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):
@@ -98,7 +98,13 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
 
                 improved_prompt = gr.Textbox(label="Prompt Mejorado", interactive=False)
                 improve_btn = gr.Button("Mejora mi prompt")
-                improve_btn.click(fn=lambda prompt: improve_prompt(prompt), inputs=[prompt], outputs=[improved_prompt, prompt])
+
+                def improve_prompt_wrapper(prompt):
+                    improved_text = improve_prompt(prompt)
+                    return prompt, improved_text
+
+                improve_btn.click(fn=improve_prompt_wrapper, inputs=[prompt], outputs=[prompt, improved_prompt])
+                
                 reset_btn = gr.Button("Reset")
                 reset_btn.click(fn=lambda: [prompt.update(""), improved_prompt.update("")], inputs=None, outputs=[prompt, improved_prompt])
                 
@@ -112,6 +118,38 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
                     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])
     
-                btn = gr.Button("Generar")
-                btn.click(fn=gen, inputs=[prompt, basemodel_choice, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model_choice, process_lora], outputs=[output_res])
+                generating = gr.StatusTracker(label="Generando", status=False)
+                btn = gr.Button("Generar", variant="primary", status_tracker=generating)
+                btn.click(
+                    fn=gen,
+                    inputs=[
+                        prompt,
+                        basemodel_choice,
+                        width,
+                        height,
+                        scales,
+                        steps,
+                        seed,
+                        upscale_factor,
+                        process_upscale,
+                        lora_model_choice,
+                        process_lora,
+                    ],
+                    outputs=[output_res],
+                    error_handler=error_handler,
+                )
+                
+                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()