Update app.py

app.py

@@ -321,14 +321,20 @@ async def generate_story_audios_async(story: StoryCreationDTO, base_output: str,
     story_dir = Path(base_output) / story.storyId
     story_dir.mkdir(parents=True, exist_ok=True)
 
-    print("generating story ...")
+    print(f"[INFO] Generating story '{story.storyId}' in {story_dir}")
+
     # Semaphore ensures we don't overload GPU
     gpu_semaphore = asyncio.Semaphore(max_concurrent_gpu)
 
     async def process_sentence(chapter_dir: Path, scene: SceneDto, sentence: SentenceDto):
+        print(f"[INFO] Starting sentence '{sentence.sentenceId}' in scene '{scene.sceneId}'")
         async with gpu_semaphore:
+            print(f"[GPU] Acquired GPU for sentence '{sentence.sentenceId}'")
             # Get prosody file from cache
-            prosody_file = download_cache[sentence.prosodyReference]
+            prosody_file = download_cache.get(sentence.prosodyReference)
+            if not prosody_file:
+                print(f"[WARN] Prosody file for '{sentence.sentenceId}' not found in cache")
+                return None
 
             sentence_save_path = chapter_dir / scene.sceneId / f"{sentence.sentenceId}.wav"
             Path(sentence_save_path).parent.mkdir(parents=True, exist_ok=True)
@@ -348,7 +354,7 @@ async def generate_story_audios_async(story: StoryCreationDTO, base_output: str,
                 prosody_file,
                 str(sentence_save_path)
             )
-
+            print(f"[DONE] Generated audio for sentence '{sentence.sentenceId}' -> {generated_path}")
             return generated_path
 
     # Prepare tasks for chapters
@@ -356,9 +362,14 @@ async def generate_story_audios_async(story: StoryCreationDTO, base_output: str,
     for chapter in story.chapters:
         chapter_dir = story_dir / chapter.chapterId
         chapter_dir.mkdir(exist_ok=True)
+        print(f"[INFO] Processing chapter '{chapter.chapterId}'")
 
         # --- Chapter title ---
-        title_prosody = download_cache[chapter.title.prosodyReference]
+        title_prosody = download_cache.get(chapter.title.prosodyReference)
+        if not title_prosody:
+            print(f"[WARN] Prosody file for chapter title '{chapter.chapterId}' not found in cache")
+            continue
+
         title_save_path = chapter_dir / "title.wav"
         tagged_title = generate_tagged_text(
             chapter.title.sentence,
@@ -366,7 +377,7 @@ async def generate_story_audios_async(story: StoryCreationDTO, base_output: str,
             chapter.title.intensity
         )
 
-        # Run title generation immediately
+        print(f"[GPU] Generating title audio for chapter '{chapter.chapterId}'")
         loop = asyncio.get_event_loop()
         await loop.run_in_executor(
             None,
@@ -375,21 +386,29 @@ async def generate_story_audios_async(story: StoryCreationDTO, base_output: str,
             title_prosody,
             str(title_save_path)
         )
+        print(f"[DONE] Generated title audio for chapter '{chapter.chapterId}' -> {title_save_path}")
 
         # --- Scenes ---
         scene_tasks = []
         for scene in chapter.scenes:
-            # Download scene files (prosody, sfx, bg) concurrently
+            print(f"[INFO] Downloading files for scene '{scene.sceneId}'")
             await download_scene_files(scene)
 
             for sentence in scene.sentences:
                 scene_tasks.append(process_sentence(chapter_dir, scene, sentence))
 
-        # Run all sentences in this chapter concurrently but limited by GPU semaphore
-        chapter_tasks.append(asyncio.gather(*scene_tasks))
+        if scene_tasks:
+            print(f"[INFO] Running {len(scene_tasks)} sentences for chapter '{chapter.chapterId}' concurrently")
+            chapter_tasks.append(asyncio.gather(*scene_tasks))
+        else:
+            print(f"[WARN] No sentences found in chapter '{chapter.chapterId}'")
 
     # Wait for all chapters to complete
-    await asyncio.gather(*chapter_tasks)
+    if chapter_tasks:
+        await asyncio.gather(*chapter_tasks)
+        print(f"[INFO] Completed generating all chapters for story '{story.storyId}'")
+    else:
+        print(f"[WARN] No chapters/tasks to process for story '{story.storyId}'")
 
 #_______________ Concatenating the generated audios to make the final story (post-processing)_______________________