Update app.py

app.py

@@ -6,7 +6,6 @@ import json
 from image_processing import downscale_image, limit_colors, convert_to_grayscale, convert_to_black_and_white
 import logging
 
-
 class SomeClass:
     def __init__(self):
         self.images = []
@@ -23,12 +22,10 @@ def update_selection(selected_state: gr.SelectData):
     updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
     return (
         gr.update(placeholder=new_placeholder),
-        updated_text,  # Retorna o texto Markdown atualizado
+        updated_text,
         selected_state
     )
 
-
-
 def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
     selected_lora_index = selected_state.index
     selected_lora = loras[selected_lora_index]
@@ -69,12 +66,24 @@ 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.Row():
-                downscale = gr.Slider(minimum=1, maximum=10, step=1, label="Downscale")
-                limit_colors = gr.Checkbox(label="Limit Colors")
-                grayscale = gr.Checkbox(label="Grayscale")
-                black_and_white = gr.Checkbox(label="Black and White")
+            
+            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)
+                    
             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])