update app.py(add tab Debug), time_tag change

app.py

@@ -166,6 +166,14 @@ def generate_voice_and_video(
     )
 
 
+def list_files(directory):
+    try:
+        files = os.listdir(directory)
+        return "\n".join(files)
+    except Exception as e:
+        return str(e)
+
+
 def sadtalker_demo():
     download_model()
     with gr.Blocks(
@@ -282,7 +290,7 @@ def sadtalker_demo():
                     value="",
                     visible=True,
                     lines=3,
-                    max_lines=4
+                    max_lines=4,
                 )
 
             def enable_generate(audio, text, image):
@@ -300,6 +308,7 @@ def sadtalker_demo():
 
         with gr.Tab("Lịch sử video"):
             with gr.Row(elem_classes="gr-row"):
+                refresh_btn = gr.Button("🔄 Refresh File List")
                 choices = ["Vui lòng chọn video"] + list_videos()
                 video_list = gr.Dropdown(
                     value="Vui lòng chọn video",
@@ -315,7 +324,16 @@ def sadtalker_demo():
                     scale=1,
                     show_download_button=True,
                 )
+            refresh_btn.click(fn=lambda: gr.update(choices=choices), outputs=video_list)
             video_list.change(lambda x: x, inputs=video_list, outputs=video_player)
+        with gr.Tab("Debug"):
+            with gr.Blocks() as demo:
+                directory_input = gr.Textbox(label="Enter Directory Path", value=".")
+                file_list_output = gr.Textbox(label="Files", lines=10)
+
+            directory_input.change(
+                fn=list_files, inputs=directory_input, outputs=file_list_output
+            )
 
         btn_generate.click(
             generate_voice_and_video,

src/gradio_demo.py

@@ -6,6 +6,7 @@ from src.test_audio2coeff import Audio2Coeff
 from src.facerender.animate import AnimateFromCoeff
 from src.generate_batch import get_data
 from src.generate_facerender_batch import get_facerender_data
+from datetime import datetime
 
 from src.utils.init_path import init_path
 
@@ -62,7 +63,7 @@ class SadTalker():
             raise(RuntimeError('Unknown model: {}'.format(facerender)))
             
 
-        time_tag = str(uuid.uuid4())
+        time_tag = datetime.now().strftime("%Y%m%d_%H%M%S") + "_" + str(uuid.uuid4())
         save_dir = os.path.join(result_dir, time_tag)
         os.makedirs(save_dir, exist_ok=True)