Upload 25 files

src/firebase/firebase_provider.py

@@ -75,12 +75,14 @@ def download_file_from_storage(file_name, destination_path):
 
 async def upload_base64_image_to_storage(base64_image: str, file_name: str) -> str:
     """
-    Upload a base64 image to Firebase Storage
+    Upload a base64 image to Firebase Storage.
+    
     param:
-        base64_image: str - The base64 encoded image
-        file_name: str - The name of the file to be uploaded
+        base64_image: str - The base64 encoded image.
+        file_name: str - The name of the file to be uploaded.
+    
     return:
-        str - The public URL of the uploaded file
+        str - The public URL of the uploaded file.
     """
     try:
         # Decode the base64 image
@@ -88,10 +90,13 @@ async def upload_base64_image_to_storage(base64_image: str, file_name: str) -> s
 
         # Convert the decoded image to a JPG file
         image = Image.open(io.BytesIO(image_data))
+        
+        # Generate a temporary file path
         temp_file_path = f"{tempfile.gettempdir()}/{file_name}.jpg"
 
-        async with aiofiles.open(temp_file_path, mode="wb") as f:
-            image.save(f, format="JPEG")
+        # Since image.save is a blocking operation, run it in a thread pool
+        loop = asyncio.get_event_loop()
+        await loop.run_in_executor(None, lambda: image.save(temp_file_path, format="JPEG"))
 
         # Upload the JPG file to Firebase Storage
         public_url = await upload_file_to_storage(temp_file_path, f"{file_name}.jpg")
@@ -117,7 +122,12 @@ async def process_images(base64_images: List[str]) -> List[str]:
             base64.urlsafe_b64encode(timestamp.encode()).decode().rstrip("=")
         )
         file_name = f"image_{encoded_timestamp}_{idx}"
+
+        # If you're performing image processing, make sure to use a thread pool
         tasks.append(upload_base64_image_to_storage(base64_image, file_name))
 
+    # Gather all async tasks
     final_output = await asyncio.gather(*tasks)
+    print("final_output", final_output)
     return final_output
+

src/utils/helper.py

@@ -0,0 +1,31 @@
+import os
+import torch
+import faiss
+import base64
+from PIL import Image
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
+from io import BytesIO
+from src.modules import FeatureExtractor
+from fastapi.middleware.cors import CORSMiddleware
+import zipfile
+from pydantic import BaseModel, Field
+import json
+from dotenv import load_dotenv
+
+
+
+
+def base64_to_image(base64_str: str) -> Image.Image:
+    try:
+        image_data = base64.b64decode(base64_str)
+        image = Image.open(BytesIO(image_data)).convert("RGB")
+        return image
+    except Exception as e:
+        raise HTTPException(status_code=400, detail="Invalid Base64 image")
+
+
+def image_to_base64(image: Image.Image) -> str:
+    buffered = BytesIO()
+    image.save(buffered, format="JPEG")
+    return base64.b64encode(buffered.getvalue()).decode("utf-8")