Update server.py

server.py

@@ -1,65 +1,83 @@
-from fastapi import FastAPI, File, UploadFile, Form
+from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
 import shutil
 import os
 import uvicorn
 
-# 1. Initialize the App (Only ONCE)
-app = FastAPI()
+# 1. Initialize the app (Only ONCE)
+app = FastAPI(title="ORBIT Visual Commerce API")
 
-# 2. CORS Middleware
+# 2. Correct CORS Configuration (Crucial for Netlify)
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
+    allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# 3. Ensure the images directory exists BEFORE mounting
+# 3. Ensure the static folder exists BEFORE mounting
 IMAGE_DIR = "Productimages"
 if not os.path.exists(IMAGE_DIR):
     os.makedirs(IMAGE_DIR)
 
 app.mount(f"/{IMAGE_DIR}", StaticFiles(directory=IMAGE_DIR), name=IMAGE_DIR)
 
-# 4. Import and Load Engine
-# We wrap this in try-except because a 4GB model often causes memory errors
+# 4. Initialize the Engine
+# Note: This will load your 4GB model into RAM
 try:
     from orbiitt_engine import OrbiittEngine
     engine = OrbiittEngine()
-    print("Engine loaded successfully.")
+    print("Successfully loaded OrbiittEngine.")
 except Exception as e:
-    print(f"CRITICAL: Failed to load engine: {e}")
+    print(f"Error loading OrbiittEngine: {e}")
     engine = None
 
 @app.get("/")
 def read_root():
     return {
-        "status": "online",
-        "engine_loaded": engine is not None,
-        "current_dir": os.getcwd(),
-        "files_visible": os.listdir(".")
+        "message": "Server is running",
+        "engine_status": "Ready" if engine else "Loading/Error",
+        "endpoints": ["/search", "/Productimages"]
     }
 
 @app.post("/search")
-async def search_endpoint(text: str = Form(None), weight: float = Form(0.5), file: UploadFile = File(None)):
+async def search_endpoint(
+    text: str = Form(None), 
+    weight: float = Form(0.5), 
+    file: UploadFile = File(None)
+):
     if engine is None:
-        return {"error": "Engine not initialized"}
-        
-    temp_path = None
-    if file:
-        temp_path = f"temp_{file.filename}"
-        with open(temp_path, "wb") as buffer:
-            shutil.copyfileobj(file.file, buffer)
+        raise HTTPException(status_code=503, detail="Engine not initialized")
 
+    temp_path = None
     try:
-        results = engine.search(text_query=text, image_file=temp_path, text_weight=weight)
+        if file:
+            # Use a unique temp name to avoid collisions
+            temp_path = f"temp_{file.filename}"
+            with open(temp_path, "wb") as buffer:
+                # Correctly read the async file content
+                content = await file.read()
+                buffer.write(content)
+
+        # Call the search engine
+        results = engine.search(
+            text_query=text, 
+            image_file=temp_path, 
+            text_weight=weight
+        )
+        return {"results": results}
+
+    except Exception as e:
+        print(f"Search error: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
     finally:
+        # Always clean up the temp file
         if temp_path and os.path.exists(temp_path):
             os.remove(temp_path)
-            
-    return {"results": results}
 
+# Corrected the execution block
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)