demo

app.py

@@ -59,26 +59,6 @@ def _log_viewer_file(ply_path: Path):
     }
     print("[VIEWER-RETURN]", json.dumps(info))
 
-def _write_sample_cube_ply(dst):
-    """Write a tiny ASCII PLY point-cloud cube; handy to sanity-check the Model3D wiring."""
-    import random
-    dst = Path(dst)
-    dst.parent.mkdir(parents=True, exist_ok=True)
-    pts = []
-    for _ in range(2000):
-        x, y, z = (random.uniform(-0.5, 0.5) for _ in range(3))
-        r, g, b = (random.randint(80, 255) for _ in range(3))
-        pts.append((x, y, z, r, g, b))
-    with open(dst, "w") as f:
-        f.write("ply\nformat ascii 1.0\n")
-        f.write(f"element vertex {len(pts)}\n")
-        f.write("property float x\nproperty float y\nproperty float z\n")
-        f.write("property uchar red\nproperty uchar green\nproperty uchar blue\n")
-        f.write("end_header\n")
-        for x, y, z, r, g, b in pts:
-            f.write(f"{x} {y} {z} {r} {g} {b}\n")
-    return str(dst)
-
 # -----------------------------
 # Ensure diff-gaussian-rasterization builds for current GPU
 # -----------------------------
@@ -361,11 +341,30 @@ class FaceLiftPipeline:
             # Final CUDA cache clear
             torch.cuda.empty_cache()
             
-            # Return paths directly (Gradio components handle them automatically)
+            # Create viewer HTML with iframe pointing to external viewer
+            ply_url = f"/file={ply_path}"
+            viewer_html = f"""
+            <div style="width:100%; height:600px; position:relative; border-radius:8px; overflow:hidden; border:1px solid #333;">
+                <iframe 
+                    src="https://www.wlyu.me/FaceLift/splat/index.html?url={ply_url}" 
+                    style="width:100%; height:100%; border:none;"
+                    allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture"
+                    allowfullscreen>
+                </iframe>
+            </div>
+            <div style="text-align:center; margin-top:10px;">
+                <a href="https://www.wlyu.me/FaceLift/splat/index.html?url={ply_url}" 
+                   target="_blank" 
+                   style="color:#4CAF50; text-decoration:none; font-size:13px;">
+                    🔗 Open viewer in new tab for better performance
+                </a>
+            </div>
+            """
+            
             return (
                 str(output_path),              # Reconstruction grid
                 str(turntable_path),           # Turntable video
-                str(ply_path),                 # Viewer
+                viewer_html,                   # Viewer HTML
                 str(ply_path),                 # Download file
             )
             
@@ -413,13 +412,10 @@ def main():
                         examples_per_page=10,
                     )
 
-                # Optional: a quick sanity button to verify the viewer path wiring
-                sample_btn = gr.Button("🔧 Load Sample Cube (sanity test)")
-
             with gr.Column(scale=1):
                 out_recon  = gr.Image(label="3D Reconstruction Views")
                 out_video  = gr.PlayableVideo(label="Turntable Animation (360° View)", height=600)
-                out_viewer = gr.Model3D(label="Interactive 3D Viewer (.ply)", height=600)
+                out_viewer = gr.HTML(label="🎮 Interactive 3D Viewer")
                 out_ply    = gr.File(label="Download 3D Model (.ply)")
 
         # Wrapper: call the pipeline and forward outputs in the exact order expected
@@ -433,14 +429,6 @@ def main():
             outputs=[out_recon, out_video, out_viewer, out_ply],
         )
 
-        # Wire the sample cube sanity test
-        def _load_sample_viewer():
-            sample_path = _write_sample_cube_ply(Path("outputs") / "sample_cube.ply")
-            print("[VIEWER-SAMPLE]", os.path.abspath(sample_path))
-            return sample_path
-
-        sample_btn.click(_load_sample_viewer, inputs=[], outputs=[out_viewer])
-
         demo.queue(max_size=10)
         demo.launch(share=True, server_name="0.0.0.0", server_port=7860, show_error=True)