Update app.py

app.py

@@ -71,29 +71,31 @@ with gr.Blocks() as app:
             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)
-                
-                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")
+    with gr.Tabs():
+        with gr.TabItem("Color"):
+            enable_color_limit = gr.Checkbox(label="Enable", value=False)
+            palette_size_color = gr.Slider(label="Palette Size", minimum=1, maximum=256, step=1, value=16)
+            quantization_methods_color = gr.Radio(choices=["Median Cut", "Maximum Coverage", "Fast Octree"], label="Colors Quantization Method")
+            dither_methods_color = gr.Radio(choices=["None", "Floyd-Steinberg"], label="Colors Dither Method")
+            k_means_color = gr.Checkbox(label="Enable k-means for color quantization", value=True)
+        
+        with gr.TabItem("Grayscale"):
+            enable_grayscale = gr.Checkbox(label="Enable", value=False)
+            palette_size_gray = gr.Slider(label="Palette Size", minimum=1, maximum=256, step=1, value=16)
+            quantization_methods_gray = gr.Radio(choices=["Median Cut", "Maximum Coverage", "Fast Octree"], label="Colors Quantization Method")
+            dither_methods_gray = gr.Radio(choices=["None", "Floyd-Steinberg"], label="Colors Dither Method")
+            k_means_gray = gr.Checkbox(label="Enable k-means for color quantization", value=True)
+        
+        with gr.TabItem("Black and white"):
+            enable_black_and_white = gr.Checkbox(label="Enable", value=False)
+            inverse_black_and_white = gr.Checkbox(label="Inverse", value=False)
+            threshold_black_and_white = gr.Slider(label="Threshold", minimum=1, maximum=256, step=1, value=128)
+        
+        with gr.TabItem("Custom color palette"):
+            enable_custom_palette = gr.Checkbox(label="Enable", value=False)
+            palette_image = gr.Image(label="Color palette image", type="pil")
+            palette_size_custom = gr.Slider(label="Palette Size", minimum=1, maximum=256, step=1, value=16)
+            dither_methods_custom = gr.Radio(choices=["None", "Floyd-Steinberg"], label="Colors Dither Method")
             
             post_process_button = gr.Button("Apply Post-Processing")