Update app.py

app.py

@@ -89,6 +89,27 @@ def generate_pdf(images, explanations, title="AI-Generated Story Scenes"):
 def reset_fields():
     return "", "", "", "", "", "Fantasy"
 
+def recreate_scene_handler(
+    num_scenes, theme, char_count, character_names, dialogue,
+    dialogue_speaker, char_styles, char_moods, bg_style,
+    images, summaries, explanations, scene_number_to_recreate):
+
+    index = int(scene_number_to_recreate) - 1
+    if index < 0 or index >= len(images):
+        return (
+            None, "", "❌ Invalid input. You have not generated Scene {} yet.".format(scene_number_to_recreate),
+            images, summaries, explanations,
+            f"⚠️ Scene {scene_number_to_recreate} is not available. Generate it first.", gr.update(visible=False)
+        )
+
+    return generate_scene(
+        num_scenes, theme, char_count, character_names, dialogue,
+        dialogue_speaker, char_styles, char_moods, bg_style,
+        images, summaries, explanations,
+        recreate_mode=True, current_index=index
+    )
+
+
 
 # -------------------- Core Logic --------------------
 
@@ -97,13 +118,11 @@ def generate_scene(num_scenes, theme, char_count, character_names, dialogue,
     images, summaries, explanations, recreate_mode=False, current_index=0):
 
     prompt = (
-        f"Scene {current_index + 1} set in {theme}. Number of characters: {char_count}. "
-        f"Characters: {character_names}. Outfits: {char_styles}. Mood: {char_moods}. "
-        f"Style: {bg_style}."
+        f"A {bg_style}-style illustration for Scene {current_index + 1} with {char_count} characters in a '{theme}' setting. "
+        f"Characters: {character_names}. They are dressed as: {char_styles}. Current mood: {char_moods}. "
     )
-
     if dialogue:
-        prompt += f' The character "{dialogue_speaker}" says: "{dialogue}". '
+        prompt += f'The character "{dialogue_speaker}" says: "{dialogue}". Display this in a speech bubble.'
         prompt += f' Please visualize this as a speech bubble above {dialogue_speaker}, like in a cartoon.'
 
     image = generate_image_from_text(prompt)
@@ -336,6 +355,7 @@ with gr.Blocks(
     with gr.Row():
         generate_btn = gr.Button("➕ Generate The Scene", elem_id="generate-btn")
         ai_coauthor_btn = gr.Button("🤖 Let AI Write This One", elem_id="ai-write-btn")
+        recreate_scene_index = gr.Number(label="🔢 Scene Number to Recreate", precision=0, value=1)
         recreate_btn = gr.Button("🔄 Recreate The Scene", elem_id="recreate-btn")
         reset_btn = gr.Button("\n⏭️ Reset", elem_id="reset-btn")
 
@@ -376,12 +396,12 @@ with gr.Blocks(
 
 
     recreate_btn.click(
-        fn=generate_scene,
+        fn=recreate_scene_handler,
         inputs=[
             scene_total, theme, char_count, character_names, dialogue,
             dialogue_speaker, char_styles, char_moods, bg_style,
             scene_images, scene_summaries, scene_explanations,
-            recreate_mode, current_scene_index
+            recreate_scene_index
         ],
         outputs=[
             image_output, summary_output, explanation_output,
@@ -390,6 +410,7 @@ with gr.Blocks(
         ]
     )
 
+
    
     ai_coauthor_btn.click(
         fn=ai_write_scene,