Update main.py

main.py

@@ -113,10 +113,35 @@ def save_image_locally(image, filename):
     image.save(filepath, format="PNG")
     return filepath
 
+
+def make_request_with_retries(url, headers, payload, retries=3, delay=2):
+    """
+    Makes an HTTP POST request with retries in case of failure.
+    :param url: The URL for the request.
+    :param headers: Headers to include in the request.
+    :param payload: Payload to include in the request.
+    :param retries: Number of retries on failure.
+    :param delay: Delay between retries.
+    :return: Response JSON from the server.
+    """
+    for attempt in range(retries):
+        try:
+            with requests.Session() as session:
+                response = session.post(url, headers=headers, json=payload, timeout=30)
+                response.raise_for_status()
+                return response.json()
+        except requests.exceptions.RequestException as e:
+            if attempt < retries - 1:
+                time.sleep(delay)
+                continue
+            else:
+                raise HTTPException(status_code=500, detail=f"Request failed after {retries} attempts: {str(e)}")
+
+
 def fetch_image(url):
     try:
         with requests.Session() as session:
-            response = session.get(url, timeout=10)
+            response = session.get(url, timeout=30)
             response.raise_for_status()
             return Image.open(BytesIO(response.content))
     except Exception as e: