fixing clear button

S2I/commons/controller.py

@@ -70,7 +70,7 @@ class Sketch2ImageController():
 
         if type_flag == 'live-sketch':
             img = Image.fromarray(np.array(image["composite"])[:, :, -1])
-        elif type_flag == 'upload':
+        elif type_flag == 'url-sketch':
             img = image["composite"]
 
         img = img.convert("RGB")

app.py

@@ -102,29 +102,6 @@ def get_meta_from_image(input_img, type_image):
     return img_pil
 
 
-# def get_meta_from_image(input_img, type_image):
-#     if input_img is None:
-#         return gr.update(value=None)
-
-#     file_content, _ = read_temp_file(input_img)
-    
-#     # Read the image using Pillow
-#     img = Image.open(io.BytesIO(file_content)).convert("RGB")
-#     img_np = np.array(img)
-
-#     if type_image == 'RGB':
-#         sketch = convert_to_pencil_sketch(img_np)
-#         processed_img = 255 - sketch
-#     elif type_image == 'SKETCH':
-#         processed_img = 255 - img_np
-
-#     # Convert the processed image back to PIL Image
-#     img_pil = Image.fromarray(processed_img.astype('uint8'))
-    
-#     return img_pil
-
-
-
 with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
     gr.HTML(
                 """
@@ -259,7 +236,7 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
                                         label="Type of Image (Color Image or Sketch Image)",
                                         interactive=True)
                                 input_type = gr.Radio(
-                                                choices=["live-sketch", "upload"],
+                                                choices=["live-sketch", "url-sketch"],
                                                 value="live-sketch",
                                                 label="Type Sketch2Image models",
                                                 interactive=True)
@@ -315,15 +292,6 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
     outputs = [result, download_sketch]
     prompt.submit(fn=assign_gpu, inputs=inputs, outputs=outputs, api_name=False)
 
-    # input_image.change(
-    #             fn=get_meta_from_image,
-    #             inputs=[
-    #                 input_image, type_image
-    #             ],
-    #             outputs=[
-    #                 image
-    #             ]
-    #         )
 
     style.change(
         lambda x: controller.styles[x],
@@ -337,16 +305,12 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
         outputs=outputs,
         api_name=False,
     )
-    clear_button.click(fn=clear_session, inputs=[], outputs=[image, result]).then(
-        fn=assign_gpu,
-        inputs=inputs,
-        outputs=outputs,
-        api_name=False,
-    )
+    clear_button.click(fn=clear_session, inputs=[], outputs=[image, result])
     val_r.change(assign_gpu, inputs=inputs, outputs=outputs, queue=False, api_name=False)
     run_button.click(fn=assign_gpu, inputs=inputs, outputs=outputs, api_name=False)
     image.change(assign_gpu, inputs=inputs, outputs=outputs, queue=False, api_name=False)
     url_image.submit(fn=get_meta_from_image, inputs=[url_image, type_image], outputs=[image])
+    url_image.change(fn=get_meta_from_image, inputs=[url_image, type_image], outputs=[image])
 if __name__ == '__main__':
     demo.queue()
     demo.launch(debug=True, share=False)