Update app.py

app.py

@@ -1,3 +1,79 @@
+import asyncio
+import mimetypes
+import os
+import tempfile
+import glob
+import fitz  # PyMuPDF
+import random
+import gradio as gr
+from docx import Document
+from audio_processing import async_text_to_speech, text_to_speech
+from content_generation import create_content, CONTENT_TYPES
+from video_processing import create_video_func
+from moviepy.editor import AudioFileClip, VideoFileClip, CompositeAudioClip
+from utils import (combine_videos, get_pexels_video, get_bgm_file, download_video)
+from video_processing import create_video
+from content_generation import create_content, CONTENT_TYPES
+from openai import OpenAI
+
+# Khởi tạo client OpenAI với API key từ biến môi trường
+client = OpenAI(api_key=os.environ.get('OPENAI_API_KEY'))
+
+def create_docx(content, output_path):
+    """
+    Tạo file docx từ nội dung.
+    """
+    doc = Document()
+    doc.add_paragraph(content)
+    doc.save(output_path)
+
+def process_pdf(file_path):
+    """
+    Xử lý file PDF và trích xuất nội dung.
+    """
+    doc = fitz.open(file_path)
+    text = ""
+    for page in doc:
+        text += page.get_text()
+    return text
+
+def process_docx(file_path):
+    """
+    Xử lý file DOCX và trích xuất nội dung.
+    """
+    doc = Document(file_path)
+    text = ""
+    for para in doc.paragraphs:
+        text += para.text
+    return text
+
+def get_bgm_file_list():
+    """
+    Trả về danh sách các tệp nhạc nền.
+    """
+    # Giả sử bạn có một thư mục chứa các tệp nhạc nền
+    song_dir = "/data/bg-music"
+    return [os.path.basename(file) for file in glob.glob(os.path.join(song_dir, "*.mp3"))]
+
+def extract_key_contents(script, num_contents=10):
+    """
+    Trích xuất các ý chính từ script.
+    """
+    try:
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"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ừ."},
+                {"role": "user", "content": script}
+            ]
+        )
+        # Sửa lỗi truy cập đối tượng không thể subscript
+        key_contents = response.choices[0].message.content.split('\n')
+        return key_contents[:num_contents]
+    except Exception as e:
+        print(f"Lỗi khi trích xuất nội dung: {str(e)}")
+        return []
+
 def interface():
     with gr.Blocks() as app:
         gr.Markdown("# Ứng dụng Tạo Nội dung và Video")