ok

app.py

@@ -41,24 +41,17 @@ class ProcessRequest(BaseModel):
 def download_image_as_pil(url: str, timeout: int = 10) -> Image.Image:
     try:
         response = requests.get(url, stream=True, headers=REQUESTS_HEADERS, timeout=timeout)
-        print(f"URL: {url}, Status Code: {response.status_code}, Content-Type: {response.headers.get('Content-Type')}")
-        
         if response.status_code == 200 and 'image' in response.headers.get('Content-Type', ''):
             try:
-                with open("temp_image.jpg", "wb") as f:
-                    f.write(response.content)  # Save the image to a temporary file
-                return Image.open("temp_image.jpg").convert("RGB")
+                return Image.open(response.raw).convert("RGB")
             except UnidentifiedImageError:
                 print(f"Unidentified image file from URL: {url}")
-        else:
-            print(f"Non-image content for URL: {url}")
         return None
     except Exception as e:
         print(f"Error downloading image: {e}")
         return None
 
 
-
 def process_batch(batch_products, batch_images, results):
     try:
         # Generate embeddings
@@ -172,8 +165,6 @@ async def process_images(request: ProcessRequest):
 async def health_check():
     return {"status": "healthy"}
 
-
-##Local test
 # if __name__ == "__main__":
 #     import uvicorn
 #     uvicorn.run(app, host="0.0.0.0", port=8000)

requirements.txt

@@ -10,6 +10,7 @@ datasets
 Pillow
 python-dotenv
 tqdm
+fastapi
 
 # WooCommerce API
 woocommerce