Update app.py

app.py

@@ -1,5 +1,4 @@
-from fastapi import FastAPI, BackgroundTasks, UploadFile, File, Form
-from fastapi.responses import FileResponse
+from fastapi import FastAPI, BackgroundTasks
 from pydantic import BaseModel
 from typing import List
 from pathlib import Path
@@ -32,7 +31,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # Supabase client
 # -----------------------------
 SUPABASE_URL = "https://kvlxvhdgacktsgykyckm.supabase.co/"
-SUPABASE_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6Imt2bHh2aGRnYWNrdHNneWt5Y2ttIiwicm9sZSI6InNlcnZpY2Vfcm9sZSIsImlhdCI6MTc3MTk2MTQ5MSwiZXhwIjoyMDg3NTM3NDkxfQ.tzfHcbzwzctHDDDp3vk4JGz30ajN2szncAV-1wK7_pM"
+SUPABASE_KEY = "YOUR_SUPABASE_KEY_HERE"
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 
 # -----------------------------
@@ -43,13 +42,11 @@ download_cache = {}  # URL -> local path
 # -----------------------------
 # Helper to get cached file (downloads if missing)
 # -----------------------------
-async def get_cached_file(url: str, subfolder: str) -> str | None:
+async def get_cached_file(url: str, subfolder: str, skip_on_startup: bool = False) -> str | None:
     """
     Returns local cached path for URL.
     Downloads and stores in subfolder if missing.
-    If download fails:
-        - Raise error for prosody or bg_music
-        - Skip (return None) for sfx
+    skip_on_startup: if True, don't raise errors even for prosody/bg_music (used during startup).
     """
     if url in download_cache:
         return download_cache[url]
@@ -76,11 +73,11 @@ async def get_cached_file(url: str, subfolder: str) -> str | None:
         return local_path
 
     except Exception as e:
-        if subfolder == "sfx":
-            print(f"Warning: Failed to download SFX {url}, skipping: {e}")
+        if subfolder == "sfx" or skip_on_startup:
+            print(f"Warning: Failed to download {subfolder} file {url}, skipping: {e}")
             return None
         else:
-            # Raise error for essential files
+            # Raise error if essential and not in startup
             raise RuntimeError(f"Failed to download {subfolder} file {url}: {e}") from e
 
 # -----------------------------
@@ -97,9 +94,6 @@ def list_all_files(bucket_name: str):
         files.append((f["name"], url))
     return files
 
-async def download_to_cache(url: str, subfolder: str):
-    await get_cached_file(url, subfolder)
-
 async def preload_all_assets():
     print("Starting Supabase asset preloading...")
     tasks = []
@@ -113,7 +107,8 @@ async def preload_all_assets():
     for bucket_name, subfolder in buckets.items():
         files = list_all_files(bucket_name)
         for _, url in files:
-            tasks.append(download_to_cache(url, subfolder))
+            # skip_on_startup=True prevents crash at startup
+            tasks.append(get_cached_file(url, subfolder, skip_on_startup=True))
 
     await asyncio.gather(*tasks)
     print(f"Preloading completed. {len(download_cache)} files cached on disk.")
@@ -251,6 +246,9 @@ async def generate_story_audios(story: StoryCreationDTO, base_output: str):
         chapter_dir.mkdir(exist_ok=True)
 
         prosody_file_title = await get_cached_file(chapter.title.prosodyReference, "prosody")
+        if prosody_file_title is None:
+            raise RuntimeError(f"Missing prosody file for chapter title: {chapter.title.prosodyReference}")
+
         title_save_path = chapter_dir / "title.wav"
         tagged_text_title = generate_tagged_text(
             chapter.title.sentence,
@@ -265,6 +263,9 @@ async def generate_story_audios(story: StoryCreationDTO, base_output: str):
 
             for sentence in scene.sentences:
                 prosody_file = await get_cached_file(sentence.prosodyReference, "prosody")
+                if prosody_file is None:
+                    raise RuntimeError(f"Missing prosody file for sentence: {sentence.sentenceId}")
+
                 sentence_save_path = scene_dir / f"{sentence.sentenceId}.wav"
                 tagged_text = generate_tagged_text(
                     sentence.sentence,
@@ -309,11 +310,14 @@ async def concat_story_audio(story: StoryCreationDTO, base_output: str, final_pa
 
             if scene.location.path:
                 sfx_file = await get_cached_file(scene.location.path, "sfx")
-                sfx_file_wav = ensure_wav(sfx_file)
-                scene_audio = scene_audio.overlay(AudioSegment.from_wav(sfx_file_wav))
+                if sfx_file:
+                    sfx_file_wav = ensure_wav(sfx_file)
+                    scene_audio = scene_audio.overlay(AudioSegment.from_wav(sfx_file_wav))
 
             if scene.bgMusic and scene.bgMusic.musicPath:
                 bg_file = await get_cached_file(scene.bgMusic.musicPath, "bg_music")
+                if bg_file is None:
+                    raise RuntimeError(f"Missing background music file: {scene.bgMusic.musicPath}")
                 bg_file_wav = ensure_wav(bg_file)
                 bg_audio = AudioSegment.from_file(bg_file_wav)
                 bg_audio = bg_audio - (1 - scene.bgMusic.volume) * 30