Update api/src/main.py

api/src/main.py

@@ -10,19 +10,23 @@ import sys
 from pathlib import Path
 import requests as http_requests
 
-# Path Setup
+# --- Path Setup ---
 root_dir = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(root_dir))
 
 # Import your core engine
-from core import processor
+try:
+    from core import processor
+except ImportError:
+    print("Error: 'core.processor' nahi mila. Check karein ke 'core' folder sahi jagah hai.")
 
 app = FastAPI()
 
-# CORS Fix for Mobile/Web
+# --- CORS Fix (Sirf ek baar clean code) ---
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
+    allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
@@ -37,103 +41,86 @@ def get_dirs():
 
 @app.get("/")
 def health_check():
-    return {"status": "active", "service": "CarVite AI"}
+    return {"status": "active", "service": "CarVite AI Backend"}
 
+# --- Main Processing Endpoint ---
+# Dono routes add kar diye hain taake frontend jo bhi call kare chal jaye
 @app.post("/process")
 @app.post("/upload-image")
 async def process_car_image(
     image: UploadFile = File(...),
-    action: str = Form("remove"),  # 'remove' or 'replace'
+    action: str = Form("remove"),  # Default: remove background
     bg_color: str = Form(None),
     bg_url: str = Form(None),
 ):
     input_dir, output_dir = get_dirs()
-    
-    # Save original image with unique ID
     file_id = int(time.time() * 1000)
-    file_extension = Path(image.filename).suffix or ".jpg"
-    input_filename = f"in_{file_id}{file_extension}"
-    input_path = input_dir / input_filename
     
-    try:
-        with input_path.open("wb") as buffer:
-            shutil.copyfileobj(image.file, buffer)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"File save error: {str(e)}")
+    # Save Input Image
+    ext = Path(image.filename).suffix or ".jpg"
+    input_path = input_dir / f"in_{file_id}{ext}"
+    
+    with input_path.open("wb") as buffer:
+        shutil.copyfileobj(image.file, buffer)
 
     try:
-        # 1. Core Background Removal
-        # Isnet model use ho raha hai car cutout nikalne k liye
+        # 1. Background Remove (Hamesha pehle cutout nikalte hain)
         processed_img, _ = processor.process_image(str(input_path), model_name="isnet-general-use")
         processed_img = processed_img.convert("RGBA")
+        
+        output_path = output_dir / f"out_{file_id}.png"
 
-        output_filename = f"out_{file_id}.png"
-        output_path = output_dir / output_filename
-
-        # 2. Case: JUST REMOVE BACKGROUND (Transparent PNG)
-        if action == "remove":
+        # 2. Logic: Sirf Remove karna hai ya Replace?
+        if action == "remove" and not bg_url and not bg_color:
+            # Sirf Transparent PNG save karo
             processed_img.save(output_path, "PNG")
-            return FileResponse(str(output_path), media_type="image/png")
-
-        # 3. Case: ENHANCE / REPLACE BACKGROUND
-        # Agar user ne color ya URL diya hai toh replace karega
-        final_img = processed_img # Fallback
-
-        if bg_url:
-            try:
+        
+        else:
+            # Background Replacement Logic
+            final_img = processed_img
+            
+            if bg_url:
+                # Agar URL se background lagana hai
                 bg_res = http_requests.get(bg_url, timeout=10)
-                bg_res.raise_for_status()
                 bg_img = Image.open(io.BytesIO(bg_res.content)).convert("RGBA")
                 bg_img = bg_img.resize(processed_img.size, Image.LANCZOS)
-                
-                # Combine layers
-                combined = Image.new("RGBA", processed_img.size)
-                combined.paste(bg_img, (0, 0))
-                combined.paste(processed_img, (0, 0), processed_img)
-                final_img = combined
-            except Exception as e:
-                print(f"BG URL Error: {e}")
-                # Agar background fail ho jaye toh sirf cutout bhej do bajaye crash hone k
-                final_img = processed_img
-
-        elif bg_color:
-            try:
-                # Color code like '#FFFFFF' or 'red'
+                final_img = Image.alpha_composite(bg_img, processed_img)
+            
+            elif bg_color:
+                # Agar solid color lagana hai
                 bg_img = Image.new("RGBA", processed_img.size, bg_color)
-                bg_img.paste(processed_img, (0, 0), processed_img)
-                final_img = bg_img
-            except Exception as e:
-                print(f"BG Color Error: {e}")
-                final_img = processed_img
+                final_img = Image.alpha_composite(bg_img, processed_img)
+            
+            final_img.save(output_path, "PNG")
 
-        # Save and Send
-        final_img.save(output_path, "PNG")
         return FileResponse(str(output_path), media_type="image/png")
 
     except Exception as e:
-        print(f"AI Processor Error: {str(e)}")
+        print(f"AI Error: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Processing failed: {str(e)}")
     finally:
         image.file.close()
 
-# Special endpoint for custom background uploads
+# --- Custom Background Upload Endpoint ---
 @app.post("/replace-background")
 async def replace_custom_bg(
-    image: UploadFile = File(...),
-    background: UploadFile = File(...),
+    image: UploadFile = File(..., description="Car Image"),
+    background: UploadFile = File(..., description="Custom BG Image"),
     car_size: float = Form(0.65),
     smart_placement: bool = Form(True)
 ):
     input_dir, output_dir = get_dirs()
-    file_id = int(time.time())
+    file_id = int(time.time() * 1000)
     
     fg_path = input_dir / f"fg_{file_id}_{image.filename}"
     bg_path = input_dir / f"bg_{file_id}_{background.filename}"
+    output_path = output_dir / f"final_{file_id}.png"
     
     with fg_path.open("wb") as f: shutil.copyfileobj(image.file, f)
     with bg_path.open("wb") as b: shutil.copyfileobj(background.file, b)
 
     try:
+        # Core replacement function call
         result_img = processor.replace_background(
             foreground_input=str(fg_path),
             background_input=str(bg_path),
@@ -141,8 +128,20 @@ async def replace_custom_bg(
             smart_placement=smart_placement
         )
         
-        output_path = output_dir / f"custom_{file_id}.png"
         result_img.save(output_path, "PNG")
         return FileResponse(str(output_path), media_type="image/png")
+
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+    finally:
+        image.file.close()
+        background.file.close()
+
+# --- Serve Processed Images (Just in case) ---
+@app.get("/output/{filename}")
+def get_output(filename: str):
+    _, output_dir = get_dirs()
+    file_path = output_dir / filename
+    if not file_path.exists():
+        raise HTTPException(status_code=404, detail="File not found")
+    return FileResponse(file_path)