Update app.py

app.py

@@ -103,28 +103,34 @@ class StoryCreationDTO(BaseModel):
 
 import httpx
 import tempfile
+import asyncio
 
-async def download_file_from_url(url: str) -> str | None:
+async def download_file_from_url(url: str, retries: int = 3, delay: float = 2.0) -> str | None:
     """
     Downloads a file from a URL and returns the path to a temporary file.
-    Returns None if the download fails.
+    Retries on failure up to `retries` times, waiting `delay` seconds between attempts.
+    Returns None if all attempts fail.
     """
-    try:
-        async with httpx.AsyncClient() as client:
-            response = await client.get(url, timeout=60.0)  # optional timeout
-            response.raise_for_status()  # raises for non-200 status codes
-
-        # Save to a temporary file
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
-        temp_file.write(response.content)
-        temp_file.close()
-        print(f"downloaded {url} successfully")
-        return temp_file.name
-
-    except Exception as e:
-        # Catch any exception (network, timeout, invalid response)
-        print(f"Failed to download {url}: {e}")
-        return None
+    for attempt in range(1, retries + 1):
+        try:
+            async with httpx.AsyncClient(timeout=60.0) as client:  # increased timeout
+                response = await client.get(url)
+                response.raise_for_status()  # raises for non-200 status codes
+
+            # Save to a temporary file
+            temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
+            temp_file.write(response.content)
+            temp_file.close()
+            print(f"Downloaded {url} successfully on attempt {attempt}")
+            return temp_file.name
+
+        except Exception as e:
+            print(f"Attempt {attempt} failed for {url}: {e}")
+            if attempt < retries:
+                await asyncio.sleep(delay)  # wait before retrying
+
+    print(f"All {retries} attempts failed for {url}")
+    return None
 
 #-----------------------------------------------------------