main

app.py

@@ -6,28 +6,28 @@ def run_ttv():
     try:
         result = subprocess.run(['python', 'ttv.py'], capture_output=True, text=True)
         
-        # Lấy danh sách file mp3 trong thư mục temp
-        mp3_files = [f for f in os.listdir('temp') if f.endswith('.mp3')]
-        if mp3_files:
-            # Trả về đường dẫn đến file mp3 mới nhất
-            latest_mp3 = os.path.join('temp', mp3_files[-1])
-            return result.stdout, latest_mp3
+        # Lấy danh sách file video trong thư mục temp
+        video_files = [f for f in os.listdir('temp') if f.endswith(('.mp4', '.avi', '.mov'))]
+        if video_files:
+            # Trả về đường dẫn đến file video mới nhất
+            latest_video = os.path.join('temp', video_files[-1])
+            return result.stdout, latest_video
         return result.stdout, None
     except Exception as e:
         return f"Có lỗi xảy ra: {str(e)}", None
 
 # Tạo giao diện
 with gr.Blocks() as demo:
-    gr.Markdown("# Ứng dụng Text-to-Voice")
+    gr.Markdown("# Ứng dụng Text-to-Video")
     
     with gr.Row():
-        run_button = gr.Button("Chạy Text-to-Voice")
+        run_button = gr.Button("Chạy Text-to-Video")
         output = gr.Textbox(label="Kết quả")
     
-    # Thêm audio player
-    audio_output = gr.Audio(label="File âm thanh đã tạo")
+    # Thay đổi từ Audio sang Video
+    video_output = gr.Video(label="File video đã tạo")
     
-    run_button.click(fn=run_ttv, outputs=[output, audio_output])
+    run_button.click(fn=run_ttv, outputs=[output, video_output])
 
 # Chạy ứng dụng
 if __name__ == "__main__":