demo

app.py

@@ -16,6 +16,9 @@
 """
 FaceLift: Single Image 3D Face Reconstruction
 Generates 3D head models from single images using multi-view diffusion and GS-LRM.
+
+Note: To enable the interactive 3D viewer, this Space needs write access to wlyu/FaceLift_demo.
+Set the HF_TOKEN environment variable in Space settings with a token that has write access.
 """
 
 # Disable HF fast transfer if hf_transfer is not installed
@@ -62,7 +65,14 @@ def _log_viewer_file(ply_path: Path):
 def upload_ply_to_hf(ply_path: Path, repo_id: str = "wlyu/FaceLift_demo") -> str:
     """Upload PLY file to HuggingFace and return the public URL."""
     try:
-        api = HfApi()
+        # Get HF token from environment (automatically available in HF Spaces)
+        hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
+        
+        if not hf_token:
+            print("⚠️  No HF_TOKEN found in environment, skipping upload")
+            return None
+        
+        api = HfApi(token=hf_token)
         ply_filename = ply_path.name
         
         # Upload to tmp_ply folder
@@ -74,6 +84,7 @@ def upload_ply_to_hf(ply_path: Path, repo_id: str = "wlyu/FaceLift_demo") -> str
             path_in_repo=path_in_repo,
             repo_id=repo_id,
             repo_type="model",
+            token=hf_token,
         )
         
         # Return the public URL
@@ -83,8 +94,8 @@ def upload_ply_to_hf(ply_path: Path, repo_id: str = "wlyu/FaceLift_demo") -> str
         
     except Exception as e:
         print(f"⚠️  Failed to upload to HuggingFace: {e}")
-        # Fallback to local path
-        return f"/file={ply_path}"
+        print("    Make sure the Space has write access to the repository")
+        return None
 
 # -----------------------------
 # Ensure diff-gaussian-rasterization builds for current GPU
@@ -371,29 +382,53 @@ class FaceLiftPipeline:
             # Final CUDA cache clear
             torch.cuda.empty_cache()
             
-            # Create viewer HTML with iframe pointing to external viewer
-            viewer_url = f"https://www.wlyu.me/FaceLift/splat/index.html?url={hf_ply_url}"
-            
-            viewer_html = f"""
-            <div style="width:100%; height:600px; position:relative; border-radius:8px; overflow:hidden; border:1px solid #333; background:#000;">
-                <iframe 
-                    src="{viewer_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; padding:10px;">
-                <a href="{viewer_url}" 
-                   target="_blank" 
-                   style="display:inline-block; color:#fff; background:#4CAF50; padding:10px 20px; text-decoration:none; font-size:14px; border-radius:6px; font-weight:500;">
-                    🎮 Open Interactive Viewer in New Tab
-                </a>
-                <p style="color:#666; font-size:12px; margin-top:8px;">
-                    Drag to rotate • Scroll to zoom • Right-click to pan
-                </p>
-            </div>
-            """
+            # Create viewer HTML
+            if hf_ply_url:
+                # Successfully uploaded - show iframe viewer
+                viewer_url = f"https://www.wlyu.me/FaceLift/splat/index.html?url={hf_ply_url}"
+                
+                viewer_html = f"""
+                <div style="width:100%; height:600px; position:relative; border-radius:8px; overflow:hidden; border:1px solid #333; background:#000;">
+                    <iframe 
+                        src="{viewer_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; padding:10px;">
+                    <a href="{viewer_url}" 
+                       target="_blank" 
+                       style="display:inline-block; color:#fff; background:#4CAF50; padding:10px 20px; text-decoration:none; font-size:14px; border-radius:6px; font-weight:500;">
+                        🎮 Open Interactive Viewer in New Tab
+                    </a>
+                    <p style="color:#666; font-size:12px; margin-top:8px;">
+                        Drag to rotate • Scroll to zoom • Right-click to pan
+                    </p>
+                </div>
+                """
+            else:
+                # Upload failed - provide instructions to use viewer manually
+                viewer_base_url = "https://www.wlyu.me/FaceLift/splat/index.html"
+                
+                viewer_html = f"""
+                <div style="padding:40px; text-align:center; background:#f5f5f5; border-radius:8px; border:1px solid #ddd;">
+                    <div style="font-size:48px; margin-bottom:20px;">🎮</div>
+                    <h3 style="margin:0 0 15px 0; color:#333;">Interactive 3D Viewer</h3>
+                    <p style="color:#666; margin-bottom:25px; line-height:1.6;">
+                        Download the PLY file below, then drag and drop it into the viewer<br>
+                        or use the viewer with a public URL
+                    </p>
+                    <a href="{viewer_base_url}" 
+                       target="_blank" 
+                       style="display:inline-block; color:#fff; background:#4CAF50; padding:12px 24px; text-decoration:none; font-size:15px; border-radius:6px; font-weight:500; margin-bottom:15px;">
+                        🔗 Open Interactive Viewer
+                    </a>
+                    <p style="color:#888; font-size:13px; margin-top:15px;">
+                        <strong>Controls:</strong> Drag to rotate • Scroll to zoom • Right-click to pan
+                    </p>
+                </div>
+                """
             
             return (
                 str(output_path),              # Reconstruction grid