Update webgui.py

webgui.py

@@ -325,7 +325,49 @@ with gr.Blocks() as demo:
             </div>
             """)
 
-    def generate_video(uploaded_img, uploaded_audio, facemask_dilation_ratio=default_values["facemask_dilation_ratio"],
+    # def generate_video(uploaded_img, uploaded_audio, facemask_dilation_ratio=default_values["facemask_dilation_ratio"],
+    #                    facecrop_dilation_ratio=default_values["facecrop_dilation_ratio"],
+    #                    context_frames=default_values["context_frames"],
+    #                    context_overlap=default_values["context_overlap"],
+    #                    cfg=default_values["cfg"],
+    #                    steps=default_values["steps"],
+    #                    sample_rate=default_values["sample_rate"],
+    #                    fps=default_values["fps"],
+    #                    device=default_values["device"],
+    #                    width=default_values["width"],
+    #                    height=default_values["height"],
+    #                    length=default_values["length"] ):
+
+    #     final_output_path = process_video(
+    #         uploaded_img, uploaded_audio, width, height, length, seed, facemask_dilation_ratio, facecrop_dilation_ratio, context_frames, context_overlap, cfg, steps, sample_rate, fps, device
+    #     )        
+    #     output_video= final_output_path
+    #     return final_output_path
+
+    # generate_button.click(
+    #     generate_video,
+    #     inputs=[
+    #         uploaded_img,
+    #         uploaded_audio,
+    #         # width,
+    #         # height,
+    #         # length,
+    #         # seed,
+    #         # facemask_dilation_ratio,
+    #         # facecrop_dilation_ratio,
+    #         # context_frames,
+    #         # context_overlap,
+    #         # cfg,
+    #         # steps,
+    #         # sample_rate,
+    #         # fps,
+    #         # device
+    #     ],
+    #     outputs=output_video,
+    #     show_api=False
+    # )
+    def generate_video(uploaded_img, uploaded_audio,
+                       facemask_dilation_ratio=default_values["facemask_dilation_ratio"],
                        facecrop_dilation_ratio=default_values["facecrop_dilation_ratio"],
                        context_frames=default_values["context_frames"],
                        context_overlap=default_values["context_overlap"],
@@ -339,32 +381,24 @@ with gr.Blocks() as demo:
                        length=default_values["length"] ):
 
         final_output_path = process_video(
-            uploaded_img, uploaded_audio, width, height, length, seed, facemask_dilation_ratio, facecrop_dilation_ratio, context_frames, context_overlap, cfg, steps, sample_rate, fps, device
+            uploaded_img, 
+            uploaded_audio, width, height, 
+            length, facemask_dilation_ratio, 
+            facecrop_dilation_ratio, context_frames, 
+            context_overlap, cfg, steps, 
+            sample_rate, fps, device
         )        
-        output_video= final_output_path
+        output_video = final_output_path
         return final_output_path
 
     generate_button.click(
         generate_video,
         inputs=[
             uploaded_img,
-            uploaded_audio,
-            # width,
-            # height,
-            # length,
-            # seed,
-            # facemask_dilation_ratio,
-            # facecrop_dilation_ratio,
-            # context_frames,
-            # context_overlap,
-            # cfg,
-            # steps,
-            # sample_rate,
-            # fps,
-            # device
+            uploaded_audio
         ],
         outputs=output_video,
-        show_api=False
+        show_progress=True
     )
 parser = argparse.ArgumentParser(description='EchoMimic')
 parser.add_argument('--server_name', type=str, default='0.0.0.0', help='Server name')