Update app.py

app.py

@@ -148,17 +148,19 @@ def update_gallery_on_upload(input_video, session_id):
     return None, target_dir, image_paths
 
 @spaces.GPU()
-def generate_splat_from_video(video_path, session_id=None):
+def generate_splats_from_video(video_path, session_id=None):
     
     if session_id is None:
         session_id = uuid.uuid4().hex
 
-    images_folder, image_paths = extract_frames(video_path, session_id)
-    
-    return generate_splat_from_images(images_folder, session_id), image_paths
+    # unpack the 3 outputs from the helper
+    plyfile, rgb_vid, depth_vid = generate_splats_from_images(images_folder, session_id)
+
+    # return all four in order
+    return plyfile, rgb_vid, depth_vid, image_paths
     
 @spaces.GPU()
-def generate_splat_from_images(images_folder, session_id=None):
+def generate_splats_from_images(images_folder, session_id=None):
 
     if session_id is None:
         session_id = uuid.uuid4().hex
@@ -334,13 +336,13 @@ if __name__ == "__main__":
                                 depth_video,
                                 image_gallery
                             ],
-                            fn=generate_splat_from_video,
+                            fn=generate_splats_from_video,
                             cache_examples=True,
                         )
                             
                
         submit_btn.click(
-            fn=generate_splat_from_images,
+            fn=generate_splats_from_images,
             inputs=[target_dir_output, session_state],
             outputs=[reconstruction_output, rgb_video, depth_video])