Update app.py

app.py

@@ -94,14 +94,10 @@ def reset_fields():
 
 def generate_scene(num_scenes, theme, char_count, character_names, dialogue,
     dialogue_speaker, char_styles, char_moods, bg_style,
-    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)
+    images, summaries, explanations, recreate_mode=False, current_index=0):
 
     prompt = (
-        f"Scene {len(images)+1} set in {theme}. Number of characters: {char_count}. "
+        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}."
     )
@@ -114,11 +110,19 @@ def generate_scene(num_scenes, theme, char_count, character_names, dialogue,
     summary = summarize_scene(prompt)
     explanation = explain_scene(image) if image else "Explanation unavailable."
 
-    images.append(image)
-    summaries.append(summary)
-    explanations.append(explanation)
-
-    status = f"✅ Scene {len(images)} of {int(num_scenes)} generated."
+    if recreate_mode:
+        images[current_index] = image
+        summaries[current_index] = summary
+        explanations[current_index] = explanation
+    else:
+        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)
+        images.append(image)
+        summaries.append(summary)
+        explanations.append(explanation)
+
+    status = f"✅ Scene {current_index + 1} {'recreated' if recreate_mode else 'generated'}."
     done_visible = len(images) == int(num_scenes)
 
     return image, summary, explanation, images, summaries, explanations, status, gr.update(visible=done_visible)
@@ -358,7 +362,8 @@ with gr.Blocks(
         inputs=[
             scene_total, theme, char_count, character_names, dialogue,
             dialogue_speaker, char_styles, char_moods, bg_style,
-            scene_images, scene_summaries, scene_explanations
+            scene_images, scene_summaries, scene_explanations,
+            gr.State(False), current_scene_index
         ],
         outputs=[
             image_output, summary_output, explanation_output,
@@ -367,12 +372,14 @@ with gr.Blocks(
         ]
     )
 
+
     recreate_btn.click(
         fn=generate_scene,
         inputs=[
             scene_total, theme, char_count, character_names, dialogue,
             dialogue_speaker, char_styles, char_moods, bg_style,
-            scene_images, scene_summaries, scene_explanations
+            scene_images, scene_summaries, scene_explanations,
+            recreate_mode, current_scene_index
         ],
         outputs=[
             image_output, summary_output, explanation_output,
@@ -380,6 +387,7 @@ with gr.Blocks(
             status, done_btn
         ]
     )
+
    
     ai_coauthor_btn.click(
         fn=ai_write_scene,