Update app.py

app.py

@@ -1,219 +1,226 @@
-import asyncio
-import mimetypes
-import openai
-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
-
-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"))]
-
-# Lấy API key từ biến môi trường
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-
-# Khởi tạo client OpenAI
-openai.api_key = OPENAI_API_KEY
-    
-def extract_key_contents(script, num_contents=10):
-    """
-    Trích xuất các ý chính từ script.
-    """
-    try:
-        response = openai.ChatCompletion.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}
-            ]
-        )
-        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 []
-
-# Giao diện Gradio
-def interface():
-    with gr.Blocks() as app:
-        gr.Markdown("# Ứng dụng Tạo Nội dung và Video")
-
-        with gr.Tab("Tạo Nội dung"):
-            prompt = gr.Textbox(label="Nhập yêu cầu nội dung")
-            file_upload = gr.File(label="Tải lên file kèm theo", type="filepath")
-
-            # Sử dụng gr.Radio thay vì gr.CheckboxGroup
-            content_type = gr.Radio(label="Chọn loại nội dung",
-                                    choices=CONTENT_TYPES,
-                                    value=None)  # Giá trị mặc định là không có gì được chọn
-
-            content_button = gr.Button("Tạo Nội dung")
-            content_output = gr.Textbox(label="Nội dung tạo ra", interactive=True)
-            confirm_button = gr.Button("Xác nhận nội dung")
-            download_docx = gr.File(label="Tải xuống file DOCX", interactive=False)
-            download_audio = gr.File(label="Tải xuống file âm thanh", interactive=False)
-            status_message = gr.Label(label="Trạng thái")
-
-            def generate_content(prompt, file, content_type):
-                try:
-                    status = "Đang xử lý..."
-                    if file and os.path.exists(file):
-                        mime_type, _ = mimetypes.guess_type(file)
-                        if mime_type == "application/pdf":
-                            file_content = process_pdf(file)
-                            prompt = f"{prompt}\n\nDưới đây là nội dung của file tài liệu:\n\n{file_content}"
-                        elif mime_type in (
-                            "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
-                            "application/msword"):
-                            file_content = process_docx(file)
-                            prompt = f"{prompt}\n\nDưới đây là nội dung của file tài liệu:\n\n{file_content}"
-                        else:
-                            raise ValueError("Định dạng file không được hỗ trợ.")
-
-                    if not content_type:
-                        raise ValueError("Vui lòng chọn một loại nội dung")
-
-                    script_content = create_content(prompt, content_type, "Tiếng Việt")
-                    docx_path = "script.docx"
-                    create_docx(script_content, docx_path)
-
-                    status = "Đã tạo nội dung thành công!"
-                    return script_content, docx_path, status
-                except Exception as e:
-                    status = f"Đã xảy ra lỗi: {str(e)}"
-                    return "", None, status
-
-            async def confirm_content(content):
-                docx_path = "script.docx"
-                create_docx(content, docx_path)
-
-                audio_path = await async_text_to_speech(content, "alloy", "Tiếng Việt")
-                return docx_path, audio_path, "Nội dung đã được xác nhận và âm thanh đã được tạo!"
-
-            content_button.click(generate_content,
-                                 inputs=[prompt, file_upload, content_type],
-                                 outputs=[content_output, download_docx, status_message])
-
-            confirm_button.click(lambda x: asyncio.run(confirm_content(x)),
-                                 inputs=[content_output],
-                                 outputs=[download_docx, download_audio, status_message])
-        
-        # Định nghĩa danh sách giọng đọc
-        VOICES = ["alloy", "echo", "fable", "onyx", "nova", "shimmer"]
-
-        with gr.Tab("Tạo Âm thanh"):
-            text_input = gr.Textbox(label="Nhập văn bản để chuyển đổi")
-            voice_select = gr.Dropdown(label="Chọn giọng đọc", choices=VOICES)  # Dropdown cho voice_select
-            audio_button = gr.Button("Tạo Âm thanh")
-            audio_output = gr.Audio(label="Âm thanh tạo ra")
-            download_audio = gr.File(label="Tải xuống file âm thanh", interactive=False)
-
-            def text_to_speech_func(text, voice):
-                try:
-                    audio_path = text_to_speech(text, voice, "Tiếng Việt")
-                    return audio_path, audio_path
-                except Exception as e:
-                    print(f"Lỗi khi chuyển đổi văn bản thành giọng nói: {e}")
-                    return None, None
-
-            audio_button.click(text_to_speech_func,
-                               inputs=[text_input, voice_select],
-                               outputs=[audio_output, download_audio])
-
-        with gr.Tab("Tạo Video"):
-            script_input = gr.Textbox(label="Nhập kịch bản")
-            audio_file = gr.File(label="Chọn file âm thanh", type="filepath")
-            keywords_output = gr.Textbox(label="Từ khóa", interactive=True)
-            max_clip_duration = gr.Slider(minimum=2, maximum=5, step=1, label="Thời lượng tối đa mỗi video (giây)")
-            join_order = gr.Checkbox(label="Ghép ngẫu nhiên", value=True)
-            bgm_files = gr.Dropdown(choices=get_bgm_file_list(), label="Chọn nhạc nền")
-            video_output = gr.Video(label="Video tạo ra")
-            video_button = gr.Button("Tạo Video")
-            status_message = gr.Label(label="Trạng thái")  # Thêm thông báo trạng thái
-
-            def create_video_func(script, audio_file, max_clip_duration, join_order, bgm_file):
-                """ Tạo video từ các thông tin đầu vào. """
-                try:
-                    status_message.update("Đang xử lý...")  # Cập nhật trạng thái
-
-                    # 1. Tính toán thời lượng video
-                    audio_clip = AudioFileClip(audio_file)
-                    video_duration = audio_clip.duration
-
-                    # 2. Trích xuất từ khóa từ kịch bản
-                    keywords = extract_key_contents(script)
-                    video_paths = []
-                    for keyword in keywords:
-                        video_url = get_pexels_video(keyword.strip())
-                        if video_url:
-                            video_path = download_video(video_url)
-                            video_paths.append(video_path)
-
-                    # 3. Ghép video
-                    temp_dir = tempfile.mkdtemp()
-                    if join_order:
-                        random.shuffle(video_paths)
-                    combined_video_path = os.path.join(temp_dir, "combined_video.mp4")
-                    combine_videos(combined_video_path, video_paths, audio_file, max_clip_duration)
-
-                    # 4. Gộp audio và nhạc nền
-                    final_video_path = "final_video.mp4"
-                    bgm_clip = AudioFileClip(bgm_file)
-                    final_audio = CompositeAudioClip([audio_clip, bgm_clip])
-                    final_video = VideoFileClip(combined_video_path).set_audio(final_audio)
-                    final_video.write_videofile(final_video_path)
-
-                    status_message.update("Video đã được tạo thành công!")  # Cập nhật trạng thái
-                    return final_video_path
-                except Exception as e:
-                    status_message.update(f"Lỗi khi tạo video: {e}")  # Cập nhật trạng thái
-                    return None
-                    return app
-
-# Khởi chạy ứng dụng
-if __name__ == "__main__":
-    app = interface()
+import asyncio
+import mimetypes
+import openai
+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
+
+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"))]
+
+# Lấy API key từ biến môi trường
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+# Khởi tạo client OpenAI
+openai.api_key = OPENAI_API_KEY
+    
+def extract_key_contents(script, num_contents=10):
+    """
+    Trích xuất các ý chính từ script.
+    """
+    try:
+        response = openai.ChatCompletion.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}
+            ]
+        )
+        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 []
+
+# ... existing imports and functions ...
+
+# Giao diện Gradio
+def interface():
+    with gr.Blocks() as app:
+        gr.Markdown("# Ứng dụng Tạo Nội dung và Video")
+
+        with gr.Tab("Tạo Nội dung"):
+            prompt = gr.Textbox(label="Nhập yêu cầu nội dung")
+            file_upload = gr.File(label="Tải lên file kèm theo", type="filepath")
+
+            # Sử dụng gr.Radio thay vì gr.CheckboxGroup
+            content_type = gr.Radio(label="Chọn loại nội dung",
+                                    choices=CONTENT_TYPES,
+                                    value=None)  # Giá trị mặc định là không có gì được chọn
+
+            content_button = gr.Button("Tạo Nội dung")
+            content_output = gr.Textbox(label="Nội dung tạo ra", interactive=True)
+            confirm_button = gr.Button("Xác nhận nội dung")
+            download_docx = gr.File(label="Tải xuống file DOCX", interactive=False)
+            download_audio = gr.File(label="Tải xuống file âm thanh", interactive=False)
+            status_message = gr.Label(label="Trạng thái")
+
+            def generate_content(prompt, file, content_type):
+                try:
+                    status = "Đang xử lý..."
+                    if file and os.path.exists(file):
+                        mime_type, _ = mimetypes.guess_type(file)
+                        if mime_type == "application/pdf":
+                            file_content = process_pdf(file)
+                            prompt = f"{prompt}\n\nDưới đây là nội dung của file tài liệu:\n\n{file_content}"
+                        elif mime_type in (
+                            "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+                            "application/msword"):
+                            file_content = process_docx(file)
+                            prompt = f"{prompt}\n\nDưới đây là nội dung của file tài liệu:\n\n{file_content}"
+                        else:
+                            raise ValueError("Định dạng file không được hỗ trợ.")
+
+                    if not content_type:
+                        raise ValueError("Vui lòng chọn một loại nội dung")
+
+                    script_content = create_content(prompt, content_type, "Tiếng Việt")
+                    docx_path = "script.docx"
+                    create_docx(script_content, docx_path)
+
+                    status = "Đã tạo nội dung thành công!"
+                    return script_content, docx_path, status
+                except Exception as e:
+                    status = f"Đã xảy ra lỗi: {str(e)}"
+                    return "", None, status
+
+            async def confirm_content(content):
+                docx_path = "script.docx"
+                create_docx(content, docx_path)
+
+                audio_path = await async_text_to_speech(content, "alloy", "Tiếng Việt")
+                return docx_path, audio_path, "Nội dung đã được xác nhận và âm thanh đã được tạo!"
+
+            content_button.click(generate_content,
+                                 inputs=[prompt, file_upload, content_type],
+                                 outputs=[content_output, download_docx, status_message])
+
+            confirm_button.click(lambda x: asyncio.run(confirm_content(x)),
+                                 inputs=[content_output],
+                                 outputs=[download_docx, download_audio, status_message])
+        
+        # Định nghĩa danh sách giọng đọc
+        VOICES = ["alloy", "echo", "fable", "onyx", "nova", "shimmer"]
+
+        with gr.Tab("Tạo Âm thanh"):
+            text_input = gr.Textbox(label="Nhập văn bản để chuyển đổi")
+            voice_select = gr.Dropdown(label="Chọn giọng đọc", choices=VOICES)  # Dropdown cho voice_select
+            audio_button = gr.Button("Tạo Âm thanh")
+            audio_output = gr.Audio(label="Âm thanh tạo ra")
+            download_audio = gr.File(label="Tải xuống file âm thanh", interactive=False)
+
+            def text_to_speech_func(text, voice):
+                try:
+                    audio_path = text_to_speech(text, voice, "Tiếng Việt")
+                    return audio_path, audio_path
+                except Exception as e:
+                    print(f"Lỗi khi chuyển đổi văn bản thành giọng nói: {e}")
+                    return None, None
+
+            audio_button.click(text_to_speech_func,
+                               inputs=[text_input, voice_select],
+                               outputs=[audio_output, download_audio])
+
+        with gr.Tab("Tạo Video"):
+            script_input = gr.Textbox(label="Nhập kịch bản")
+            audio_file = gr.File(label="Chọn file âm thanh", type="filepath")
+            keywords_output = gr.Textbox(label="Từ khóa", interactive=True)
+            max_clip_duration = gr.Slider(minimum=2, maximum=5, step=1, label="Thời lượng tối đa mỗi video (giây)")
+            join_order = gr.Checkbox(label="Ghép ngẫu nhiên", value=True)
+            bgm_files = gr.Dropdown(choices=get_bgm_file_list(), label="Chọn nhạc nền")
+            video_output = gr.Video(label="Video tạo ra")
+            video_button = gr.Button("Tạo Video")
+            status_message = gr.Label(label="Trạng thái")  # Thêm thông báo trạng thái
+
+            def create_video_func(script, audio_file, max_clip_duration, join_order, bgm_file):
+                """ Tạo video từ các thông tin đầu vào. """
+                try:
+                    status_message.update("Đang xử lý...")  # Cập nhật trạng thái
+
+                    # 1. Tính toán thời lượng video
+                    audio_clip = AudioFileClip(audio_file)
+                    video_duration = audio_clip.duration
+
+                    # 2. Trích xuất từ khóa từ kịch bản
+                    keywords = extract_key_contents(script)
+                    video_paths = []
+                    for keyword in keywords:
+                        video_url = get_pexels_video(keyword.strip())
+                        if video_url:
+                            video_path = download_video(video_url)
+                            video_paths.append(video_path)
+
+                    # 3. Ghép video
+                    temp_dir = tempfile.mkdtemp()
+                    if join_order:
+                        random.shuffle(video_paths)
+                    combined_video_path = os.path.join(temp_dir, "combined_video.mp4")
+                    combine_videos(combined_video_path, video_paths, audio_file, max_clip_duration)
+
+                    # 4. Gộp audio và nhạc nền
+                    final_video_path = "final_video.mp4"
+                    bgm_clip = AudioFileClip(bgm_file)
+                    final_audio = CompositeAudioClip([audio_clip, bgm_clip])
+                    final_video = VideoFileClip(combined_video_path).set_audio(final_audio)
+                    final_video.write_videofile(final_video_path)
+
+                    status_message.update("Video đã được tạo thành công!")  # Cập nhật trạng thái
+                    return final_video_path
+                except Exception as e:
+                    status_message.update(f"Lỗi khi tạo video: {e}")  # Cập nhật trạng thái
+                    return None
+
+            video_button.click(create_video_func,
+                               inputs=[script_input, audio_file, max_clip_duration, join_order, bgm_files],
+                               outputs=[video_output, status_message])
+
+    return app
+
+# Khởi chạy ứng dụng
+if __name__ == "__main__":
+    app = interface()
     app.launch()