Upload app.py

app.py

@@ -39,7 +39,15 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-index = faiss.read_index("./model/db_vit_b_16.index")
+index_path = "./model/db_vit_b_16.index"
+
+if not os.path.exists(index_path):
+    raise FileNotFoundError(f"Index file not found: {index_path}")
+
+try:
+    index = faiss.read_index(index_path)
+except RuntimeError as e:
+    raise RuntimeError(f"Error reading FAISS index: {e}")
 feature_extractor = FeatureExtractor(base_model="vit_b_16")
 
 if torch.backends.mps.is_built():
@@ -124,20 +132,18 @@ async def search_image(body: ImageSearchBody):
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
 
-from src.firebase.firebase_provider import (
-    upload_base64_image_to_storage,
-)
+from src.firebase.firebase_provider import process_images
 
 
 class Body(BaseModel):
-    base64_image: str = Field(..., title="Base64 Image String")
-    image_name: str = Field(..., title="Image Name")
+    base64_image: list[str] = Field(..., title="Base64 Image String")
     model_config = {
         "json_schema_extra": {
             "examples": [
                 {
-                    "base64_image": "iVBORw0KGgoAAAANSUhEUgAAABQAAAAUCAYAAACNiR0NAAABdUlEQVR42mNk",
-                    "image_name": "image1",
+                    "base64_image": [
+                        "iVBORw0KGgoAAAANSUhEUgAAABQAAAAUCAYAAACNiR0NAAABdUlEQVR42mNk",
+                    ]
                 }
             ]
         }
@@ -145,9 +151,9 @@ class Body(BaseModel):
 
 
 @app.post("/upload_image")
-def upload_image(body: Body):
+async def upload_image(body: Body):
     try:
-        public_url = upload_base64_image_to_storage(body.base64_image, body.image_name)
+        public_url = await process_images(body.base64_image)
         return JSONResponse(content={"public_url": public_url}, status_code=200)
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)