Update app.py

app.py

@@ -55,21 +55,15 @@ async def convert_text_to_speech(text, language='vi', output_file="output.wav"):
     except Exception as e:
         raise Exception(f"Lỗi khi chuyển đổi văn bản thành giọng nói: {str(e)}")
 
-def create_video(audio_file, mc_choice, text_color, output_video="output.mp4"):
+def create_video(audio_file, custom_text, text_color, output_video="output.mp4"):
     """
-    Tạo video với người MC dẫn chương trình.
+    Tạo video với văn bản tùy chỉnh.
     """
-    # Chọn file video nền dựa trên lựa chọn MC
-    background_videos = {
-        "Ms. Leah": "mc1_background.mp4",
-        "Ms. Jenny": "mc2_background.mp4",
-        "Ms. Soo-jin": "mc3_background.mp4",
-        "Ms. Daisy": "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
+    # Chọn file video nền mặc định
+    background_video = "default_background.mp4"  # Đặt tên file video nền mặc định
     
     # Định nghĩa văn bản trên video
-    text_on_video = f"MC: {mc_choice}"
+    text_on_video = custom_text if custom_text else ""
     
     # Định nghĩa màu sắc chữ
     color_map = {
@@ -85,7 +79,7 @@ def create_video(audio_file, mc_choice, text_color, output_video="output.mp4"):
     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", background_video,  # Video nền mặc định
         "-i", audio_file,      # File audio đầu vào
         "-vf", f"drawtext=text='{text_on_video}':fontcolor={font_color}:fontsize=50:fontfile=/LHanoienne.otf:x=10:y=10:box=1:[email protected]:boxborderw=5",  # Văn bản trên video
         "-t", "45.7",          # Thời lượng video
@@ -117,7 +111,7 @@ def interface():
                     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")
+                    custom_text_input = gr.Textbox(label="Nhập văn bản tùy chỉnh (trống nếu không muốn hiển thị)", value="")
                     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)
@@ -154,7 +148,7 @@ def interface():
                     status = f"Đã xảy ra lỗi: {str(e)}"
                     return "", None, True, None, status
             
-            async def confirm_content(content_output, use_generated_text, manual_text_input, language, mc_choice, text_color):
+            async def confirm_content(content_output, use_generated_text, manual_text_input, language, custom_text_input, text_color):
                 try:
                     if use_generated_text:
                         text_to_convert = content_output
@@ -170,7 +164,7 @@ def interface():
                     audio_path = await convert_text_to_speech(text_to_convert, language)
                     status_message.value = "Đã chuyển đổi văn bản thành giọng nói thành công!"
                     status_message.value = "Đang tạo video..."
-                    video_path = create_video(audio_path, mc_choice, text_color)
+                    video_path = create_video(audio_path, custom_text_input, text_color)
                     status_message.value = "Đã tạo video thành công!"
                     return docx_path, audio_path, video_path, status_message.value
                 except Exception as e:
@@ -187,7 +181,7 @@ def interface():
             
             # Liên kết sự kiện click của nút Xác nhận nội dung
             confirm_button.click(confirm_content,
-                                 inputs=[content_output, use_generated_text, manual_text_input, language, mc_choice, text_color],
+                                 inputs=[content_output, use_generated_text, manual_text_input, language, custom_text_input, text_color],
                                  outputs=[download_docx, download_audio, download_video, status_message])
     
     return app