Update app.py

app.py

@@ -199,41 +199,76 @@ with gr.Blocks() as demo:
     raw_depth = gr.State()  # 🔑 保存模型输出
 
     with gr.Row():
-        # ========== Left ==========
-        with gr.Column(scale=1):
-            inp = gr.Image(type="numpy", label="Input Image", height=360)
-
-            gr.Markdown("### Examples (click to load)")
-            gr.Examples(examples=example_paths, inputs=inp)
-
-            gr.Markdown("### Examples from Gemini (click to load)")
-            gr.Examples(examples=example_gen_paths, inputs=inp)
-
-            btn_infer = gr.Button("Run Inference", variant="primary")
-            btn_100m = gr.Button("Visualize (100m)")
-            btn_10m = gr.Button("Visualize (10m)")
-
-            gr.Markdown(
-                """
-                <small>
-                <b>Visualization range:</b><br>
-                • <b>100m</b>: recommended for <b>outdoor</b> scenes<br>
-                • <b>10m</b>: recommended for <b>indoor</b> scenes<br>
-                (Only affects visualization, not the raw depth output)
-                </small>
-                """,
-                elem_id="vis_hint",
-            )
-
-        # ========== Right ==========
-        with gr.Column(scale=2):
-            with gr.Row():
-                out_color = gr.Image(label="Depth (Color)", height=260, scale=20)
-                colorbar_color = gr.Image(label="Scale", height=260, scale=1, show_label=False, min_width=100)
-            with gr.Row():
-                out_gray = gr.Image(label="Depth (Gray)", height=260, scale=20)
-                colorbar_gray = gr.Image(label="Scale", height=260, scale=1, show_label=False, min_width=100)
-            out_npy = gr.File(label="Depth (.npy)")
+
+    # ========== Left ==========
+    # 左侧列（Input Image）
+    with gr.Column(scale=10):
+        inp = gr.Image(
+            type="numpy",
+            label="Input Image",
+            height=360
+        )
+
+        gr.Markdown("### Examples (click to load)")
+        gr.Examples(examples=example_paths, inputs=inp)
+
+        gr.Markdown("### Examples from Gemini (click to load)")
+        gr.Examples(examples=example_gen_paths, inputs=inp)
+
+        btn_infer = gr.Button("Run Inference", variant="primary")
+        btn_100m = gr.Button("Visualize (100m)")
+        btn_10m = gr.Button("Visualize (10m)")
+
+        gr.Markdown(
+            """
+            <small>
+            <b>Visualization range:</b><br>
+            • <b>100m</b>: recommended for <b>outdoor</b> scenes<br>
+            • <b>10m</b>: recommended for <b>indoor</b> scenes<br>
+            (Only affects visualization, not the raw depth output)
+            </small>
+            """,
+            elem_id="vis_hint",
+        )
+
+    # ========== Right ==========
+    # 右侧整体（包含 中间列 + colorbar 列）
+    with gr.Column(scale=11):
+
+        # -------- Row 1: Color Depth --------
+        with gr.Row():
+            # 中间列（必须和左侧等宽）
+            with gr.Column(scale=10):
+                out_color = gr.Image(
+                    label="Depth (Color)",
+                    height=260
+                )
+
+            # colorbar 列（很窄）
+            with gr.Column(scale=1, min_width=80):
+                colorbar_color = gr.Image(
+                    label="Scale",
+                    height=260,
+                    show_label=False
+                )
+
+        # -------- Row 2: Gray Depth --------
+        with gr.Row():
+            with gr.Column(scale=10):
+                out_gray = gr.Image(
+                    label="Depth (Gray)",
+                    height=260
+                )
+
+            with gr.Column(scale=1, min_width=80):
+                colorbar_gray = gr.Image(
+                    label="Scale",
+                    height=260,
+                    show_label=False
+                )
+
+        out_npy = gr.File(label="Depth (.npy)")
+
 
     # 1️⃣ 跑模型
     btn_infer.click(