Fix disable enhance

app.py

@@ -233,7 +233,7 @@ with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
                 visible=False,
                 interactive=False,
             )
-
+        
         with gr.Accordion("Advanced Settings", open=False):
             negative_prompt = gr.Text(
                 label="Negative prompt",
@@ -242,8 +242,6 @@ with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
                 visible=True,
             )
             
-            # Removed checkbox and enriched prompt display from here
-
             with gr.Tabs() as resolution_tabs:
                 with gr.TabItem("Preset Resolutions"):
                     resolution_dropdown = gr.Dropdown(
@@ -300,7 +298,18 @@ with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
 
         # Examples should explicitly target only the prompt input
         max_length = 180
-        examples_component = gr.Examples(examples=examples, inputs=prompt, example_labels=[ex[:max_length] + "..." if len(ex) > max_length else ex for ex in examples])
+        
+        # Function to handle example clicks - sets prompt and disables enrichment
+        def set_example(example):
+            return example, False
+            
+        gr.Examples(
+            examples=examples,
+            inputs=prompt,
+            outputs=[prompt, use_prompt_enrichment],
+            fn=set_example,
+            example_labels=[ex[:max_length] + "..." if len(ex) > max_length else ex for ex in examples]
+        )
     
     # Update width and height when resolution dropdown changes
     resolution_dropdown.change(
@@ -309,13 +318,6 @@ with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
         outputs=[width, height]
     )
     
-    # Disable enrichment when an example is clicked
-    examples_component.click(
-        fn=lambda: False,
-        inputs=[],
-        outputs=[use_prompt_enrichment]
-    )
-    
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
@@ -330,7 +332,6 @@ with gr.Blocks(css=css, theme="ParityError/Interstellar") as demo:
             num_inference_steps,
             use_prompt_enrichment,
         ],
-        # Outputs now include updates for the enriched prompt display components
         outputs=[result, seed, enriched_prompt_display, enriched_prompt_text, enrichment_error],
     )