Update app.py

app.py

@@ -38,10 +38,18 @@ def trim_video(video_path, output_dir="trimmed_videos", max_duration=2):
             # If the video is within the duration, return the original path
             return video_path
 
-def extract_frames_with_labels(video_path, base_output_dir="frames"):
+def load_driving_video(video_path):
     if is_shared_ui :
         video_path = trim_video(video_path)
         print("Path to the (trimmed) driving video:", video_path)
+        frames_data = extract_frames_with_labels(video_path)
+        return video_path, frames_data, gr.update(open="True")
+    else:
+        frames_data = extract_frames_with_labels(video_path)
+        return video_path, frames_data, gr.update(open="True")
+        
+def extract_frames_with_labels(video_path, base_output_dir="frames"):
+
     # Generate a timestamped folder name
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     output_dir = os.path.join(base_output_dir, f"frames_{timestamp}")
@@ -79,7 +87,7 @@ def extract_frames_with_labels(video_path, base_output_dir="frames"):
     # Release the video capture object
     video_capture.release()
     
-    return frame_data, gr.update(open=True)
+    return  frame_data
 
 # 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, progress=gr.Progress(track_tqdm=True)):
@@ -130,6 +138,7 @@ div#frames-gallery{
     overflow: scroll!important;
 }
 """
+example_frame_data = extract_frames_with_labels("./assets/driving_video.mp4")
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# X-Portrait: Expressive Portrait Animation with Hierarchical Motion Attention")
@@ -164,9 +173,9 @@ with gr.Blocks(css=css) as demo:
                 submit_btn = gr.Button("Submit")
                 gr.Examples(
                     examples=[
-                        ["./assets/source_image.png", "./assets/driving_video.mp4"]
+                        ["./assets/source_image.png", "./assets/driving_video.mp4", example_frame_data]
                     ],
-                    inputs=[source_image, driving_video]
+                    inputs=[source_image, driving_video, driving_frames]
                 )
             with gr.Column():
                 video_output = gr.Video(label="Output Video")
@@ -184,10 +193,10 @@ with gr.Blocks(css=css) as demo:
                 """)
 
 
-    driving_video.change(
-        fn = extract_frames_with_labels,
+    driving_video.upload(
+        fn = load_driving_video,
         inputs = [driving_video],
-        outputs = [driving_frames, frames_gallery_panel],
+        outputs = [driving_video, driving_frames, frames_gallery_panel],
         queue = False
     )