add extra tools

app.py

@@ -32,13 +32,19 @@ def clean_adverse(
 
 def clean(
         image: Image.Image,
-        use_adverse_clean: bool = True,
-        use_scunet_clean: bool = True,
+
+        diameter: float = 5,
+        sigma_color: float = 8,
+        sigma_space: float = 8,
+        radius: float = 4,
+        eps: float = 16,
+
+        use_scunet_clean: bool = False,
         use_nafnet_clean: bool = False
 ) -> Image.Image:
     image = load_image(image)
-    if use_adverse_clean:
-        image = clean_adverse(image)
+
+    image = clean_adverse(image, diameter, sigma_color, sigma_space, radius, eps)
     if use_scunet_clean:
         image = restore_with_scunet(image)
     if use_nafnet_clean:
@@ -52,16 +58,70 @@ if __name__ == '__main__':
             with gr.Column():
                 gr_input_image = gr.Image(label='Input Image', type="pil")
                 gr_submit = gr.Button(value='FUCK YOU MIST!')
-                gr_adverse = gr.Checkbox(label='Use Adverse Clean', value=True)
-                gr_scunet = gr.Checkbox(label='Use SCUNET', value=True)
-                gr_nafnet = gr.Checkbox(label='Use NafNET', value=False)
+                with gr.Accordion("Advanced Config", open=False):
+                    diameter_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=5,
+                        label="Diameter (default = 5)",
+                        interactive=True,
+                    )
+                    sigma_color_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=8,
+                        label="SigmaColor (default = 8)",
+                        interactive=True,
+                    )
+                    sigma_space_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=8,
+                        label="SigmaSpace (default = 8)",
+                        interactive=True,
+                    )
+
+                    radius_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=4,
+                        label="Radius (default = 4)",
+                        interactive=True,
+                    )
+                    eps_slider = gr.Slider(
+                        minimum=1,
+                        maximum=30,
+                        step=1,
+                        value=16,
+                        label="Accuracy (default = 16)",
+                        interactive=True,
+                    )
+
+                with gr.Accordion("Extra Restoration", open=False):
+                    gr_scunet = gr.Checkbox(label='Use SCUNET', value=False)
+                    gr_nafnet = gr.Checkbox(label='Use NafNET', value=False)
 
             with gr.Column():
                 gr_output_image = gr.Image(label='Output Image', type="pil")
 
             gr_submit.click(
                 fn=clean,
-                inputs=[gr_input_image, gr_adverse, gr_scunet, gr_nafnet],
+                inputs=[
+                    gr_input_image,
+
+                    diameter_slider,
+                    sigma_color_slider,
+                    sigma_space_slider,
+                    radius_slider,
+                    eps_slider,
+
+                    gr_scunet,
+                    gr_nafnet,
+                ],
                 outputs=[gr_output_image],
             )