Update demo

app.py

@@ -376,24 +376,65 @@ GSPLAT_HEAD = """
   function ensureViewer() {
     if (renderer) return;
     const container = document.getElementById("splat-container");
-    renderer  = new SPLAT.WebGLRenderer();
+    if (!container) {
+      console.error("Container not found!");
+      return;
+    }
+    
+    renderer = new SPLAT.WebGLRenderer();
+    renderer.canvas.style.width = '100%';
+    renderer.canvas.style.height = '100%';
     container.appendChild(renderer.canvas);
-    scene   = new SPLAT.Scene();
-    camera  = new SPLAT.Camera();
+    
+    scene = new SPLAT.Scene();
+    camera = new SPLAT.Camera();
+    camera.data.setPosition(0, 0, 2);
+    
     controls = new SPLAT.OrbitControls(camera, renderer.canvas);
-    const loop = () => { controls.update(); renderer.render(scene, camera); requestAnimationFrame(loop); };
+    
+    const loop = () => { 
+      controls.update(); 
+      renderer.render(scene, camera); 
+      requestAnimationFrame(loop); 
+    };
     requestAnimationFrame(loop);
+    console.log("Viewer initialized");
   }
 
   async function loadSplat(url) {
+    console.log("Loading splat from:", url);
     ensureViewer();
+    
+    if (!scene) {
+      console.error("Scene not initialized!");
+      return;
+    }
+    
     // clear previous
     scene.children.length = 0;
-    await SPLAT.Loader.LoadAsync(url, scene, ()=>{});
+    
+    try {
+      await SPLAT.Loader.LoadAsync(url, scene, (progress) => {
+        console.log("Loading progress:", progress);
+      });
+      console.log("Splat loaded successfully!");
+      
+      // Reset camera position after loading
+      camera.data.setPosition(0, 0, 2);
+    } catch (error) {
+      console.error("Error loading splat:", error);
+    }
   }
 
   // Expose callable function for Gradio
   window.__load_splat__ = loadSplat;
+  
+  // Initialize viewer when DOM is ready
+  if (document.readyState === 'loading') {
+    document.addEventListener('DOMContentLoaded', ensureViewer);
+  } else {
+    setTimeout(ensureViewer, 100);
+  }
 </script>
 """
 
@@ -416,10 +457,11 @@ def main():
             return new_session_id()
 
         # After generation: copy ply into per-session folder and return viewer URL
-        def _prep_viewer_url(ply_path: str, session_id: str) -> str:
+        def _prep_viewer_url(ply_path: str, session_id: str):
             if not ply_path or not os.path.exists(ply_path):
-                return ""
-            return copy_to_session_and_get_url(ply_path, session_id)
+                return "Model file not found", ""
+            url = copy_to_session_and_get_url(ply_path, session_id)
+            return "Loading model into viewer...", url
         
         # Wrapper to return only the outputs we want to display
         def _generate_and_filter_outputs(image_path, auto_crop, guidance_scale, random_seed, num_steps):
@@ -432,7 +474,7 @@ def main():
         gr.Markdown("""
         ### 💡 Tips for Best Results
         - **Works best with near-frontal portrait images**
-        - **Note:** These checkpoints were not trained with accessories (glasses, hats, etc.). Portraits containing accessories may produce suboptimal results.
+        - **The provided checkpoints were not trained with accessories (glasses, hats, etc.). Portraits containing accessories may produce suboptimal results.**
         """)
         
         with gr.Row():
@@ -455,7 +497,8 @@ def main():
             with gr.Column(scale=1):
                 gr.Markdown("### Interactive Gaussian Splat Viewer")
                 viewer = gr.HTML("<div id='splat-container' style='width:100%;height:600px;background:#000;border:1px solid #ccc;border-radius:8px;'></div>")
-                url_box = gr.Textbox(label="Scene URL (auto-filled)", interactive=False, visible=False)
+                viewer_status = gr.Textbox(label="Viewer Status", value="Waiting for model...", interactive=False)
+                url_box = gr.Textbox(label="Model URL (for debugging)", interactive=False)
                 reload_btn = gr.Button("Reload Viewer", size="sm")
                 
                 out_recon = gr.Image(label="3D Reconstruction")
@@ -473,10 +516,10 @@ def main():
         ).then(
             fn=_prep_viewer_url,
             inputs=[out_ply, session],
-            outputs=url_box,
+            outputs=[viewer_status, url_box],
         ).then(
             fn=None, inputs=url_box, outputs=None,
-            js="(url)=>window.__load_splat__(url)"
+            js="(url)=>{console.log('Calling __load_splat__ with:', url); return window.__load_splat__(url);}"
         )
 
         # Manual reload if needed