Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -1,6 +1,12 @@
 import gradio as gr
 from PIL import Image
 
+
+SER_SKETCH=None
+
+def get_img_from_sketchpad(img):
+    return Image.fromarray(img["composite"])
+
 def ui(controller):#get_first_result,get_second_result): #controller):
     with gr.Blocks() as ui:
         with gr.Row():
@@ -36,9 +42,10 @@ 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.input(get_img_from_sketchpad, outputs=SER_SKETCH, inputs=sketch, show_progress="hidden")
+        
         improve_sketch.click(fn=controller.get_first_result,
-                             inputs=[Image.fromarray(sketch['composite']), first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],
+                             inputs=[SER_SKETCH), first_prompt, first_negative_prompt], #[sketch, first_prompt, first_negative_prompt],
                              outputs=improved_sketch_view)
 
         result_button.click(fn=controller.get_second_result,
@@ -46,7 +53,7 @@ def ui(controller):#get_first_result,get_second_result): #controller):
                              outputs=result)
 
         firts_prompt_helper.click(fn=controller.get_help_w_prompt,
-                                 inputs=[Image.fromarray(sketch['composite'])], #[sketch],
+                                 inputs=[SER_SKETCH], #[sketch],
                                  outputs=first_prompt)
 
         second_prompt_helper.click(fn=controller.get_help_w_prompt,