Update routers/models.py

routers/models.py

@@ -69,24 +69,6 @@ def get_hf_service():
     return hf_service
 
 
-def cleanup_model_file(glb_path: Path):
-    """Cleanup function to delete model file after download.
-
-    This runs as a background task after the response is sent to the client.
-
-    Args:
-        glb_path: Path to the GLB file to delete
-    """
-    try:
-        if glb_path.exists():
-            glb_path.unlink()
-            logger.info(f"✓ Deleted GLB file after download: {glb_path}")
-        else:
-            logger.warning(f"GLB file not found for deletion: {glb_path}")
-    except Exception as delete_error:
-        logger.warning(f"Failed to delete GLB file after download: {delete_error}")
-
-
 @router.post("/generate", response_model=GenerationResponse)
 async def generate_model(request: PromptRequest):
     """Generate a 3D model from a text prompt.
@@ -199,21 +181,18 @@ async def download_model(model_id: str, background_tasks: BackgroundTasks):
         raise HTTPException(status_code=404, detail="Model file not found")
 
     try:
-        # Apply brightness normalization to GLB file before serving
-        logger.info("Applying brightness normalization to GLB file...")
-        # normalize_materials_for_ar(glb_path)
+        # Apply brightness normalization for AR visibility before serving
+        logger.info(
+            "Applying brightness normalization to GLB file for AR visibility..."
+        )
+        normalize_materials_for_ar(glb_path)
         logger.info("✓ Brightness normalization applied to GLB")
 
-        # Read normalized GLB file content
+        # Read GLB file content
         with open(glb_path, "rb") as f:
             glb_content = f.read()
 
-        logger.info(
-            f"Serving normalized GLB file: {glb_path} ({len(glb_content)} bytes)"
-        )
-
-        # Schedule cleanup task to run after response is sent
-        # background_tasks.add_task(cleanup_model_file, glb_path)
+        logger.info(f"Serving GLB file: {glb_path} ({len(glb_content)} bytes)")
 
         # Create response
         response = Response(