update UI

S2I/commons/controller.py

@@ -58,10 +58,10 @@ class Sketch2ImageController():
         self.load_pipeline(zero_options=options)
         # prompt = prompt_template.replace("{prompt}", prompt)
 
-        if type_flag == 'live-sketch':
-            img = Image.fromarray(np.array(image["composite"])[:, :, -1])
-        elif type_flag == 'url-sketch':
+        if input_type:
             img = image["composite"]
+        else:
+            img = Image.fromarray(np.array(image["composite"])[:, :, -1])
         
         img = img.convert("RGB")
         img = img.resize((512, 512))

app.py

@@ -221,33 +221,36 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
             )
             with gr.Group():
                 with gr.Row():
-                    url_image = gr.Textbox(label="Image URLS", value="")
-                    type_image = gr.Radio(
-                                    choices=["RGB", "SKETCH"],
-                                    value="SKETCH",
-                                    label="Type of Image (Color Image or Sketch Image)",
-                                    interactive=True)
-                    with gr.Row():
-                            ui_mode = gr.Radio(
-                                    choices=["Light Mode", "Dark Mode"],
-                                    value="Light Mode",
-                                    label="Switch Light/Dark Mode UI",
-                                    interactive=True)
-                            zero_gpu_options = gr.Radio(
-                                        choices=["GPU", "CPU"],
-                                        value="GPU",
-                                        label="GPU & CPU Options Spaces",
+                    with gr.Group():
+                        input_type = gr.Checkbox(label="Use URL Image Sketch", value=False)
+                        url_image = gr.Textbox(label="Image URLS", value="")
+                        type_image = gr.Radio(
+                                        choices=["RGB", "SKETCH"],
+                                        value="SKETCH",
+                                        label="Type of Image (Color Image or Sketch Image)",
                                         interactive=True)
-                            model_options = gr.Radio(
-                                        choices=["350k", "350k-adapter"],
-                                        value="350k-adapter",
-                                        label="Type Sketch2Image models",
-                                        interactive=True)
-                            half_model = gr.Radio(
-                                        choices=["float32", "float16"],
-                                        value="float16",
-                                        label="Demo Speed",
+                    with gr.Group():
+                        with gr.Row():
+                                ui_mode = gr.Radio(
+                                        choices=["Light Mode", "Dark Mode"],
+                                        value="Light Mode",
+                                        label="Switch Light/Dark Mode UI",
                                         interactive=True)
+                                zero_gpu_options = gr.Radio(
+                                            choices=["GPU", "CPU"],
+                                            value="GPU",
+                                            label="GPU & CPU Options Spaces",
+                                            interactive=True)
+                                model_options = gr.Radio(
+                                            choices=["350k", "350k-adapter"],
+                                            value="350k-adapter",
+                                            label="Type Sketch2Image models",
+                                            interactive=True)
+                                half_model = gr.Radio(
+                                            choices=["float32", "float16"],
+                                            value="float16",
+                                            label="Demo Speed",
+                                            interactive=True)
 
         with gr.Column(elem_id="column_output"):
             gr.Markdown("## IMAGE GENERATE", elem_id="output_header")
@@ -268,12 +271,6 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
                     clear_button = gr.Button("Reset Sketch Session", min_width=10, variant='primary')
                 with gr.Accordion("S2I Advances Option", open=True):
                         with gr.Row():
-                                input_type = gr.Radio(
-                                    choices=["live-sketch", "url-sketch"],
-                                    value="live-sketch",
-                                    label="Type Sketch2Image models",
-                                    interactive=True)
-                                                                        
                                 style = gr.Dropdown(
                                     label="Style",
                                     choices=controller.STYLE_NAMES,