improve

app.py

@@ -14,7 +14,7 @@ from diffusers import (
     EulerDiscreteScheduler  # <-- Added import
 )
 from share_btn import community_icon_html, loading_icon_html, share_js
-# from gallery_history import fetch_gallery_history, show_gallery_history
+from gallery_history import fetch_gallery_history, show_gallery_history
 from illusion_style import css
 
 BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
@@ -189,18 +189,16 @@ with gr.Blocks(css=css) as app:
         inference,
         inputs=[control_image, prompt, negative_prompt, guidance_scale, controlnet_conditioning_scale, control_start, control_end, strength, seed, sampler],
         outputs=[result_image, share_group, used_seed]
+    ).then(
+       fn=fetch_gallery_history, inputs=[prompt, result_image], outputs=history, queue=False
     )
-    # ).then(
-    #    fn=fetch_gallery_history, inputs=[prompt, result_image], outputs=history, queue=False
-    # )
     run_btn.click(
         inference,
         inputs=[control_image, prompt, negative_prompt, guidance_scale, controlnet_conditioning_scale, control_start, control_end, strength, seed, sampler],
         outputs=[result_image, share_group, used_seed]
+    ).then(
+        fn=fetch_gallery_history, inputs=[prompt, result_image], outputs=history, queue=False
     )
-    # ).then(
-    #     fn=fetch_gallery_history, inputs=[prompt, result_image], outputs=history, queue=False
-    # )
     share_button.click(None, [], [], _js=share_js)
 app.queue(max_size=20)