Spaces:

PLUS-WAVE
/

InfiniSplat

Running on Zero

akhaliq HF Staff commited on 4 days ago

Commit

769d659

1 Parent(s): 85f764f

Fix Invalid URL crash in runPipeline and surface file sizes

new URL(artifactUrl) threw because the artifact URL is a relative path
("/gradio_api/file=..."). Drop the URL parse; instead return the file
size from each @app .api() endpoint alongside the URL, so the UI can
show real byte counts in the Scene tree and Telemetry grid.

Files changed (1) hide show

app.py +18 -12

app.py CHANGED Viewed

@@ -88,7 +88,7 @@ def reconstruct(image_path: FileData) -> dict:
         seconds=round(time.perf_counter() - started, 3),
         bytes=artifact.stat().st_size,
     )
-    return {"url": _file_url(artifact)}
 @app.api(name="export_ply", queue=True, concurrency_limit=2)
@@ -107,7 +107,7 @@ def export_ply(artifact_url: str) -> dict:
         seconds=round(time.perf_counter() - started, 3),
         bytes=scene_ply.stat().st_size,
     )
-    return {"url": _file_url(scene_ply)}
 @app.api(name="export_viewer", queue=True, concurrency_limit=2)
@@ -126,7 +126,7 @@ def export_viewer(scene_ply_url: str) -> dict:
         sog_bytes=exported.scene_sog.stat().st_size,
         viewer_html_bytes=exported.viewer_html.stat().st_size,
     )
-    return {"url": _file_url(exported.viewer_html)}
 @app.api(name="export_html", queue=True, concurrency_limit=2)
@@ -144,7 +144,7 @@ def export_html(viewer_html_url: str) -> dict:
         seconds=round(time.perf_counter() - started, 3),
         bytes=standalone.stat().st_size,
     )
-    return {"url": _file_url(standalone)}
 @app.api(name="viewer_html", queue=False)
@@ -1293,11 +1293,12 @@ INDEX_HTML = r"""<!doctype html>
           client.predict("/reconstruct", { image_path: handle_file(pendingFile) })
         );
         const artifactUrl = reconWrap.result.data[0].url;
         stageTimings.gpu = reconWrap.ms;
         setStage("reconstruct", "done", fmtSec(reconWrap.ms / 1000));
         document.getElementById("telGpu").textContent = fmtSec(reconWrap.ms / 1000);
-        document.getElementById("telArtifact").textContent = fmtBytes(parseInt(new URL(artifactUrl).searchParams.get("size") || "0", 10)) || "ready";
-        logLine("reconstruct", "ok", `${fmtSec(reconWrap.ms / 1000)} · artifact ready`);
         // Stage 2: PLY export
         setViewerState("loading", { title: "Preparing PLY", detail: "Filtering Gaussians", progress: 45, stageLabel: "STAGE 2 / 4 · CPU" });
@@ -1307,9 +1308,11 @@ INDEX_HTML = r"""<!doctype html>
           client.predict("/export_ply", { artifact_url: artifactUrl })
         );
         const plyUrl = plyWrap.result.data[0].url;
-        setStage("ply", "done", fmtSec(plyWrap.ms / 1000));
         setDownload(dlPly, plyUrl, "Download PLY");
-        logLine("ply", "ok", `${fmtSec(plyWrap.ms / 1000)} · ${plyUrl.split("/").pop()}`);
         // Stage 3: Viewer
         setViewerState("loading", { title: "Encoding viewer", detail: "Building WebGL scene", progress: 70, stageLabel: "STAGE 3 / 4 · ENCODE" });
@@ -1319,14 +1322,16 @@ INDEX_HTML = r"""<!doctype html>
           client.predict("/export_viewer", { scene_ply_url: plyUrl })
         );
         const viewerUrl = viewWrap.result.data[0].url + "?v=" + Date.now();
-        setStage("viewer", "done", fmtSec(viewWrap.ms / 1000));
         viewerFrame.src = viewerUrl;
         viewerFrame.onload = () => {
           viewerFrame.classList.add("ready");
           setViewerState("ready");
           viewportMeta.textContent = "ready";
         };
-        logLine("viewer", "ok", `${fmtSec(viewWrap.ms / 1000)} · WebGL scene ready`);
         // Stage 4: HTML
         setViewerState("loading", { title: "Bundling HTML", detail: "Embedding assets for download", progress: 92, stageLabel: "STAGE 4 / 4 · BUNDLE" });
@@ -1336,12 +1341,13 @@ INDEX_HTML = r"""<!doctype html>
           client.predict("/export_html", { viewer_html_url: viewerUrl.replace(/[?#].*/, "") })
         );
         const htmlUrl = htmlWrap.result.data[0].url;
-        setStage("html", "done", fmtSec(htmlWrap.ms / 1000));
         setDownload(dlHtml, htmlUrl, "Download HTML");
         const totalSec = (performance.now() - pipelineStart) / 1000;
         document.getElementById("telTotal").textContent = fmtSec(totalSec);
-        logLine("html", "ok", `${fmtSec(htmlWrap.ms / 1000)} · standalone ready`);
         setViewerState("ready");
         setStatus("idle", { gpu: "idle", queue: 0, stage: "4 / 4" });

         seconds=round(time.perf_counter() - started, 3),
         bytes=artifact.stat().st_size,
     )
+    return {"url": _file_url(artifact), "size": artifact.stat().st_size}
 @app.api(name="export_ply", queue=True, concurrency_limit=2)
         seconds=round(time.perf_counter() - started, 3),
         bytes=scene_ply.stat().st_size,
     )
+    return {"url": _file_url(scene_ply), "size": scene_ply.stat().st_size}
 @app.api(name="export_viewer", queue=True, concurrency_limit=2)
         sog_bytes=exported.scene_sog.stat().st_size,
         viewer_html_bytes=exported.viewer_html.stat().st_size,
     )
+    return {"url": _file_url(exported.viewer_html), "size": exported.viewer_html.stat().st_size}
 @app.api(name="export_html", queue=True, concurrency_limit=2)
         seconds=round(time.perf_counter() - started, 3),
         bytes=standalone.stat().st_size,
     )
+    return {"url": _file_url(standalone), "size": standalone.stat().st_size}
 @app.api(name="viewer_html", queue=False)
           client.predict("/reconstruct", { image_path: handle_file(pendingFile) })
         );
         const artifactUrl = reconWrap.result.data[0].url;
+        const artifactSize = reconWrap.result.data[0].size;
         stageTimings.gpu = reconWrap.ms;
         setStage("reconstruct", "done", fmtSec(reconWrap.ms / 1000));
         document.getElementById("telGpu").textContent = fmtSec(reconWrap.ms / 1000);
+        document.getElementById("telArtifact").textContent = fmtBytes(artifactSize);
+        logLine("reconstruct", "ok", `${fmtSec(reconWrap.ms / 1000)} · ${fmtBytes(artifactSize)}`);
         // Stage 2: PLY export
         setViewerState("loading", { title: "Preparing PLY", detail: "Filtering Gaussians", progress: 45, stageLabel: "STAGE 2 / 4 · CPU" });
           client.predict("/export_ply", { artifact_url: artifactUrl })
         );
         const plyUrl = plyWrap.result.data[0].url;
+        const plySize = plyWrap.result.data[0].size;
+        setStage("ply", "done", fmtBytes(plySize));
         setDownload(dlPly, plyUrl, "Download PLY");
+        document.getElementById("telPly").textContent = fmtBytes(plySize);
+        logLine("ply", "ok", `${fmtSec(plyWrap.ms / 1000)} · ${fmtBytes(plySize)}`);
         // Stage 3: Viewer
         setViewerState("loading", { title: "Encoding viewer", detail: "Building WebGL scene", progress: 70, stageLabel: "STAGE 3 / 4 · ENCODE" });
           client.predict("/export_viewer", { scene_ply_url: plyUrl })
         );
         const viewerUrl = viewWrap.result.data[0].url + "?v=" + Date.now();
+        const viewerSize = viewWrap.result.data[0].size;
+        setStage("viewer", "done", fmtBytes(viewerSize));
+        document.getElementById("telViewer").textContent = fmtBytes(viewerSize);
         viewerFrame.src = viewerUrl;
         viewerFrame.onload = () => {
           viewerFrame.classList.add("ready");
           setViewerState("ready");
           viewportMeta.textContent = "ready";
         };
+        logLine("viewer", "ok", `${fmtSec(viewWrap.ms / 1000)} · ${fmtBytes(viewerSize)}`);
         // Stage 4: HTML
         setViewerState("loading", { title: "Bundling HTML", detail: "Embedding assets for download", progress: 92, stageLabel: "STAGE 4 / 4 · BUNDLE" });
           client.predict("/export_html", { viewer_html_url: viewerUrl.replace(/[?#].*/, "") })
         );
         const htmlUrl = htmlWrap.result.data[0].url;
+        const htmlSize = htmlWrap.result.data[0].size;
+        setStage("html", "done", fmtBytes(htmlSize));
         setDownload(dlHtml, htmlUrl, "Download HTML");
         const totalSec = (performance.now() - pipelineStart) / 1000;
         document.getElementById("telTotal").textContent = fmtSec(totalSec);
+        logLine("html", "ok", `${fmtSec(htmlWrap.ms / 1000)} · ${fmtBytes(htmlSize)}`);
         setViewerState("ready");
         setStatus("idle", { gpu: "idle", queue: 0, stage: "4 / 4" });