Upload app.py

app.py

@@ -127,6 +127,13 @@ def undo_edit(image_history):
     else:
         return None, [], "No previous version available"
 
+# Function to set output image as input for continuous editing
+def continue_editing(output_image):
+    if output_image is not None:
+        return output_image, "Ready to continue editing the current image"
+    else:
+        return None, "No edited image available to continue editing"
+
 # Create Gradio UI
 def create_ui():
     with gr.Blocks(title="Gemini Image Editor") as app:
@@ -156,6 +163,7 @@ def create_ui():
                 with gr.Row():
                     edit_btn = gr.Button("Edit Image")
                     undo_btn = gr.Button("Undo Last Edit")
+                    continue_btn = gr.Button("Continue Editing")
                 
             with gr.Column():
                 output_image = gr.Image(type="pil", label="Edited Image")
@@ -173,6 +181,13 @@ def create_ui():
             inputs=[image_history],
             outputs=[output_image, image_history, status]
         )
+        
+        # Add handler for continue editing button
+        continue_btn.click(
+            fn=continue_editing,
+            inputs=[output_image],
+            outputs=[input_image, status]
+        )
     
     return app