Update app.py

app.py

@@ -84,6 +84,14 @@ def process_docx(file_path):
         text += para.text
     return text
 
+def get_bgm_file_list():
+    """
+    Trả về danh sách các tệp nhạc nền.
+    """
+    # Giả sử bạn có một thư mục chứa các tệp nhạc nền
+    song_dir = "/data/bg-music"
+    return [os.path.basename(file) for file in glob.glob(os.path.join(song_dir, "*.mp3"))]
+
 # Giao diện Gradio
 def interface():
     with gr.Blocks() as app:
@@ -171,8 +179,7 @@ def interface():
             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")
+            bgm_files = gr.Dropdown(choices=get_bgm_file_list(), label="Chọn nhạc nền")
             video_output = gr.Video(label="Video tạo ra")
 
             def create_video_func(script, audio_file, keywords, max_clip_duration, join_order, bgm_file):