Update app.py

app.py

@@ -91,24 +91,26 @@ with gr.Blocks(css=css) as demo:
   with gr.Box(elem_id="main-image"):
     #with gr.Accordion(open=True, label="Your color sketch") as sketch_accordion:  
     with gr.Row():
-        with gr.Row():
-            image = gr.Image(interactive=True, tool="color-sketch", source="canvas", type="pil")
-        with gr.Row():
-            aspect = gr.Radio(["Square", "Horizontal", "Vertical"], value="Square", label="Aspect Ratio")
-        button_run = gr.Button("I've finished my sketch",elem_id="main_button")
-        
-        prompts = []
-        colors = []
-        color_row = [None] * MAX_COLORS
-        with gr.Column(visible=False) as post_sketch:
-          general_prompt = gr.Textbox(label="General Prompt")
-          for n in range(MAX_COLORS):
-            with gr.Row(visible=False) as color_row[n]:
-              with gr.Box(elem_id="color-bg"):
-                colors.append(gr.HTML('<div class="color-bg-item" style="background-color: black"></div>'))
-              prompts.append(gr.Textbox(label="Prompt for this color"))
-          final_run_btn = gr.Button("Generate!")
-        out_image = gr.Image()
+        with gr.Column():
+            with gr.Row():
+                image = gr.Image(interactive=True, tool="color-sketch", source="canvas", type="pil")
+            with gr.Row():
+                aspect = gr.Radio(["Square", "Horizontal", "Vertical"], value="Square", label="Aspect Ratio")
+            button_run = gr.Button("I've finished my sketch",elem_id="main_button")
+            
+            prompts = []
+            colors = []
+            color_row = [None] * MAX_COLORS
+            with gr.Column(visible=False) as post_sketch:
+              general_prompt = gr.Textbox(label="General Prompt")
+              for n in range(MAX_COLORS):
+                with gr.Row(visible=False) as color_row[n]:
+                  with gr.Box(elem_id="color-bg"):
+                    colors.append(gr.HTML('<div class="color-bg-item" style="background-color: black"></div>'))
+                  prompts.append(gr.Textbox(label="Prompt for this color"))
+              final_run_btn = gr.Button("Generate!")
+        with gr.Column():
+            out_image = gr.Image()
   
   css_height = gr.HTML("<style>#main-image{width: 512px} .fixed-height{height: 512px !important}</style>")