Update app.py

app.py

@@ -58,7 +58,7 @@ def process_sketch(image, binary_matrixes):
   for n in range(how_many_colors-1):
     visibilities[n] = gr.update(visible=True)
     colors[n] = colors_fixed[n]
-  return [gr.update(visible=True), binary_matrixes, *visibilities, *colors]
+  return [gr.update(open=False), gr.update(visible=True), binary_matrixes, *visibilities, *colors]
 
 def process_generation(binary_matrixes, master_prompt, *prompts):
     clipped_prompts = prompts[:len(binary_matrixes)]
@@ -86,11 +86,12 @@ def update_css(aspect):
 with gr.Blocks(css=css) as demo:
   binary_matrixes = gr.State([])
   with gr.Box(elem_id="main-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")
+    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 = []
@@ -103,11 +104,11 @@ with gr.Blocks(css=css) as demo:
             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!")
-      gallery = gr.Gallery()
+      out_image = gr.Image()
   
   css_height = gr.HTML("<style>#main-image{width: 512px} .fixed-height{height: 512px !important}</style>")
   
   aspect.change(update_css, inputs=aspect, outputs=css_height)
-  button_run.click(process_sketch, inputs=[image, binary_matrixes], outputs=[post_sketch, binary_matrixes, *color_row, *colors])
-  final_run_btn.click(process_generation, inputs=[binary_matrixes, general_prompt, *prompts], outputs=gallery)
+  button_run.click(process_sketch, inputs=[image, binary_matrixes], outputs=[sketch_accordion, post_sketch, binary_matrixes, *color_row, *colors])
+  final_run_btn.click(process_generation, inputs=[binary_matrixes, general_prompt, *prompts], outputs=out_image)
 demo.launch()