Update ui/gradio_ui.py

ui/gradio_ui.py

@@ -5,21 +5,27 @@ def ui(controller):#get_first_result,get_second_result): #controller):
         with gr.Row():
             with gr.Column():
                 sketch=gr.Image(sources = 'upload', label='Model image', type = 'pil')
+                
+            with gr.Column():
+                improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")        
+
+        #with gr.Row():
+            with gr.Column():
+                result=gr.Image(type="pil", label="Improved Sketch")
+                
+
+        with gr.Row():
+            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")
                 #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():
-                improved_sketch_view = gr.Image(type="pil", label="Improved Sketch")
 
-        
-
-        #with gr.Row():
             with gr.Column():
-                result=gr.Image(type="pil", label="Improved Sketch")
                 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")
                 result_button = gr.Button(value="Paint It", variant="primary")
+                
 
         improve_sketch.click(fn=controller.get_first_result,
                              inputs=[sketch, first_prompt, first_negative_prompt],