Update app.py

app.py

@@ -215,18 +215,18 @@ with gr.Blocks(css=css) as demo:
 
     # Event handling
     generation_event = run_button.click(
-        fn=infer,
-        inputs=[enhanced_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs=[result, seed]
-    ).then(
-        fn=append_to_history,
-        inputs=[result, enhanced_prompt, seed, width, height, guidance_scale, num_inference_steps, history_state],
-        outputs=history_state
-    ).then(
-        fn=create_history_html,
-        inputs=history_state,
-        outputs=history_display
-    )
+    fn=infer,
+    inputs=[enhanced_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+    outputs=[result, seed]  # Image is OUTPUT here
+).then(
+    fn=append_to_history,
+    inputs=[result, enhanced_prompt, seed, width, height, guidance_scale, num_inference_steps, history_state],  # Image becomes INPUT here
+    outputs=history_state
+).then(
+    fn=create_history_html,
+    inputs=history_state,
+    outputs=history_display
+)
 
     enhanced_prompt.submit(
         fn=infer,