Changed image upload technique

app/main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 
@@ -9,10 +9,9 @@ import io
 from app.services.embedding_service import EmbeddingService
 from app.services.similarity_service import SimilarityService
 from app.services.ocr_service import OCRService
-from app.schemas import CardResponse
+from app.schemas import CardResponse, ImagePayload
 
-import requests
-import os
+import base64
 
 
 # ---------------
@@ -51,14 +50,13 @@ def health():
 
 
 @app.post("/predict", response_model=CardResponse)
-async def predict(file: UploadFile = File(...)):
+async def predict(payload: ImagePayload):
     try:
-        image_bytes = await file.read()
+        image_bytes = base64.b64decode(payload.image_b64)  # <-- decode base64
         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:

app/schemas.py

@@ -36,4 +36,13 @@ class CardResponse(BaseModel):
     hp: Optional[str] = None
     types: Optional[list[str]] = None
     moves: Optional[list[Move]] = None
-    similar_cards: list[SimilarCard] = []
+    similar_cards: list[SimilarCard] = []
+
+
+# --------------------------------
+# ----- IMAGE PAYLOAD ------------
+# --------------------------------
+
+class ImagePayload(BaseModel):
+    image_b64: str
+    filename: str