Update app.py

app.py

@@ -105,66 +105,66 @@ import httpx
 import tempfile
 import asyncio
 
-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.
-    Retries on failure up to `retries` times, waiting `delay` seconds between attempts.
-    Returns None if all attempts fail.
-    """
-    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
-
-# download_cache = {}
-
 # 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.
-#     If download fails after `retries` attempts, returns None instead of raising an error.
-#     Caches successful downloads to avoid repeated requests.
+#     Retries on failure up to `retries` times, waiting `delay` seconds between attempts.
+#     Returns None if all attempts fail.
 #     """
-#     if url in download_cache:
-#         print(f"{url} is got from cache")
-#         return download_cache[url]
-    
 #     for attempt in range(1, retries + 1):
 #         try:
-#             async with httpx.AsyncClient(timeout=60.0) as client:
+#             async with httpx.AsyncClient(timeout=60.0) as client:  # increased timeout
 #                 response = await client.get(url)
-#                 response.raise_for_status()
+#                 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()
-
-#             download_cache[url] = temp_file.name
+#             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)
+#                 await asyncio.sleep(delay)  # wait before retrying
 
-#     print(f"All {retries} attempts failed for {url}, skipping...")
+#     print(f"All {retries} attempts failed for {url}")
 #     return None
 
+download_cache = {}
+
+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.
+    If download fails after `retries` attempts, returns None instead of raising an error.
+    Caches successful downloads to avoid repeated requests.
+    """
+    if url in download_cache:
+        print(f"{url} is got from cache")
+        return download_cache[url]
+    
+    for attempt in range(1, retries + 1):
+        try:
+            async with httpx.AsyncClient(timeout=60.0) as client:
+                response = await client.get(url)
+                response.raise_for_status()
+
+            temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
+            temp_file.write(response.content)
+            temp_file.close()
+
+            download_cache[url] = temp_file.name
+            return temp_file.name
+
+        except Exception as e:
+            print(f"Attempt {attempt} failed for {url}: {e}")
+            if attempt < retries:
+                await asyncio.sleep(delay)
+
+    print(f"All {retries} attempts failed for {url}, skipping...")
+    return None
+
 #-----------------------------------------------------------
 
 #takes the text to be said and path to the prosody audio and path to save the generated audio and returns path to the generated audio 
@@ -215,7 +215,7 @@ async def generate_story_audios(story: StoryCreationDTO, base_output: str):
             audio_file=prosody_file_title,
             save_path=title_save_path
         )
-        os.remove(prosody_file_title)
+        # os.remove(prosody_file_title)
 
         for scene in chapter.scenes:
             scene_dir = chapter_dir / scene.sceneId
@@ -236,7 +236,7 @@ async def generate_story_audios(story: StoryCreationDTO, base_output: str):
                     audio_file=prosody_file,
                     save_path=sentence_save_path
                 )
-                os.remove(prosody_file)
+                # os.remove(prosody_file)
 
 #_______________ Concatenating the generated audios to make the final story (post-processing)_______________________
 
@@ -295,7 +295,7 @@ async def concat_story_audio(story: StoryCreationDTO, base_output: str, final_pa
                     sfx_file_wav = ensure_wav(sfx_file)
                     sfx_audio = AudioSegment.from_wav(sfx_file_wav)
                     scene_audio = scene_audio.overlay(sfx_audio)
-                    os.remove(sfx_file)
+                    # os.remove(sfx_file)
                 else:
                     print(f"SFX skipped for {scene.location.locationName}")
 
@@ -314,7 +314,7 @@ async def concat_story_audio(story: StoryCreationDTO, base_output: str, final_pa
                     bg_audio = bg_audio * loops
                 bg_audio = bg_audio[:len(scene_audio)]  # trim to match scene
                 scene_audio = scene_audio.overlay(bg_audio)
-                os.remove(bg_file)
+                # os.remove(bg_file)
 
             # Add 2 seconds of silence between scenes
             scene_audio += AudioSegment.silent(duration=2000)
@@ -534,6 +534,12 @@ async def get_results(task_id: str):
     # Ensure result exists and has all required fields
     result = task.get("result")
     if result and all(k in result for k in ("fileName", "duration", "audioBase64")):
+        #clearing cache
+        for file_path in download_cache.values():
+            if os.path.exists(file_path):
+                os.remove(file_path)
+        download_cache.clear()
+        
         return {"status": "completed", **result}
     else:
         # If result is missing fields, mark as still processing