Update app.py

app.py

@@ -2,16 +2,18 @@ import asyncio
 import mimetypes
 import os
 import tempfile
-import glob
 import fitz  # PyMuPDF
 import random
 import gradio as gr
 from docx import Document
 from content_generation import create_content, CONTENT_TYPES
 from openai import OpenAI
+from gradio_client import Client, handle_file
+import subprocess
 
 # 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'))
+client_openai = OpenAI(api_key=os.environ.get('OPENAI_API_KEY'))
+client_vixtts = Client("thinhlpg/vixtts-demo")
 
 def create_docx(content, output_path):
     """
@@ -41,10 +43,66 @@ def process_docx(file_path):
         text += para.text
     return text
 
+def convert_text_to_speech(text, language='vi'):
+    """
+    Chuyển đổi văn bản thành giọng nói.
+    """
+    result = client_vixtts.predict(
+        prompt=text,
+        language=language,
+        audio_file_pth=handle_file(''),
+        normalize_text=True,
+        api_name="/predict"
+    )
+    return result.audio_file_pth
+
+def create_video(audio_file, mc_choice, text_color, output_video="output.mp4"):
+    """
+    Tạo video với người MC dẫn chương trình.
+    """
+    # Chọn file video nền dựa trên lựa chọn MC
+    background_videos = {
+        "MC1": "mc1_background.mp4",
+        "MC2": "mc2_background.mp4",
+        "MC3": "mc3_background.mp4",
+        "MC4": "mc4_background.mp4"
+    }
+    background_video = background_videos.get(mc_choice, "mc1_background.mp4")  # Mặc định là MC1 nếu không tìm thấy
+    
+    # Định nghĩa văn bản trên video
+    text_on_video = f"MC: {mc_choice}"
+    
+    # Định nghĩa màu sắc chữ
+    color_map = {
+        "Trắng": "white",
+        "Đỏ": "red",
+        "Xanh dương": "blue",
+        "Xanh lá": "green",
+        "Vàng": "yellow"
+    }
+    font_color = color_map.get(text_color, "white")  # Mặc định là trắng nếu không tìm thấy
+    
+    # Lệnh ffmpeg để tạo video
+    command = [
+        "ffmpeg",
+        "-stream_loop", "-1",  # Lặp lại video nền vô hạn
+        "-i", background_video,  # Video nền (tùy thuộc vào MC)
+        "-i", audio_file,      # File audio đầu vào
+        "-vf", f"drawtext=text='{text_on_video}':fontcolor={font_color}:fontsize=50:fontfile=/path/to/font.ttf:x=(w-text_w)/2:y=10:box=1:[email protected]:boxborderw=5",  # Văn bản trên video
+        "-t", "45.7",          # Thời lượng video
+        "-c:v", "libx264",     # Codec video
+        "-c:a", "aac",         # Codec audio
+        "-shortest",           # Dừng khi audio kết thúc
+        "-y",                  # Ghi đè file nếu tồn tại
+        output_video           # File video đầu ra
+    ]
+    # Chạy lệnh ffmpeg
+    subprocess.run(command, check=True)
+    return output_video
+
 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"):
             with gr.Row():
                 with gr.Column():
@@ -57,10 +115,20 @@ def interface():
                 
                 with gr.Column():
                     content_output = gr.Textbox(label="Nội dung tạo ra", interactive=True)
+                    use_generated_text = gr.Checkbox(label="Sử dụng văn bản được tạo bởi LLM", value=True)
+                    manual_text_input = gr.Textbox(label="Nhập văn bản thủ công", visible=False)
+                    language = gr.Dropdown(label="Chọn ngôn ngữ", choices=["vi", "en"], value="vi")
+                    mc_choice = gr.Dropdown(label="Chọn MC", choices=["MC1", "MC2", "MC3", "MC4"], value="MC1")
+                    text_color = gr.Dropdown(label="Chọn màu chữ", choices=["Trắng", "Đỏ", "Xanh dương", "Xanh lá", "Vàng"], value="Trắng")
                     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)
+                    download_video = gr.File(label="Tải xuống file video", interactive=False)
                     status_message = gr.Label(label="Trạng thái")
-
+            
+            def toggle_manual_text(use_generated_text):
+                return not use_generated_text
+            
             def generate_content(prompt, file, content_type):
                 try:
                     status = "Đang xử lý..."
@@ -76,32 +144,43 @@ def interface():
                             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
+                    return script_content, docx_path, True, None, status
                 except Exception as e:
                     status = f"Đã xảy ra lỗi: {str(e)}"
-                    return "", None, status
-
-            async def confirm_content(content):
+                    return "", None, True, None, status
+            
+            async def confirm_content(content_output, use_generated_text, manual_text_input, language, mc_choice, text_color):
+                if use_generated_text:
+                    text_to_convert = content_output
+                else:
+                    text_to_convert = manual_text_input
+                
+                if not text_to_convert:
+                    return None, None, None, "Vui lòng cung cấp văn bản để chuyển đổi."
+                
                 docx_path = "script.docx"
-                create_docx(content, docx_path)
-
+                create_docx(text_to_convert, docx_path)
+                audio_path = convert_text_to_speech(text_to_convert, language)
+                video_path = create_video(audio_path, mc_choice, text_color)
+                return docx_path, audio_path, video_path, "Đã xác nhận và tạo video thành công!"
+            
+            use_generated_text.change(toggle_manual_text, inputs=[use_generated_text], outputs=[manual_text_input.visible])
             content_button.click(generate_content,
                                  inputs=[prompt, file_upload, content_type],
-                                 outputs=[content_output, download_docx, status_message])
-
+                                 outputs=[content_output, download_docx, use_generated_text, manual_text_input, status_message])
+            
+            confirm_button.click(confirm_content,
+                                 inputs=[content_output, use_generated_text, manual_text_input, language, mc_choice, text_color],
+                                 outputs=[download_docx, download_audio, download_video, status_message])
+    
     return app
 
-
-
 # Khởi chạy ứng dụng
 if __name__ == "__main__":
     app = interface()