Update app.py

app.py

@@ -1,13 +1,15 @@
 import asyncio
 import mimetypes
 import os
-
+import tempfile
 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 utils import get_bgm_file  # Import get_bgm_file
+from moviepy.editor import AudioFileClip, CompositeAudioClip
+from utils import (combine_videos, find_matching_image, get_pexels_image,split_audio, get_bgm_file)
+from video_processing import create_video
 
 # Danh sách giọng đọc
 VOICES = ["alloy", "echo", "fable", "onyx", "nova", "shimmer"]
@@ -170,27 +172,63 @@ def interface():
                           inputs=[text_input, voice_select],
                           outputs=[audio_output, download_audio])
 
-    with gr.Tab("Tạo Video"):  # Added the colon here
-      script_input = gr.Textbox(label="Nhập kịch bản cho video")
-      dataset_path = gr.Textbox(
-          label="Đường dẫn đến dataset ảnh",
-          placeholder="Để trống nếu không sử dụng dataset")
-      use_dataset = gr.Checkbox(label="Sử dụng dataset ảnh")
-      audio_file = gr.File(label="Chọn file âm thanh", type="filepath")
-      create_video_button = gr.Button("Tạo Video")
-      video_output = gr.Video(label="Video tạo ra")
-      download_video = gr.File(label="Tải xuống file video",
-                                 interactive=False)
-
-      create_video_button.click(fn=create_video_func,
-                                inputs=[
-                                    script_input, audio_file, dataset_path,
-                                    use_dataset
-                                ],
-                                outputs=[video_output, download_video])
-
-  return app
-
+    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)  # Mặc định là ghép ngẫu nhiên
+            bgm_files = gr.Dropdown(choices=get_bgm_file_list(), label="Chọn nhạc nền")  # Hàm get_bgm_file_list() cần được định nghĩa
+            create_video_button = gr.Button("Tạo Video")
+            video_output = gr.Video(label="Video tạo ra")
+
+        def create_video_func(script,
+                      audio_file,
+                      keywords,
+                      max_clip_duration,
+                      join_order,
+                      bgm_file):
+            """
+            Tạo video từ các thông tin đầu vào.
+            """
+            try:
+                # 1. Tính toán thời lượng video
+                audio_clip = AudioFileClip(audio_file)
+                video_duration = audio_clip.duration
+
+                # 2. Tìm kiếm và tải video từ Pexels
+                video_paths = []
+                for keyword in keywords:
+                for _ in range(
+                        int(video_duration //
+                            max_clip_duration)):  # Chia thời lượng video cho thời lượng tối đa mỗi clip
+                    image_url = get_pexels_image(keyword)
+                    if image_url:
+                        video_path = download_video_from_pexels(
+                            image_url)  # Cần định nghĩa hàm download_video_from_pexels
+                        video_paths.append(video_path)
+
+                # 3. Ghép video
+                temp_dir = tempfile.mkdtemp()
+                if join_order:  # Ghép ngẫu nhiên
+                    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)
+        
+                return final_video_path
+            except Exception as e:
+                print(f"Lỗi khi tạo video: {e}")
+                return None
+    return app
+      
 # Khởi chạy ứng dụng
 if __name__ == "__main__":
   app = interface()