Update app.py

app.py

@@ -91,23 +91,24 @@ 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():
-        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.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()
   
   css_height = gr.HTML("<style>#main-image{width: 512px} .fixed-height{height: 512px !important}</style>")