demo

app.py

@@ -46,10 +46,44 @@ import spaces
 import subprocess
 import sys
 
-# Outputs directory for generated files
+# Outputs directory for generated files (ensures folder exists even if CWD differs)
 OUTPUTS_DIR = Path.cwd() / "outputs"
 OUTPUTS_DIR.mkdir(exist_ok=True)
 
+def _as_str_path(p):
+    """Coerce any Path-like to a plain string for Gradio components."""
+    from pathlib import Path as _Path
+    return str(p) if isinstance(p, (str, _Path)) else ""
+
+def _log_viewer_file(ply_path: Path):
+    """Print a concise JSON line about the viewer file so users can debug from Space logs."""
+    info = {
+        "ply_path": str(Path(ply_path).absolute()),
+        "exists": Path(ply_path).exists(),
+        "size_bytes": (Path(ply_path).stat().st_size if Path(ply_path).exists() else None)
+    }
+    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
 # -----------------------------
@@ -135,7 +169,7 @@ class FaceLiftPipeline:
         self.camera_indices = [2, 1, 0, 5, 4, 3]
         
         # Load models (keep on CPU for ZeroGPU compatibility)
-        print("Loading models...")
+        print("Loading models... (gradio", getattr(gr, "__version__", "unknown"), ")")
         try:
             self.mvdiffusion_pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
                 str(workspace_dir / "checkpoints/mvdiffusion/pipeckpts"),
@@ -326,10 +360,19 @@ class FaceLiftPipeline:
             turntable_path = output_dir / "turntable.mp4"
             imageseq2video(turntable_frames, str(turntable_path), fps=30)
             
+            # Log the viewer file for quick debugging
+            _log_viewer_file(ply_path)
+            
             # Final CUDA cache clear
             torch.cuda.empty_cache()
             
-            return str(output_path), str(turntable_path), str(ply_path)
+            # Return a Model3D.update for the viewer and a string path for the File component
+            return (
+                str(output_path),              # Reconstruction grid
+                str(turntable_path),           # Turntable video
+                gr.Model3D.update(value=_as_str_path(ply_path)),  # Viewer
+                _as_str_path(ply_path),        # Download file
+            )
             
         except Exception as e:
             import traceback
@@ -349,13 +392,6 @@ def main():
 
     with gr.Blocks(title="FaceLift: Single Image 3D Face Reconstruction") as demo:
 
-        # Wrapper to return outputs for display
-        def _generate_and_filter_outputs(image_path, auto_crop, guidance_scale, random_seed, num_steps):
-            output_path, turntable_path, ply_path = \
-                pipeline.generate_3d_head(image_path, auto_crop, guidance_scale, random_seed, num_steps)
-            
-            return output_path, turntable_path, str(ply_path), ply_path
-
         gr.Markdown("## FaceLift: Single Image 3D Face Reconstruction.")
         
         gr.Markdown("""
@@ -382,12 +418,33 @@ 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)
-                # Interactive 3D viewer for the generated Gaussian PLY (uses three.js under the hood)
+                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_ply   = gr.File(label="Download 3D Model (.ply)")
+                out_ply    = gr.File(label="Download 3D Model (.ply)")
+
+        # Wrapper: call the pipeline and forward outputs in the exact order expected
+        @spaces.GPU(duration=1)  # trivial; only formats outputs
+        def _generate_and_filter_outputs(image_path, auto_crop, guidance_scale, random_seed, num_steps):
+            (
+                output_path,
+                turntable_path,
+                viewer_update,
+                ply_dl_path,
+            ) = pipeline.generate_3d_head(image_path, auto_crop, guidance_scale, random_seed, num_steps)
+            # Ensure correct types for each component
+            if not isinstance(viewer_update, gr.components.Model3D.Update):
+                viewer_update = gr.Model3D.update(value=_as_str_path(viewer_update))
+            return (
+                _as_str_path(output_path),
+                _as_str_path(turntable_path),
+                viewer_update,
+                _as_str_path(ply_dl_path),
+            )
 
         # Run generation and display all outputs
         run_btn.click(
@@ -396,6 +453,14 @@ 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 gr.Model3D.update(value=_as_str_path(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)