Update app.py

app.py

@@ -55,43 +55,51 @@ with gr.Blocks() as app:
     title = gr.Markdown("# artificialguybr LoRA portfolio")
     description = gr.Markdown("### This is a Pixel Art Generator using SD Loras.")
     selected_state = gr.State()
+    
     with gr.Row():
         gallery = gr.Gallery([(item["image"], item["title"]) for item in loras], label="LoRA Gallery", allow_preview=False, columns=3)
+        
         with gr.Column():
             prompt_title = gr.Markdown("### Click on a LoRA in the gallery to create with it")
             selected_info = gr.Markdown("")
+            
             with gr.Row():
                 prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
                 button = gr.Button("Run")
+            
             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)
                     number_of_colors = gr.Slider(label="Palette Size", minimum=1, maximum=256, step=1, value=16)
+                
                 with gr.TabItem("Grayscale"):
                     is_grayscale = gr.Checkbox(label="Enable", value=False)
                     number_of_shades = gr.Slider(label="Palette Size", minimum=1, maximum=256, step=1, value=16)
+                
                 with gr.TabItem("Black and white"):
                     is_black_and_white = gr.Checkbox(label="Enable", value=False)
                     black_and_white_threshold = gr.Slider(label="Threshold", minimum=1, maximum=256, step=1, value=128)
+                
                 with gr.TabItem("Custom color palette"):
                     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")
-                    
+                with gr.TabItem("Post-Processing"):
+                    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()