Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 import tempfile
 import uuid
 import math
+import time
 from typing import *
 from datetime import datetime
 from pathlib import Path
@@ -148,12 +149,33 @@ def end_session(req: gr.Request):
         shutil.rmtree(user_dir)
 
 def remove_background(input: Image.Image) -> Image.Image:
-    with tempfile.NamedTemporaryFile(suffix='.png') as f:
-        input = input.convert('RGB')
-        input.save(f.name)
-        output = rmbg_client.predict(handle_file(f.name), api_name="/image")[0][0]
-        output = Image.open(output)
-        return output
+    # Try multiple times if GPU is busy
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as f:
+                input = input.convert('RGB')
+                input.save(f.name)
+                f_path = f.name
+            
+            # Call external API
+            output = rmbg_client.predict(handle_file(f_path), api_name="/image")[0][0]
+            
+            # Cleanup temp input
+            try:
+                os.remove(f_path)
+            except:
+                pass
+                
+            return Image.open(output)
+            
+        except Exception as e:
+            print(f"RMBG attempt {attempt+1} failed: {e}")
+            if attempt < max_retries - 1:
+                time.sleep(2) # Wait a bit before retry
+            else:
+                print("Returning original image as fallback.")
+                return input # Fallback: return original if RMBG fails
 
 def preprocess_image(input: Image.Image) -> Image.Image:
     """Preprocess the input image: remove bg, crop, resize."""
@@ -397,46 +419,26 @@ def generate_3d(
 
         # 5. Auto-Rotation Logic (Animation)
         if auto_rotate:
-            # Create an orbit animation
-            # Radius approx 1.5 units (assuming model is unit scale centered)
             radius = 2.0 
-            frames = 120 # Number of frames for one rotation
+            frames = 120 
             
             for i in range(frames):
                 angle = 2 * math.pi * (i / frames)
                 # Orbit in X-Z plane (Y is up)
                 x = radius * math.sin(angle)
                 z = radius * math.cos(angle)
-                y = 0.5 # Slightly elevated
-                
-                # Simple look-at transform logic
-                # Position: (x, y, z)
-                # Target: (0, 0, 0)
-                # Up: (0, 1, 0)
-                
-                # Rerun expects translation and rotation (quat or mat3)
-                # For simplicity in this demo, we just log translation to visualize path
-                # OR we log a camera object moving.
+                y = 0.5
                 
-                # Logging a Pinhole camera that moves
-                rr.set_time_sequence("frame", i)
+                # Use the correct recording object to set time
+                rec.set_time_sequence("frame", i)
                 
-                # Calculate Rotation Matrix to look at center
-                # Forward vector (cam to target)
+                # Rotation Matrix logic
                 fwd = np.array([-x, -y, -z])
                 fwd = fwd / np.linalg.norm(fwd)
-                # Right vector
                 up = np.array([0, 1, 0])
                 right = np.cross(fwd, up)
                 right = right / np.linalg.norm(right)
-                # Re-calculate Up
                 new_up = np.cross(right, fwd)
-                
-                # Rerun Camera convention: -Z forward, +Y down, +X right? 
-                # OR standard computer vision: Z forward.
-                # Since we set RIGHT_HAND_Y_UP world, we should align standard camera.
-                
-                # Construct rotation matrix [Right, Up, -Forward]
                 rot_mat = np.column_stack((right, new_up, -fwd))
                 
                 rec.log(