Update app.py

app.py

@@ -240,12 +240,14 @@ def generate_splats_from_video(video_path, session_id=None):
         session_id = uuid.uuid4().hex
 
     images_folder, image_paths = extract_frames(video_path, session_id)
-    plyfile, rgb_vid, depth_vid = generate_splats_from_images(images_folder, session_id)
+    plyfile, rgb_vid, depth_vid = generate_splats_from_images(image_paths, session_id)
 
     return plyfile, rgb_vid, depth_vid, image_paths
     
 @spaces.GPU()
-def generate_splats_from_images(images_folder, session_id=None):
+def generate_splats_from_images(image_paths, session_id=None):
+
+    print(image_paths)
 
     if session_id is None:
         session_id = uuid.uuid4().hex
@@ -256,13 +258,6 @@ def generate_splats_from_images(images_folder, session_id=None):
 
     base_dir = os.path.join(os.environ["ANYSPLAT_PROCESSED"], session_id)
 
-    all_files = (
-        sorted(os.listdir(images_folder))
-        if os.path.isdir(images_folder)
-        else []
-    )
-    all_files = [f"{i}: {filename}" for i, filename in enumerate(all_files)]
-
     print("Running run_model...")
     with torch.no_grad():
         plyfile, video, depth_colored = get_reconstructed_scene(base_dir, model, device)
@@ -413,7 +408,7 @@ if __name__ == "__main__":
                
         submit_btn.click(
             fn=generate_splats_from_images,
-            inputs=[target_dir_output, session_state],
+            inputs=[image_gallery, session_state],
             outputs=[reconstruction_output, rgb_video, depth_video])
 
         input_video.upload(