TDN-M commited on
Commit
be74313
·
verified ·
1 Parent(s): eebfcc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import asyncio
2
  import mimetypes
3
- import openai
4
  import os
5
  import tempfile
6
  import glob
@@ -15,6 +14,10 @@ from moviepy.editor import AudioFileClip, VideoFileClip, CompositeAudioClip
15
  from utils import (combine_videos, get_pexels_video, get_bgm_file, download_video)
16
  from video_processing import create_video
17
  from content_generation import create_content, CONTENT_TYPES
 
 
 
 
18
 
19
  def create_docx(content, output_path):
20
  """
@@ -52,18 +55,12 @@ def get_bgm_file_list():
52
  song_dir = "/data/bg-music"
53
  return [os.path.basename(file) for file in glob.glob(os.path.join(song_dir, "*.mp3"))]
54
 
55
- # Lấy API key từ biến môi trường
56
- OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
57
-
58
- # Khởi tạo client OpenAI
59
- openai.api_key = OPENAI_API_KEY
60
-
61
  def extract_key_contents(script, num_contents=10):
62
  """
63
  Trích xuất các ý chính từ script.
64
  """
65
  try:
66
- response = openai.ChatCompletion.create(
67
  model="gpt-3.5-turbo",
68
  messages=[
69
  {"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},
 
1
  import asyncio
2
  import mimetypes
 
3
  import os
4
  import tempfile
5
  import glob
 
14
  from utils import (combine_videos, get_pexels_video, get_bgm_file, download_video)
15
  from video_processing import create_video
16
  from content_generation import create_content, CONTENT_TYPES
17
+ from openai import OpenAI
18
+
19
+ # Khởi tạo client OpenAI
20
+ client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])
21
 
22
  def create_docx(content, output_path):
23
  """
 
55
  song_dir = "/data/bg-music"
56
  return [os.path.basename(file) for file in glob.glob(os.path.join(song_dir, "*.mp3"))]
57
 
 
 
 
 
 
 
58
  def extract_key_contents(script, num_contents=10):
59
  """
60
  Trích xuất các ý chính từ script.
61
  """
62
  try:
63
+ response = client.completions.create(
64
  model="gpt-3.5-turbo",
65
  messages=[
66
  {"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},