Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -18,12 +18,14 @@ def ui(controller):#get_first_result,get_second_result): #controller):
             with gr.Column():
                 first_prompt = gr.Textbox(label="Prompt", lines=3)
                 first_negative_prompt = gr.Textbox(label="Negative prompt", lines=3, value="sketch, lowres, error, extra digit, fewer digits, cropped, worst quality,low quality, normal quality, jpeg artifacts, blurry")
+                firts_prompt_helper = gr.Button(value="Prompt Helper", variant="primary")
                 #controlnet_scale = gr.Slider(minimum=0.5, maximum=1.25, value=0.5, step=0.01, label="Contr")
                 improve_sketch = gr.Button(value="Improve Sketch", variant="primary")
 
             with gr.Column():
                 second_prompt = gr.Textbox(label="Prompt", lines=3)
                 second_negative_prompt = gr.Textbox(label="Negative prompt", lines=3, value="disfigured, extra digit, fewer digits, cropped, worst quality, low quality")
+                second_prompt_helper = gr.Button(value="Prompt Helper", variant="primary")
                 result_button = gr.Button(value="Paint It", variant="primary")
                 
 
@@ -35,5 +37,14 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                              inputs=[improved_sketch_view, second_prompt, second_negative_prompt],
                              outputs=result)
 
+        firts_prompt_helper.click(fn=controller.get_help_w_prompt,
+                                 inputs=[sketch],
+                                 outputs=first_prompt)
+
+        second_prompt_helper.click(fn=controller.get_help_w_prompt,
+                                 inputs=[improved_sketch_view],
+                                 outputs=first_prompt)
+        
+
 
     ui.queue().launch(debug=True)