Added try/except to catch main.py errors"

app/main.py

@@ -1,5 +1,7 @@
 from fastapi import FastAPI, File, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+
 from contextlib import asynccontextmanager
 from PIL import Image
 import io
@@ -50,29 +52,29 @@ def health():
 
 @app.post("/predict", response_model=CardResponse)
 async def predict(file: UploadFile = File(...)):
-    image_bytes = await file.read()
-    image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
-
-    # Generate embedding and find similar cards first
-    embedding = app.state.embedding_service.embed(image)
-    similar_cards = app.state.similarity_service.search(embedding, top_k=5)
-
-    # Use OCR for extraction
-    ocr_data = app.state.ocr_service.extract(image)
-
-    # If top match is very confident, use its metadata to fill in OCR gaps
-    if similar_cards and similar_cards[0]["score"] > 0.99:
-        top_match = similar_cards[0]
-        ocr_data["name"] = ocr_data["name"] or top_match["name"]
-        ocr_data["types"] = ocr_data["types"] or top_match["types"]
-
-    return CardResponse(
-        name=ocr_data.get("name"),
-        hp=ocr_data.get("hp"),
-        types=ocr_data.get("types"),
-        moves=ocr_data.get("moves"),
-        similar_cards=similar_cards
-    )
+    try:
+        image_bytes = await file.read()
+        image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+
+        embedding = app.state.embedding_service.embed(image)
+        similar_cards = app.state.similarity_service.search(embedding, top_k=5)
+
+        ocr_data = app.state.ocr_service.extract(image)
+
+        if similar_cards and similar_cards[0]["score"] > 0.99:
+            top_match = similar_cards[0]
+            ocr_data["name"] = ocr_data["name"] or top_match["name"]
+            ocr_data["types"] = ocr_data["types"] or top_match["types"]
+
+        return CardResponse(
+            name=ocr_data.get("name"),
+            hp=ocr_data.get("hp"),
+            types=ocr_data.get("types"),
+            moves=ocr_data.get("moves"),
+            similar_cards=similar_cards
+        )
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"error": str(e)})
 
 
 import uvicorn