Update app.py

app.py

@@ -275,7 +275,8 @@ def generate_3d(
         # 2. Process Mesh
         progress(0.7, desc="Processing Mesh...")
         mesh = outputs[0]
-        # Aggressive simplification to prevent CUDA errors during post-processing
+        # FIX: Aggressive simplification to prevent CUDA errors during post-processing
+        # This keeps the mesh manageable for the fill_holes kernel
         mesh.simplify(1000000) 
         
         # 3. Export to GLB
@@ -283,7 +284,7 @@ def generate_3d(
         
         grid_size = latents[2] 
         
-        # Try-Except block for safe export (fallback if hole filling fails)
+        # Safe export with fallback if high-quality hole filling fails
         try:
             glb = o_voxel.postprocess.to_glb(
                 vertices=mesh.vertices,
@@ -302,7 +303,7 @@ def generate_3d(
             )
         except RuntimeError as e:
             print(f"Warning: Post-processing failed with remesh=True. Error: {e}")
-            print("Retrying with remesh=False...")
+            print("Retrying with remesh=False (Standard mesh generation)...")
             glb = o_voxel.postprocess.to_glb(
                 vertices=mesh.vertices,
                 faces=mesh.faces,
@@ -323,7 +324,8 @@ def generate_3d(
         timestamp = now.strftime("%Y-%m-%dT%H%M%S")
         glb_path = os.path.join(user_dir, f'trellis_output_{timestamp}.glb')
         
-        # extension_webp=False ensures compatibility with Rerun/Standard Viewers
+        # FIX: extension_webp=False ensures compatibility with Rerun/Standard Viewers
+        # This solves the "Missing data" and "Unsupported extension" errors.
         glb.export(glb_path, extension_webp=False)
         
         # --- Rerun Visualization Logic ---
@@ -341,10 +343,15 @@ def generate_3d(
             rr.init("TRELLIS-3D-Viewer", recording_id=run_id, spawn=False)
             rec = rr
             
-        # 1. SETUP STATIC ENVIRONMENT
+        # 1. OPTIMIZATION: Clear the view to avoid ghosting
+        rec.log("world", rr.Clear(recursive=True), static=True)
+
+        # 2. OPTIMIZATION: Set View Coordinates
+        # GLB files are commonly Right-Hand, Y-Up. This ensures the model stands up.
         rec.log("world", rr.ViewCoordinates.RIGHT_HAND_Y_UP, static=True)
-        
-        # Log Axes Helpers (X=Red, Y=Green, Z=Blue)
+
+        # 3. OPTIMIZATION: Add Axes Helpers
+        # Visual cues for orientation (X=Red, Y=Green, Z=Blue)
         try:
             rec.log("world/axes/x", rr.Arrows3D(vectors=[[0.5, 0, 0]], colors=[[255, 0, 0]]), static=True)
             rec.log("world/axes/y", rr.Arrows3D(vectors=[[0, 0.5, 0]], colors=[[0, 255, 0]]), static=True)
@@ -352,31 +359,10 @@ def generate_3d(
         except Exception:
             pass
 
-        # 2. LOG MODEL (Static relative to its parent 'anchor')
-        # We place the model inside 'world/anchor/model' so we can rotate 'world/anchor'
-        rec.log("world/anchor/model", rr.Asset3D(path=glb_path), static=True)
+        # 4. Log the 3D Model
+        # Use Asset3D with 'path' to embed the GLB efficiently
+        rec.log("world/model", rr.Asset3D(path=glb_path), static=True)
         
-        # 3. CREATE ROTATION ANIMATION (Auto-Rotate)
-        # We animate the 'anchor' transform over a timeline
-        progress(0.98, desc="Generating Rotation Animation...")
-        
-        num_frames = 240 # 240 frames for one full loop
-        for i in range(num_frames):
-            # Set the time cursor
-            rec.set_time_sequence("frame", i)
-            
-            # Calculate rotation angle (0 to 360 degrees)
-            angle_rad = (i / num_frames) * 2 * np.pi
-            
-            # Log the rotation transform for the anchor
-            # This rotates the model around the Y-axis
-            rec.log(
-                "world/anchor", 
-                rr.Transform3D(
-                    rotation=rr.RotationAxisAngle(axis=[0, 1, 0], radians=angle_rad)
-                )
-            )
-
         # Save the Rerun recording (.rrd)
         rrd_path = os.path.join(user_dir, f'trellis_output_{timestamp}.rrd')
         rec.save(rrd_path)