update app

app.py

@@ -136,10 +136,6 @@ model = None
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
 
-
-# -------------------------------------------------------------------------
-# Rerun visualization helper
-# -------------------------------------------------------------------------
 def predictions_to_rrd(predictions, glbfile, target_dir, frame_filter="All", show_cam=True):
     run_id = str(uuid.uuid4())
     timestamp = datetime.now().strftime("%Y-%m-%dT%H%M%S")
@@ -239,9 +235,6 @@ def predictions_to_rrd(predictions, glbfile, target_dir, frame_filter="All", sho
     return rrd_path
 
 
-# -------------------------------------------------------------------------
-# 1) Core model inference
-# -------------------------------------------------------------------------
 @spaces.GPU(duration=120)
 def run_model(target_dir, apply_mask=True, mask_edges=True, filter_black_bg=False, filter_white_bg=False):
     global model
@@ -387,10 +380,6 @@ def populate_visualization_tabs(processed_data):
         return None, None, None, []
     return update_depth_view(processed_data, 0), update_normal_view(processed_data, 0), update_measure_view(processed_data, 0)[0], []
 
-
-# -------------------------------------------------------------------------
-# 2) Handle uploaded video/images
-# -------------------------------------------------------------------------
 def handle_uploads(unified_upload, s_time_interval=1.0):
     start_time = time.time()
     gc.collect()
@@ -453,9 +442,6 @@ def handle_uploads(unified_upload, s_time_interval=1.0):
     return target_dir, image_paths
 
 
-# -------------------------------------------------------------------------
-# 3) Reconstruction
-# -------------------------------------------------------------------------
 @spaces.GPU(duration=120)
 def gradio_demo(target_dir, frame_filter="All", show_cam=True, filter_black_bg=False, filter_white_bg=False, apply_mask=True, show_mesh=True):
     if not os.path.isdir(target_dir) or target_dir == "None":
@@ -506,9 +492,6 @@ def gradio_demo(target_dir, frame_filter="All", show_cam=True, filter_black_bg=F
     )
 
 
-# -------------------------------------------------------------------------
-# 4) Helper / visualization functions
-# -------------------------------------------------------------------------
 def colorize_depth(depth_map, mask=None):
     if depth_map is None:
         return None
@@ -672,9 +655,6 @@ def update_all_views_on_filter_change(target_dir, filter_black_bg, filter_white_
         return processed_data, None, None, None, []
 
 
-# -------------------------------------------------------------------------
-# Example scene functions
-# -------------------------------------------------------------------------
 def get_scene_info(examples_dir):
     import glob
     scenes = []
@@ -702,9 +682,6 @@ def load_example_scene(scene_name, examples_dir="examples"):
     return None, target_dir, image_paths, f"Loaded '{scene_name}' — {selected_scene['num_images']} images. Click Reconstruct."
 
 
-# -------------------------------------------------------------------------
-# CSS
-# -------------------------------------------------------------------------
 CUSTOM_CSS = (GRADIO_CSS or "") + """
 /* ── Page shell ── */
 #app-shell {
@@ -831,28 +808,21 @@ CUSTOM_CSS = (GRADIO_CSS or "") + """
 
 """
 
-# -------------------------------------------------------------------------
-# 6) Build Gradio UI
-# -------------------------------------------------------------------------
 with gr.Blocks() as demo:
 
-    # Hidden state
     is_example            = gr.Textbox(visible=False, value="None")
     num_images            = gr.Textbox(visible=False, value="None")
     processed_data_state  = gr.State(value=None)
     measure_points_state  = gr.State(value=[])
     target_dir_output     = gr.Textbox(visible=False, value="None")
 
-    # ── Header ──────────────────────────────────────────────────────────
     with gr.Column(elem_id="app-shell"):
         with gr.Column(elem_id="app-header"):
             gr.Markdown("# **Map-Anything-v1**", elem_id="main-title")
             gr.Markdown("Metric 3D Reconstruction (Point Cloud and Camera Poses)")
 
-        # ── Main two-column layout ───────────────────────────────────────
         with gr.Row(equal_height=False):
 
-            # ── LEFT PANEL ─────────────────────────────────────────────
             with gr.Column(elem_id="left-panel", scale=0):
 
                 unified_upload = gr.File(
@@ -890,7 +860,6 @@ with gr.Blocks() as demo:
 
                 submit_btn = gr.Button("Reconstruct", variant="primary", elem_id="btn-reconstruct")
 
-                # Options accordion
                 with gr.Accordion("Options", open=False, elem_id="options-panel"):
                     gr.Markdown("### Point Cloud")
                     show_cam      = gr.Checkbox(label="Show cameras", value=True)
@@ -906,26 +875,21 @@ with gr.Blocks() as demo:
                     elem_id="frame-filter", show_label=True,
                 )
 
-            # ── RIGHT PANEL ────────────────────────────────────────────
             with gr.Column(elem_id="right-panel", scale=1):
 
-                # Status log
                 log_output = gr.Markdown(
                     "Upload a video or images, then click **Reconstruct**.",
                     elem_id="log-strip",
                 )
 
-                # Viewer tabs
                 with gr.Tabs(elem_id="viewer-tabs"):
 
-                    # 3-D View
                     with gr.Tab("3D View"):
                         reconstruction_output = Rerun(
                             label="Rerun 3D Viewer",
-                            height=630,
+                            height=680,
                         )
 
-                    # Depth
                     with gr.Tab("Depth"):
                         with gr.Row(elem_classes=["nav-row"]):
                             prev_depth_btn = gr.Button("◀ Prev", size="sm", scale=1)
@@ -940,7 +904,6 @@ with gr.Blocks() as demo:
                             format="png", interactive=False,
                         )
 
-                    # Normal
                     with gr.Tab("Normal"):
                         with gr.Row(elem_classes=["nav-row"]):
                             prev_normal_btn = gr.Button("◀ Prev", size="sm", scale=1)
@@ -955,7 +918,6 @@ with gr.Blocks() as demo:
                             format="png", interactive=False,
                         )
 
-                    # Measure
                     with gr.Tab("Measure"):
                         gr.Markdown(MEASURE_INSTRUCTIONS_HTML)
                         with gr.Row(elem_classes=["nav-row"]):
@@ -976,7 +938,6 @@ with gr.Blocks() as demo:
                         )
                         measure_text = gr.Markdown("")
 
-        # ── Examples ────────────────────────────────────────────────────
         with gr.Column(elem_id="examples-section"):
             gr.Markdown("## Example Scenes")
             gr.Markdown("Click a thumbnail to load the scene, then press **Reconstruct**.")
@@ -1009,11 +970,6 @@ with gr.Blocks() as demo:
                                 with gr.Column(scale=1, min_width=140):
                                     pass
 
-    # =========================================================================
-    # Event wiring
-    # =========================================================================
-
-    # Reconstruct button
     submit_btn.click(
         fn=clear_fields, inputs=[], outputs=[reconstruction_output]
     ).then(
@@ -1028,7 +984,6 @@ with gr.Blocks() as demo:
         ],
     ).then(fn=lambda: "False", inputs=[], outputs=[is_example])
 
-    # Live visualization option updates
     for trigger_inputs, trigger in [
         ([target_dir_output, frame_filter, show_cam, is_example, filter_black_bg, filter_white_bg, show_mesh], frame_filter.change),
         ([target_dir_output, frame_filter, show_cam, is_example], show_cam.change),
@@ -1055,7 +1010,6 @@ with gr.Blocks() as demo:
         [processed_data_state, depth_map, normal_map, measure_image, measure_points_state],
     )
 
-    # Upload handling
     def update_gallery_on_unified_upload(files, interval):
         if not files:
             return None, None, None
@@ -1093,14 +1047,12 @@ with gr.Blocks() as demo:
         outputs=[target_dir_output, image_gallery, log_output, resample_btn],
     )
 
-    # Measure tab
     measure_image.select(
         fn=measure,
         inputs=[processed_data_state, measure_points_state, measure_view_selector],
         outputs=[measure_image, measure_points_state, measure_text],
     )
 
-    # Depth tab navigation
     prev_depth_btn.click(
         fn=lambda pd, sel: navigate_depth_view(pd, sel, -1),
         inputs=[processed_data_state, depth_view_selector], outputs=[depth_view_selector, depth_map],
@@ -1114,7 +1066,6 @@ with gr.Blocks() as demo:
         inputs=[processed_data_state, depth_view_selector], outputs=[depth_map],
     )
 
-    # Normal tab navigation
     prev_normal_btn.click(
         fn=lambda pd, sel: navigate_normal_view(pd, sel, -1),
         inputs=[processed_data_state, normal_view_selector], outputs=[normal_view_selector, normal_map],
@@ -1128,7 +1079,6 @@ with gr.Blocks() as demo:
         inputs=[processed_data_state, normal_view_selector], outputs=[normal_map],
     )
 
-    # Measure tab navigation
     prev_measure_btn.click(
         fn=lambda pd, sel: navigate_measure_view(pd, sel, -1),
         inputs=[processed_data_state, measure_view_selector],