First

ui/story_interface.py

@@ -3,6 +3,7 @@ import time
 from utils.story_management import (
     generate_direct_comic,
     extract_comic_scenes,
+    load_narration_from_file
 )
 from config import IMAGE_STYLES, IMAGE_STYLE_INFO, AGE_GROUPS
 from datetime import datetime
@@ -17,6 +18,7 @@ def log_execution(func):
         end_time = time.time()
         end_str = datetime.fromtimestamp(end_time).strftime('%Y-%m-%d %H:%M:%S')
         duration = end_time - start_time
+
         
         return result
     return wrapper
@@ -144,24 +146,24 @@ def create_quick_comic_tab() -> None:
             """Display a loading message while story scenes are being generated."""
             return "🔄 Generating your story scenes... Please wait..."
 
-        # def generate_comic_with_length(user_prompt, comic_style, enrich_prompt, age_group, image_quality):
-        #     """Wrapper that handles the fixed num_scenes value while passing the age group and image quality."""
-        #     comic_image, save_path_display, status_display, narration = generate_direct_comic(
-        #         user_prompt,
-        #         comic_style,
-        #         12,
-        #         enrich_prompt,
-        #         3,
-        #         age_group,
-        #     )
-        #
-        #     if narration and narration.strip():
-        #         narration_formatted = f""
-        #         narration_update = gr.update(visible=True, value=narration_formatted)
-        #     else:
-        #         narration_update = gr.update(visible=True, value="")
-        #
-        #     return comic_image, save_path_display, status_display, narration_update
+        def generate_comic_with_length(user_prompt, comic_style, enrich_prompt, age_group, image_quality):
+            """Wrapper that handles the fixed num_scenes value while passing the age group and image quality."""
+            comic_image, save_path_display, status_display, narration = generate_direct_comic(
+                user_prompt,
+                comic_style,
+                12,
+                enrich_prompt,
+                3,
+                age_group,
+            )
+
+            if narration and narration.strip():
+                narration_formatted = f""
+                narration_update = gr.update(visible=True, value=narration_formatted)
+            else:
+                narration_update = gr.update(visible=True, value="")
+
+            return comic_image, save_path_display, status_display, narration_update
 
         def init_scene_viewer(comic_path: str | None) -> tuple:
             """Initialize the scene viewer with extracted scenes from the comic image."""