Update app.py
Browse files
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 =
|
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ừ."},
|