Update app.py

app.py

@@ -79,10 +79,10 @@ def extract_frames_with_labels(video_path, base_output_dir="frames"):
     # Release the video capture object
     video_capture.release()
     
-    return frame_data
+    return frame_data, gr.update(open=True)
 
 # Define a function to run your script with selected inputs
-def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_frames, num_mix, ddim_steps):
+def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_frames, num_mix, ddim_steps, progress=gr.Progress(track_tqdm=True)):
 
     # Create a unique output directory name based on current date and time
     output_dir_base = "outputs"
@@ -123,6 +123,7 @@ def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_f
     except subprocess.CalledProcessError as e:
         return f"An error occurred: {e}", None
 
+
 # Set up Gradio interface
 css="""
 div#frames-gallery{
@@ -152,8 +153,8 @@ with gr.Blocks(css=css) as demo:
                     with gr.Row():
                         best_frame = gr.Number(value=36, label="Best Frame", info="specify the frame index in the driving video where the head pose best matches the source image (note: precision of best_frame index might affect the final quality)")
                         out_frames = gr.Number(value=-1, label="Out Frames", info="number of generation frames")
-                    with gr.Accordion("Driving video Frames"):
-                        driving_frames = gr.Gallery(show_label=True, columns=6, height=512, elem_id="frames-gallery")
+                    with gr.Accordion("Driving video Frames", open=False) as frames_gallery_panel:
+                        driving_frames = gr.Gallery(show_label=True, columns=6, height=380, elem_id="frames-gallery")
                 with gr.Row():
                     seed = gr.Number(value=999, label="Seed")
                     uc_scale = gr.Number(value=5, label="UC Scale")
@@ -182,10 +183,10 @@ with gr.Blocks(css=css) as demo:
                 """)
 
 
-    driving_video.upload(
+    driving_video.change(
         fn = extract_frames_with_labels,
         inputs = [driving_video],
-        outputs = [driving_frames],
+        outputs = [driving_frames, frames_gallery_panel],
         queue = False
     )