Update app.py

app.py

@@ -66,7 +66,6 @@ with gr.Blocks() as app:
             result = gr.Image(interactive=False, label="Generated Image")
             refined_result = gr.Image(interactive=False, label="Refined Generated Image")
             post_processed_result = gr.Image(interactive=False, label="Post-Processed Image")
-            
             with gr.Tabs():
                 with gr.TabItem("Color"):
                     enable_color_limit = gr.Checkbox(label="Enable", value=False)
@@ -81,12 +80,18 @@ with gr.Blocks() as app:
                     use_color_palette = gr.Checkbox(label="Enable", value=False)
                     palette_image = gr.Image(label="Color palette image", type="pil")
                     palette_colors = gr.Slider(label="Palette Size (only for complex images)", minimum=1, maximum=256, step=1, value=16)
+                
+                # Define downscale and other variables here
+                downscale = gr.Slider(label="Downscale", minimum=1, maximum=32, step=2, value=8)
+                limit_colors = gr.Checkbox(label="Limit Colors")
+                grayscale = gr.Checkbox(label="Grayscale")
+                black_and_white = gr.Checkbox(label="Black and White")
                     
             post_process_button = gr.Button("Apply Post-Processing")
 
     gallery.select(update_selection, outputs=[prompt, selected_info, selected_state])
     prompt.submit(fn=run_lora, inputs=[prompt, selected_state], outputs=[result, refined_result])
     post_process_button.click(fn=apply_post_processing, inputs=[refined_result, downscale, limit_colors, grayscale, black_and_white], outputs=[post_processed_result])
-
+    
 app.queue(max_size=20, concurrency_count=5)
 app.launch()