Upload app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
+from gradio_client import Client, file
 from content import TAB_ABOUT
 from arch_content import TAB_ARCH
 from bench_content import TAB_BENCH
 from usage_content import TAB_USAGE
+from infer_content import INFERENCE_TAB
 
 CSS = """
 #container{max-width:1100px;margin:0 auto}
@@ -13,11 +15,30 @@ CSS = """
 .stat{text-align:center}
 .stat-value{font-size:2em;font-weight:700;color:#4f46e5}
 .stat-label{font-size:.85em;color:#6b7280}
-.video-wrap{position:relative;padding-bottom:56.25%;height:0;overflow:hidden;border-radius:12px}
-.video-wrap iframe{position:absolute;top:0;left:0;width:100%;height:100%;border:0;border-radius:12px}
 footer{display:none!important}
 """
 
+def run_inference(video_path, fps, max_frames, num_scale, keyframe_int, conf_pct):
+    if not video_path:
+        return None, None, "❌ Please upload a video first."
+    try:
+        client = Client("dennny123/lingbot-3d")
+        result = client.predict(
+            video_file=file(video_path),
+            fps=int(fps),
+            max_frames=int(max_frames),
+            num_scale_frames=int(num_scale),
+            keyframe_interval=int(keyframe_int),
+            conf_percentile=float(conf_pct),
+            api_name="/reconstruct_scene"
+        )
+        glb_path = result[1]
+        img_path = result[2]
+        status = result[5]
+        return glb_path, img_path, f"✅ {status}"
+    except Exception as e:
+        return None, None, f"❌ Error: {str(e)}"
+
 with gr.Blocks(title="LingBot-Map Explorer", css=CSS) as demo:
     with gr.Column(elem_id="container"):
         gr.Markdown(
@@ -33,6 +54,21 @@ with gr.Blocks(title="LingBot-Map Explorer", css=CSS) as demo:
             '</div>'
         )
         with gr.Tabs():
+            with gr.Tab("🔮 Run Inference"):
+                gr.Markdown(INFERENCE_TAB)
+                vid = gr.Video(label="Input video", sources=["upload"])
+                with gr.Row():
+                    fps = gr.Slider(1, 12, value=8, step=1, label="Sampling FPS")
+                    max_fr = gr.Slider(2, 24, value=24, step=1, label="Max frames")
+                with gr.Row():
+                    num_sc = gr.Slider(1, 8, value=4, step=1, label="Scale frames")
+                    kf_int = gr.Slider(1, 8, value=2, step=1, label="Keyframe interval")
+                conf = gr.Slider(0, 90, value=50, step=5, label="Confidence percentile")
+                btn = gr.Button("Build 3D Scene", variant="primary")
+                status = gr.Markdown()
+                with gr.Row():
+                    glb_out = gr.Model3D(label="3D Result", display_mode="point_cloud")
+                    img_out = gr.Image(label="Frame preview")
             with gr.Tab("📋 About"):
                 gr.Markdown(TAB_ABOUT)
             with gr.Tab("🧠 Architecture"):
@@ -42,4 +78,10 @@ with gr.Blocks(title="LingBot-Map Explorer", css=CSS) as demo:
             with gr.Tab("🚀 Usage"):
                 gr.Markdown(TAB_USAGE)
 
+    btn.click(
+        run_inference,
+        [vid, fps, max_fr, num_sc, kf_int, conf],
+        [glb_out, img_out, status]
+    )
+
 demo.launch()