Update app.py

app.py

@@ -417,6 +417,37 @@ function swDownload() {
 """
 
 
+# Gradio serves any file under its allowed paths (which includes the
+# system temp dir — where our .ply files live) at this documented route:
+#   /gradio_api/file=<path>
+# See: https://gradio.app/guides/file-access
+# Building this URL ourselves in Python is deterministic — unlike reading
+# it back off a File component's FileData, whose `.url` field can still be
+# null at the moment a change event fires (it gets filled in later by the
+# frontend's own render pass), which is what left the viewer stuck idle.
+def make_load_trigger(ply_path: str, size_mb) -> str:
+    import urllib.parse
+    url = "/gradio_api/file=" + urllib.parse.quote(ply_path, safe="/")
+    url_js = url.replace("\\", "\\\\").replace("'", "\\'")
+    return f"""
+<script>
+(function() {{
+  var tries = 0;
+  (function tryLoad() {{
+    tries++;
+    if (typeof window.swLoadUrl === 'function') {{
+      window.swLoadUrl('{url_js}', '{size_mb}');
+    }} else if (tries < 50) {{
+      setTimeout(tryLoad, 100);
+    }} else if (window.swLog) {{
+      window.swLog('ERROR: viewer script never became ready — try reloading the page');
+    }}
+  }})();
+}})();
+</script>
+"""
+
+
 # ────────────────────────────────────────────────────────────────
 # UI — css in gr.Blocks() (correct for all Gradio versions)
 # ────────────────────────────────────────────────────────────────
@@ -474,12 +505,9 @@ with gr.Blocks(css=CSS, title="SplatWeb") as demo:
             # 3D viewer — always visible, waits for content
             gr.HTML(VIEWER_SHELL)
 
-            # Hidden File component — Gradio serves this over its normal
-            # file route and gives the frontend a real URL for it (via the
-            # .change(js=...) wiring below), so the browser can fetch() the
-            # raw bytes directly instead of us shipping a giant base64
-            # string through the page.
-            ply_file = gr.File(visible=False, elem_id="sw-ply-file", label="ply")
+            # Hidden trigger — a tiny script (just a URL string, never the
+            # file bytes) telling the viewer where to fetch the model from.
+            trigger1 = gr.HTML(value="", visible=False)
 
         # ── Tab 2: Two angles + animation ───────────────────
         with gr.TabItem("🎬 Two Angles → Animation"):
@@ -541,17 +569,18 @@ with gr.Blocks(css=CSS, title="SplatWeb") as demo:
     def handle_single(img):
         print(f"[handle_single] click received, img={img}", flush=True)
         if img is None:
-            return "⚠ Please upload a photo first.", gr.update(value=None, visible=False)
+            return "⚠ Please upload a photo first.", gr.update(value="", visible=False)
         try:
             ply, status = run_sharp(img)
         except Exception as e:
             print(f"[handle_single] GPU call raised: {e}", flush=True)
             return (f"⚠ GPU rejected the request: {str(e)}\n(If this mentions quota/duration, you've hit your daily free ZeroGPU limit — wait for it to reset, or sign in with a HF account for a bigger quota.)",
-                    gr.update(value=None, visible=False))
+                    gr.update(value="", visible=False))
         print(f"[handle_single] run_sharp returned ply={ply!r} status={status!r}", flush=True)
         if ply:
-            return status, gr.update(value=ply, visible=False)
-        return status, gr.update(value=None, visible=False)
+            size_mb = round(os.path.getsize(ply) / 1024 / 1024, 2)
+            return status, gr.update(value=make_load_trigger(ply, size_mb), visible=True)
+        return status, gr.update(value="", visible=False)
 
     def handle_dual(a, b, do_anim):
         if a is None or b is None:
@@ -590,34 +619,9 @@ with gr.Blocks(css=CSS, title="SplatWeb") as demo:
 
         return msg, html_out
 
-    btn1.click(fn=handle_single, inputs=[img1], outputs=[st1, ply_file])
+    btn1.click(fn=handle_single, inputs=[img1], outputs=[st1, trigger1])
     btn2.click(fn=handle_dual,   inputs=[imgA, imgB, anim_toggle], outputs=[st2, anim_file])
 
-    # This is the load-bearing wire: whenever ply_file's value changes
-    # (including programmatically, from handle_single's return), Gradio
-    # hands this JS function the component's real FileData — the same
-    # {path, url, size, ...} object it uses internally — no DOM-scraping,
-    # no guessing routes. fn=None means this runs client-side only, no
-    # extra server round-trip.
-    ply_file.change(
-        fn=None, inputs=[ply_file], outputs=[],
-        js="""
-        (file) => {
-          const log = (m) => { if (window.swLog) window.swLog(m); else console.log('[SplatWeb]', m); };
-          log('ply_file component changed: ' + JSON.stringify(file));
-          if (file && file.url) {
-            const mb = file.size ? (file.size / 1024 / 1024).toFixed(2) : '';
-            if (window.swLoadUrl) { window.swLoadUrl(file.url, mb); }
-            else { log('ERROR: viewer script not ready yet — reload the page and try again'); }
-          } else if (file) {
-            log('ERROR: file has no .url field (got: ' + Object.keys(file).join(',') + ') — Gradio version mismatch?');
-          } else {
-            log('cleared / no file');
-          }
-        }
-        """
-    )
-
     gr.HTML("""
     <div style="text-align:center;padding:1.5rem 1rem;
                 border-top:1px solid rgba(80,130,255,0.07);margin-top:1.5rem;">