Update app.py

app.py

@@ -96,9 +96,19 @@ def generate_pdf(images, explanations):
         doc.build(story)
         return tmp.name
 
-# -------------------- Scene Management --------------------
+# -------------------- Core Logic --------------------
+
+def generate_scene(num_scenes, theme, char_count, character_names, dialogue,
+                   images, summaries, explanations):
+    
+    if len(images) >= int(num_scenes):
+        return gr.update(), gr.update(), gr.update(), images, summaries, explanations, \
+               f"✅ All {num_scenes} scenes have been generated.", gr.update(visible=True)
+
+    prompt = f"Scene {len(images)+1} set in {theme}. Number of characters: {char_count}. Characters: {character_names}."
+    if dialogue:
+        prompt += f' Include this dialogue: "{dialogue}".'
 
-def generate_scene(prompt, images, summaries, explanations):
     image = generate_image_from_text(prompt)
     summary = summarize_scene(prompt)
     explanation = explain_scene(image) if image else "Explanation unavailable."
@@ -107,91 +117,84 @@ def generate_scene(prompt, images, summaries, explanations):
     summaries.append(summary)
     explanations.append(explanation)
 
-    return images, summaries, explanations
-
-def edit_scene(index, theme, char_count, character_names, dialogue, images, summaries, explanations):
-    prompt = f"Scene {index+1} set in {theme}. Characters: {character_names}. Dialogue: '{dialogue}'"
-    image = generate_image_from_text(prompt)
-    summary = summarize_scene(prompt)
-    explanation = explain_scene(image)
-
-    images[index] = image
-    summaries[index] = summary
-    explanations[index] = explanation
-
-    return images, summaries, explanations
+    status = f"✅ Scene {len(images)} of {int(num_scenes)} generated."
+    done_visible = len(images) == int(num_scenes)
 
-def delete_scene(index, images, summaries, explanations):
-    if 0 <= index < len(images):
-        del images[index]
-        del summaries[index]
-        del explanations[index]
-    return images, summaries, explanations
+    return image, summary, explanation, images, summaries, explanations, status, gr.update(visible=done_visible)
 
 def finalize_story(images, explanations):
     if not images or not explanations:
         return None, None
+
     pdf = generate_pdf(images, explanations)
+
     with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as txt:
         for i, exp in enumerate(explanations):
             txt.write(f"Scene {i+1}:\n{exp}\n\n")
         txt_path = txt.name
+
     return txt_path, pdf
 
-# -------------------- UI Logic --------------------
+# -------------------- UI --------------------
 
-with gr.Blocks(title="AI Scene Builder with Preview & Editing") as demo:
-    gr.Markdown("## 🎬 AI Story Scene Generator with Editing and Previews")
+with gr.Blocks(title="Scene-by-Scene Story Generator") as demo:
+    gr.Markdown("## 🎬 AI Scene-by-Scene Story Creator")
+    gr.Markdown("Describe your story one scene at a time, with AI-generated images, summaries, and explanations.")
 
-    scene_count = gr.Number(label="Number of Scenes", precision=0)
-    theme = gr.Textbox(label="Global Theme")
+    with gr.Accordion("🧩 Story Setup", open=True):
+        scene_total = gr.Number(label="🔢 Number of Scenes", precision=0, value=3)
+        theme = gr.Textbox(label="🌍 Global Theme", placeholder="e.g. A magical forest full of glowing creatures")
 
-    char_count = gr.Number(label="Number of Characters", precision=0)
-    character_names = gr.Textbox(label="Character Names")
-    dialogue = gr.Textbox(label="Dialogue")
+    gr.Markdown("### ✨ Describe Your Next Scene")
 
-    generate_btn = gr.Button("Generate Scene")
-    index_to_edit = gr.Number(label="Scene Index to Edit/Delete (0-based)", value=0)
-    edit_btn = gr.Button("✏️ Edit Scene")
-    delete_btn = gr.Button("🗑️ Delete Scene")
+    with gr.Row():
+        char_count = gr.Number(label="👥 Number of Characters", precision=0, value=2)
+        character_names = gr.Textbox(label="🧙‍♂️ Character Names", placeholder="e.g. Elora, Bramble the Bear")
+        dialogue = gr.Textbox(label="💬 Dialogue (optional)", placeholder="e.g. 'Protect the forest!'")
 
-    thumbnail_gallery = gr.Gallery(label="Scene Previews").style(grid=[4], height="auto")
+    generate_btn = gr.Button("➕ Generate This Scene")
+    status = gr.Markdown()
 
-    finalize_btn = gr.Button("✅ Finalize & Export")
-    txt_file = gr.File()
-    pdf_file = gr.File()
+    image_output = gr.Image(label="🖼️ Scene Image", type="pil")
+    summary_output = gr.Markdown(label="📝 Scene Summary")
+    explanation_output = gr.Textbox(label="📖 Scene Explanation", lines=6)
 
+    with gr.Row():
+        tts_btn = gr.Button("🔊 Read Aloud")
+        tts_audio = gr.Audio(label="Audio", autoplay=False)
+
+    done_btn = gr.Button("✅ Finalize Story & Export", visible=False)
+    txt_file = gr.File(label="📄 Explanations (.txt)")
+    pdf_file = gr.File(label="📘 Scene PDF")
+
+    # Persistent session states
     scene_images = gr.State([])
     scene_explanations = gr.State([])
     scene_summaries = gr.State([])
 
-    def build_prompt(index, theme, character_names, dialogue):
-        return f"Scene {index+1} set in {theme}. Characters: {character_names}. Dialogue: '{dialogue}'"
-
     generate_btn.click(
-        lambda sc, th, cc, names, dlg, imgs, sums, expls: generate_scene(
-            build_prompt(len(imgs), th, names, dlg), imgs, sums, expls
-        ),
-        inputs=[scene_count, theme, char_count, character_names, dialogue, scene_images, scene_summaries, scene_explanations],
-        outputs=[scene_images, scene_summaries, scene_explanations, thumbnail_gallery]
-    )
-
-    edit_btn.click(
-        edit_scene,
-        inputs=[index_to_edit, theme, char_count, character_names, dialogue, scene_images, scene_summaries, scene_explanations],
-        outputs=[scene_images, scene_summaries, scene_explanations, thumbnail_gallery]
+        fn=generate_scene,
+        inputs=[
+            scene_total, theme, char_count, character_names, dialogue,
+            scene_images, scene_summaries, scene_explanations
+        ],
+        outputs=[
+            image_output, summary_output, explanation_output,
+            scene_images, scene_summaries, scene_explanations,
+            status, done_btn
+        ]
     )
 
-    delete_btn.click(
-        delete_scene,
-        inputs=[index_to_edit, scene_images, scene_summaries, scene_explanations],
-        outputs=[scene_images, scene_summaries, scene_explanations, thumbnail_gallery]
-    )
-
-    finalize_btn.click(
-        finalize_story,
+    done_btn.click(
+        fn=finalize_story,
         inputs=[scene_images, scene_explanations],
         outputs=[txt_file, pdf_file]
     )
 
-demo.launch()
+    tts_btn.click(
+        fn=text_to_speech,
+        inputs=[explanation_output],
+        outputs=[tts_audio]
+    )
+
+demo.launch()