Update app.py

app.py

@@ -132,8 +132,28 @@ import asyncio
 #     print(f"All {retries} attempts failed for {url}")
 #     return None
 
+
 download_cache = {}
 
+async def download_scene_files(scene: SceneDto):
+    tasks = []
+
+    # Sentence prosody references
+    for sentence in scene.sentences:
+        tasks.append(download_file_from_url(sentence.prosodyReference))
+
+    # Location SFX
+    if scene.location.path:
+        tasks.append(download_file_from_url(scene.location.path))
+
+    # Background music
+    if scene.bgMusic and scene.bgMusic.musicPath:
+        tasks.append(download_file_from_url(scene.bgMusic.musicPath))
+
+    # Run all downloads concurrently
+    downloaded_files = await asyncio.gather(*tasks)
+    return downloaded_files
+
 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.
@@ -218,13 +238,14 @@ async def generate_story_audios(story: StoryCreationDTO, base_output: str):
         # os.remove(prosody_file_title)
 
         for scene in chapter.scenes:
+            await download_scene_files(scene)
             scene_dir = chapter_dir / scene.sceneId
             scene_dir.mkdir(exist_ok=True)
 
             # --- Sentences audio ---
             for sentence in scene.sentences:
                 # Download the prosody reference audio from Supabase
-                prosody_file = await download_file_from_url(sentence.prosodyReference)
+                prosody_file = download_cache[sentence.prosodyReference]
                 sentence_save_path = scene_dir / f"{sentence.sentenceId}.wav"
                 tagged_text = generate_tagged_text(
                     sentence.sentence,