Update app.py

app.py

@@ -82,7 +82,7 @@ css = """
 #col-container{ margin: 0 auto; max-width: 1024px;}
 """
 
-with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
+with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
     with gr.Column(elem_id="col-container"):
         with gr.Row():
             with gr.Column(scale=3):
@@ -118,8 +118,13 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:with gr.Blocks(css=cs
                     reset_advanced = gr.Button("Reset")
                     reset_advanced.click(fn=lambda: [width.update(1280), height.update(768), scales.update(10), steps.update(20), seed.update(-1)], inputs=None, outputs=[width, height, scales, steps, seed])
     
+                prompt_state = gr.State("")
                 generating = gr.StatusTracker(label="Generando", status=False)
                 btn = gr.Button("Generar", variant="primary", status_tracker=generating)
+
+                def reset_prompt():
+                    return "", False
+
                 btn.click(
                     fn=gen,
                     inputs=[
@@ -137,19 +142,8 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:with gr.Blocks(css=cs
                     ],
                     outputs=[output_res],
                     error_handler=error_handler,
+                    after_fn=reset_prompt,
+                    outputs_after_fn=[prompt_state, generating],
                 )
                 
-                def check_prompt_change(prompt, previous_prompt):
-                    if prompt != previous_prompt:
-                        generating.update(status=True)
-                    return previous_prompt
-                
-                previous_prompt = gr.State("")
-                btn.click(
-                    fn=check_prompt_change,
-                    inputs=[prompt, previous_prompt],
-                    outputs=[previous_prompt],
-                    before_fn=True,
-                )
-    
     demo.launch()