Update app.py

app.py

@@ -119,16 +119,17 @@ def create_ui():
         
         with gr.Row():
             prompt_input = gr.Textbox(label="Enter a Prompt", placeholder="Describe the image you want to generate", lines=3)
+            enhance_checkbox = gr.Checkbox(label="Enhance Prompt", value=True)  # Checkbox for enhancing prompt
             generate_button = gr.Button("Generate Image")
         
         with gr.Row():
             output_image = gr.Image(label="Generated Image")
         
         # Link the button to the function
-        generate_button.click(fn=generate_image, inputs=[prompt_input, True], outputs=output_image)
+        generate_button.click(fn=generate_image, inputs=[prompt_input, enhance_checkbox], outputs=output_image)
     
     return ui
 
 # Run the interface
 if __name__ == "__main__":
-    create_ui().launch()
+    create_ui().launch()