Update app.py

app.py

@@ -17,24 +17,45 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Initialize CarveKit
-try:
-    interface = HiInterface(
-        object_type="object",  # Can be "object" or "hairs-like"
-        batch_size_seg=5,
-        batch_size_matting=1,
-        device='cpu',  # Use 'cuda' if GPU is available
-        seg_mask_size=640,
-        matting_mask_size=2048,
-        trimap_prob_threshold=231,
-        trimap_kernel_size=30,
-        trimap_erosion_iters=5,
-        fp16=False
-    )
-    logger.info("CarveKit interface initialized successfully")
-except Exception as e:
-    logger.error(f"Failed to initialize CarveKit: {e}")
-    interface = None
+# Initialize CarveKit with proper cache handling
+import os
+interface = None
+
+def initialize_carvekit():
+    """Initialize CarveKit with proper error handling and cache setup"""
+    global interface
+    try:
+        # Set cache directory
+        cache_dir = os.environ.get('CARVEKIT_CACHE_DIR', '/app/.cache/carvekit')
+        os.makedirs(cache_dir, exist_ok=True)
+        
+        # Set environment variable for CarveKit
+        os.environ['CARVEKIT_CACHE_DIR'] = cache_dir
+        
+        # Import CarveKit after setting up cache
+        from carvekit.api.high import HiInterface
+        
+        interface = HiInterface(
+            object_type="object",  # Can be "object" or "hairs-like"
+            batch_size_seg=5,
+            batch_size_matting=1,
+            device='cpu',  # Use 'cuda' if GPU is available
+            seg_mask_size=640,
+            matting_mask_size=2048,
+            trimap_prob_threshold=231,
+            trimap_kernel_size=30,
+            trimap_erosion_iters=5,
+            fp16=False
+        )
+        logger.info("CarveKit interface initialized successfully")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to initialize CarveKit: {e}")
+        interface = None
+        return False
+
+# Try to initialize CarveKit
+carvekit_ready = initialize_carvekit()
 
 # Create FastAPI app
 app = FastAPI(
@@ -98,7 +119,8 @@ async def health_check():
     """Health check endpoint"""
     return {
         "status": "healthy",
-        "carvekit_ready": interface is not None
+        "carvekit_ready": interface is not None,
+        "carvekit_initialized": carvekit_ready
     }
 
 @app.post("/api/remove-background")