Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -3,27 +3,15 @@ from PIL import Image, ImageOps
 import numpy as np
 
 
-#SER_SKETCH=Image.fromarray(np.zeros((1024,1024)))
-
-def get_img_from_sketchpad(img):
-    #global SER_SKETCH
-    return Image.fromarray(img["composite"]) #ImageOps.invert(Image.fromarray((img["composite"])))#,'L')
-
-#def set_ser_sketch(img):
-    #global SER_SKETCH
-    #SER_SKETCH=img
-
-def ui(controller):#get_first_result,get_second_result): #controller):
+def ui(controller):
     with gr.Blocks() as ui:
         with gr.Row():
             with gr.Column():
                 sketch=gr.Sketchpad(sources = 'upload', label='Sketch', type='pil', image_mode="RGB", brush=gr.Brush()) #gr.Image(sources = 'upload', label='Sketch', type = 'pil')
-                #ser_image=gr.Image(type="pil")#, visible=False)
-                
+                                
             with gr.Column():
-                improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")        
-
-        #with gr.Row():
+                improved_sketch_view = gr.Image(type="pil", label="Improved Sketch") 
+        
             with gr.Column():
                 result=gr.Image(type="pil", label="Colored Image")
                 
@@ -32,8 +20,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
             with gr.Column():
                 first_prompt = gr.Textbox(label="Prompt to Improve Sketch", 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")
+                firts_prompt_helper = gr.Button(value="Prompt Helper", variant="primary")                
                 improve_sketch = gr.Button(value="Improve Sketch", variant="primary")
 
             with gr.Column():
@@ -52,11 +39,9 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                         inputs=[sketch, first_prompt, second_prompt],
                         label='Examples. Be concrete as much as possible.')
                 
-        #sketch.change(fn=get_img_from_sketchpad, inputs=sketch, outputs=ser_image, show_progress="full")
-
-              
+                      
         improve_sketch.click(fn=controller.get_first_result,
-                             inputs=[sketch, first_prompt, first_negative_prompt], #[ser_image, first_prompt, first_negative_prompt],
+                             inputs=[sketch, first_prompt, first_negative_prompt], 
                              outputs=improved_sketch_view)
 
         result_button.click(fn=controller.get_second_result,
@@ -64,13 +49,11 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                              outputs=result)
 
         firts_prompt_helper.click(fn=controller.get_help_w_prompt,
-                                 inputs=[sketch], #[ser_image], #
+                                 inputs=[sketch], 
                                  outputs=first_prompt)
 
         second_prompt_helper.click(fn=controller.get_help_w_prompt,
                                  inputs=[improved_sketch_view],
-                                 outputs=second_prompt)
-        
-
+                                 outputs=second_prompt) 
 
     ui.queue().launch(debug=True)