Update app.py

app.py

@@ -223,7 +223,7 @@ def update_gallery_on_video_upload(input_video, session_id):
         return None, None, None
         
     target_dir, image_paths = extract_frames(input_video, session_id)
-    return None, target_dir, image_paths, image_paths
+    return None, target_dir, image_paths
 
 def update_gallery_on_images_upload(input_images, session_id):
 
@@ -231,7 +231,7 @@ def update_gallery_on_images_upload(input_images, session_id):
         return None, None, None
         
     target_dir, image_paths = extract_images(input_images, session_id)
-    return None, target_dir, image_paths, image_paths
+    return None, target_dir, image_paths
 
 @spaces.GPU()
 def generate_splats_from_video(video_path, session_id=None):
@@ -247,6 +247,14 @@ def generate_splats_from_video(video_path, session_id=None):
 @spaces.GPU()
 def generate_splats_from_images(image_paths, session_id=None):
 
+    processed_image_paths = []
+
+    for file_data in image_paths:
+        if isinstance(file_data, tuple):
+            file_path, _ = file_data
+            processed_image_paths.append(file_path)
+
+    image_paths = processed_image_paths
     print(image_paths)
 
     if session_id is None:
@@ -408,19 +416,19 @@ if __name__ == "__main__":
                
         submit_btn.click(
             fn=generate_splats_from_images,
-            inputs=[input_images, session_state],
+            inputs=[image_gallery, session_state],
             outputs=[reconstruction_output, rgb_video, depth_video])
 
         input_video.upload(
             fn=update_gallery_on_video_upload,
             inputs=[input_video, session_state],
-            outputs=[reconstruction_output, target_dir_output, image_gallery, input_images],
+            outputs=[reconstruction_output, target_dir_output, image_gallery],
         )
 
         input_images.upload(
             fn=update_gallery_on_images_upload,
             inputs=[input_images, session_state],
-            outputs=[reconstruction_output, target_dir_output, image_gallery, input_images],
+            outputs=[reconstruction_output, target_dir_output, image_gallery],
         )
 
         demo.unload(cleanup)