Update app.py

app.py

@@ -5,6 +5,30 @@ import datetime
 import edge_tts
 import asyncio
 import os
+import subprocess
+
+def create_video(audio_file, turn, output_video="output.mp4"):
+    # Chọn file video nền dựa trên lượt
+    background_video = "missVN.mp4" if turn == "Miss AI Vietnam" else "missCN.mp4"
+
+    # 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 lượt)
+        "-i", audio_file,      # File audio đầu vào
+        "-vf", "drawtext=text='MISS AI':fontcolor=white:fontsize=100:fontfile=/path/to/font.ttf:x=10:y=10",  # 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
 
 # Khởi tạo các mô hình từ Hugging Face
 MissAIVietnam = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -79,10 +103,13 @@ def start_debate(topic, position_1, position_2):
     history.append((initial_message, response))
 
     # Chuyển văn bản thành âm thanh với giọng của Miss AI Vietnam
-    output_file = asyncio.run(text_to_speech(response, "en-US-JennyNeural"))  # Giọng nữ tiếng Anh Mỹ
-    audio_files.append(output_file)  # Thêm file audio vào danh sách
+    output_audio = asyncio.run(text_to_speech(response, "en-US-JennyNeural"))  # Giọng nữ tiếng Anh Mỹ
+    audio_files.append(output_audio)  # Thêm file audio vào danh sách
 
-    return f"The debate has started! {turn} begins.", history, output_file
+    # Tạo video từ audio và lượt hiện tại
+    output_video = create_video(output_audio, turn)
+
+    return f"The debate has started! {turn} begins.", history, output_video
 
 # Hàm để chuyển lượt trong tranh luận
 def next_turn(topic, position_1, position_2, current_history):
@@ -105,20 +132,13 @@ def next_turn(topic, position_1, position_2, current_history):
     history.append(("", response))  # Thêm phản hồi vào lịch sử
 
     # Chuyển văn bản thành âm thanh với giọng tương ứng
-    output_file = asyncio.run(text_to_speech(response, voice))
-    audio_files.append(output_file)  # Thêm file audio vào danh sách
-
-    return f"It's now {turn}'s turn.", history, output_file
+    output_audio = asyncio.run(text_to_speech(response, voice))
+    audio_files.append(output_audio)  # Thêm file audio vào danh sách
 
-# Hàm để kết thúc tranh luận và nối các file audio
-def end_debate():
-    global audio_files
-    if not audio_files:
-        return "No debate audio found.", None
+    # Tạo video từ audio và lượt hiện tại
+    output_video = create_video(output_audio, turn)
 
-    # Nối các file audio thành một file duy nhất
-    final_audio_file = concatenate_audio_files(audio_files)
-    return "The debate has ended. Here is the full debate audio.", final_audio_file
+    return f"It's now {turn}'s turn.", history, output_video
 
 # Giao diện Gradio
 with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Roboto Mono")])) as demo:
@@ -135,18 +155,19 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Roboto Mono")]))
             status_output = gr.Textbox(label="Status", interactive=False)
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(label="Debate Arena", height=500)
-            audio_output = gr.Audio(label="Debate Audio", autoplay=True)  # Hiển thị audio hiện tại
+            video_output = gr.Video(label="Debate Video", autoplay=True)  # Hiển thị video
+            audio_output = gr.Audio(label="Debate Audio", autoplay=None)  # Hiển thị audio hiện tại
             final_audio_output = gr.Audio(label="Full Debate Audio", visible=False)  # Hiển thị audio cuối cùng
 
     start_button.click(
         fn=start_debate,
         inputs=[topic_input, position_1_input, position_2_input],
-        outputs=[status_output, chatbot, audio_output],
+        outputs=[status_output, chatbot, video_output, audio_output],
     )
     next_button.click(
         fn=next_turn,
         inputs=[topic_input, position_1_input, position_2_input, chatbot],
-        outputs=[status_output, chatbot, audio_output],
+        outputs=[status_output, chatbot, video_output, audio_output],
     )
     end_button.click(
         fn=end_debate,