changes in reading images

.gitignore

@@ -0,0 +1 @@
+.env

app.py

@@ -13,6 +13,10 @@ from time import sleep
 import json
 from functions import combine_images_side_by_side
 
+from dotenv import load_dotenv
+load_dotenv()
+
+
 app = FastAPI()
 openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
@@ -80,6 +84,14 @@ async def make_base_image(character_lora: str, character_keyword: str, outfit_de
     return(result)
 
 # Function related to virtual outfit try on
+async def safe_read_file(file: UploadFile):
+    print("safe read file")
+    print(file)
+    if file is None:
+        return None
+    content = await file.read()
+    return content if content else None
+
 
 async def run_virtual_tryon_pipeline(
     request_id,
@@ -95,14 +107,23 @@ async def run_virtual_tryon_pipeline(
         r.set(request_id, "checking incoming data...")
         front_type = ""
         back_type = ""
-        
+        print("checking incoming data")
         # Read all files once at the start
-        top_garment_data = await top_garment_image.read() if top_garment_image else None
-        bottom_garment_data = await bottom_garment_image.read() if bottom_garment_image else None
-        top_back_garment_data = await top_back_garment_image.read() if top_back_garment_image else None
-        bottom_back_garment_data = await bottom_back_garment_image.read() if bottom_back_garment_image else None
+        try:
+            print("reading top front")
+            top_garment_data = await safe_read_file(top_garment_image)
+            print("reading bottom front")
+            bottom_garment_data = await safe_read_file(bottom_garment_image)
+            print("reading top back")
+            top_back_garment_data = await safe_read_file(top_back_garment_image)
+            print("reading bottom back")
+            bottom_back_garment_data = await safe_read_file(bottom_back_garment_image)
+        except Exception as e:
+            print(f"Error reading garment images: {e}")
+            #return {"error": "Error reading garment images"}
 
         if top_garment_data is None and bottom_garment_data is None:
+            print("Missing garment images from front")
             return {"error": "Missing garment images"}
             
         if top_garment_data and bottom_garment_data:

functions.py

@@ -16,6 +16,7 @@ async def combine_images_side_by_side(image1: bytes, image2: bytes) -> str:
     Returns:
         Base64 encoded string of the combined image
     """
+    print("combine images side by side")
     # Create temp directory if it doesn't exist
     os.makedirs('tmp', exist_ok=True)
     
@@ -46,7 +47,7 @@ async def combine_images_side_by_side(image1: bytes, image2: bytes) -> str:
     
     # Clean up
     os.remove(filename)
-    
+    print("combined images side by side")
     return encoded_string
 
 # Example usage:

requirements.txt

@@ -6,4 +6,5 @@ fal-client
 requests
 openai
 redis
-Pillow>=10.0.0
+Pillow>=10.0.0
+python-dotenv