demo

app.py

@@ -39,7 +39,7 @@ import yaml
 from easydict import EasyDict as edict
 from einops import rearrange
 from PIL import Image
-from huggingface_hub import snapshot_download
+from huggingface_hub import snapshot_download, HfApi
 import spaces
 
 # Install diff-gaussian-rasterization at runtime (requires GPU)
@@ -59,6 +59,33 @@ def _log_viewer_file(ply_path: Path):
     }
     print("[VIEWER-RETURN]", json.dumps(info))
 
+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()
+        ply_filename = ply_path.name
+        
+        # Upload to tmp_ply folder
+        path_in_repo = f"tmp_ply/{ply_filename}"
+        
+        print(f"Uploading {ply_filename} to HuggingFace...")
+        api.upload_file(
+            path_or_fileobj=str(ply_path),
+            path_in_repo=path_in_repo,
+            repo_id=repo_id,
+            repo_type="model",
+        )
+        
+        # Return the public URL
+        hf_url = f"https://huggingface.co/{repo_id}/resolve/main/{path_in_repo}"
+        print(f"✓ Uploaded to: {hf_url}")
+        return hf_url
+        
+    except Exception as e:
+        print(f"⚠️  Failed to upload to HuggingFace: {e}")
+        # Fallback to local path
+        return f"/file={ply_path}"
+
 # -----------------------------
 # Ensure diff-gaussian-rasterization builds for current GPU
 # -----------------------------
@@ -338,26 +365,33 @@ class FaceLiftPipeline:
             # Log the viewer file for quick debugging
             _log_viewer_file(ply_path)
             
+            # Upload PLY to HuggingFace for public access
+            hf_ply_url = upload_ply_to_hf(ply_path)
+            
             # Final CUDA cache clear
             torch.cuda.empty_cache()
             
             # Create viewer HTML with iframe pointing to external viewer
-            ply_url = f"/file={ply_path}"
+            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;">
+            <div style="width:100%; height:600px; position:relative; border-radius:8px; overflow:hidden; border:1px solid #333; background:#000;">
                 <iframe 
-                    src="https://www.wlyu.me/FaceLift/splat/index.html?url={ply_url}" 
+                    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;">
-                <a href="https://www.wlyu.me/FaceLift/splat/index.html?url={ply_url}" 
+            <div style="text-align:center; margin-top:10px; padding:10px;">
+                <a href="{viewer_url}" 
                    target="_blank" 
-                   style="color:#4CAF50; text-decoration:none; font-size:13px;">
-                    🔗 Open viewer in new tab for better performance
+                   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>
             """