update app

app.py

@@ -275,7 +275,7 @@ def generate_3d(
         # 2. Process Mesh
         progress(0.7, desc="Processing Mesh...")
         mesh = outputs[0]
-        # FIX: Aggressive simplification to prevent CUDA errors during post-processing
+        # Aggressive simplification to prevent CUDA errors during post-processing
         mesh.simplify(1000000) 
         
         # 3. Export to GLB
@@ -323,7 +323,7 @@ def generate_3d(
         timestamp = now.strftime("%Y-%m-%dT%H%M%S")
         glb_path = os.path.join(user_dir, f'trellis_output_{timestamp}.glb')
         
-        # FIX: extension_webp=False ensures compatibility with Rerun/Standard Viewers
+        # extension_webp=False ensures compatibility with Rerun/Standard Viewers
         glb.export(glb_path, extension_webp=False)
         
         # --- Rerun Visualization Logic ---
@@ -341,22 +341,42 @@ def generate_3d(
             rr.init("TRELLIS-3D-Viewer", recording_id=run_id, spawn=False)
             rec = rr
             
-        # 1. CAMERA VIEW: Set View Coordinates (GLB is Right-Hand Y-Up)
-        # This fixes the "rotated/sleeping model" issue in the viewer
+        # 1. SETUP STATIC ENVIRONMENT
         rec.log("world", rr.ViewCoordinates.RIGHT_HAND_Y_UP, static=True)
-
-        # 2. SPECIAL FEATURES: Orientation Axes (Red=X, Green=Y, Blue=Z)
+        
+        # Log Axes Helpers (X=Red, Y=Green, Z=Blue)
         try:
-            # Arrows to show the origin
-            rec.log("world/axes/x", rr.Arrows3D(vectors=[[0.6, 0, 0]], colors=[[255, 0, 0]]), static=True)
-            rec.log("world/axes/y", rr.Arrows3D(vectors=[[0, 0.6, 0]], colors=[[0, 255, 0]]), static=True)
-            rec.log("world/axes/z", rr.Arrows3D(vectors=[[0, 0, 0.6]], colors=[[0, 0, 255]]), static=True)
+            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)
+            rec.log("world/axes/z", rr.Arrows3D(vectors=[[0, 0, 0.5]], colors=[[0, 0, 255]]), static=True)
         except Exception:
-            pass # Ignore if Arrows3D not available in this version
+            pass
 
-        # 3. Log the Model
-        rec.log("world/model", rr.Asset3D(path=glb_path), static=True)
+        # 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)
         
+        # 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)