iterative image adding

.vscode/launch.json

@@ -0,0 +1,17 @@
+{
+    "version": "0.2.0",
+    "configurations": [
+
+        {
+        "name": "Python: Current File (venv)",
+        "type": "debugpy",
+        "request": "launch",
+        "program": "${file}",
+        "console": "integratedTerminal",
+        "cwd": "${workspaceFolder}",
+        "envFile": "${workspaceFolder}/.env",
+        "python": "/home/robot_op/miniconda3/envs/stream3r/bin/python",
+        "justMyCode": true
+        },
+    ]
+}

app.py

@@ -211,7 +211,7 @@ def run_model(target_dir: str, model: STream3R, mode: str="causal", streaming: b
     print(f"Running inference in {'streaming' if streaming else 'batch'} mode...")
     dtype = torch.bfloat16 if torch.cuda.get_device_capability()[0] >= 8 else torch.float16
 
-    session_cache_path: str | None = None
+    session_cache_path: Optional[str] = None
 
     with torch.no_grad():
         with torch.amp.autocast(dtype=dtype, device_type=device):
@@ -370,8 +370,8 @@ def update_gallery_on_upload(input_video, input_images, input_zip, session_state
     Handle any new uploads (video, images, or zip) and render preview.
     """
     if not input_video and not input_images and not input_zip and not session_state:
-        default_updates = [gr.update()] * 9
-        return (None, current_target_dir, None, None, None, *default_updates)
+        default_updates = [gr.update()] * 10
+        return (gr.update(), current_target_dir, None, None, None, *default_updates)
 
     target_dir, image_paths, session_loaded = handle_uploads(
         input_video,
@@ -400,7 +400,8 @@ def update_gallery_on_upload(input_video, input_images, input_zip, session_state
     mask_sky_update = gr.update()
     prediction_mode_update = gr.update()
 
-    reconstruction_value = None
+    reconstruction_update = gr.update()
+    session_state_reset = gr.update()
 
     if session_loaded:
         settings = load_session_settings(target_dir)
@@ -450,16 +451,19 @@ def update_gallery_on_upload(input_video, input_images, input_zip, session_state
                     f"glbscene_{conf_val}_{sanitize_frame_filter_label(frame_value)}_maskb{mask_black_val}_maskw{mask_white_val}_cam{show_cam_val}_sky{mask_sky_val}_pred{pred_mode_val.replace(' ', '_')}_mode{mode_val}.glb",
                 )
                 if os.path.exists(glb_candidate):
-                    reconstruction_value = glb_candidate
+                    reconstruction_update = glb_candidate
             except (KeyError, AttributeError):
                 pass
 
+        session_state_reset = gr.update(value=None)
+
     return (
-        reconstruction_value,
+        reconstruction_update,
         target_dir,
         image_paths,
         message,
         None,
+        session_state_reset,
         streaming_update,
         mode_update,
         conf_update,
@@ -939,6 +943,7 @@ with gr.Blocks(
             dropdown,
             image_paths,
             session_file,
+            gr.update(value=None),
             False,
             mode,
             conf_thres,
@@ -974,6 +979,7 @@ with gr.Blocks(
             frame_filter,
             image_gallery,
             session_state_output,
+            session_state_input,
             streaming,
             mode,
             conf_thres,
@@ -1141,6 +1147,7 @@ with gr.Blocks(
         image_gallery,
         log_output,
         session_state_output,
+        session_state_input,
         streaming,
         mode,
         conf_thres,