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
+    return None, target_dir, image_paths, 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
+    return None, target_dir, image_paths, image_paths
 
 @spaces.GPU()
 def generate_splats_from_video(video_path, session_id=None):
@@ -408,19 +408,19 @@ if __name__ == "__main__":
                
         submit_btn.click(
             fn=generate_splats_from_images,
-            inputs=[image_gallery, session_state],
+            inputs=[input_images, 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],
+            outputs=[reconstruction_output, target_dir_output, image_gallery, input_images],
         )
 
         input_images.upload(
             fn=update_gallery_on_images_upload,
             inputs=[input_images, session_state],
-            outputs=[reconstruction_output, target_dir_output, image_gallery],
+            outputs=[reconstruction_output, target_dir_output, image_gallery, input_images],
         )
 
         demo.unload(cleanup)