Update app.py

app.py

@@ -1,8 +1,6 @@
 import logging
-
 import gradio as gr
 from rembg import new_session
-
 from cutter import remove, make_label
 from utils import *
 
@@ -12,116 +10,28 @@ remove_bg_models = {
     "U2NET Cloth Seg": "u2net_cloth_seg"
 }
 
-model_choices = keys(remove_bg_models)
-
+model_choices = list(remove_bg_models.keys())
 
 def predict(image, session, smoot, matting, bg_color):
-
     session = new_session(remove_bg_models[session])
-
     try:
         return remove(session, image, smoot, matting, bg_color)
     except ValueError as err:
         logging.error(err)
         return make_label(str(err)), None
 
-
-def change_show_mask(chk_state):
-    return gr.Image.update(visible=chk_state)
-
-
-def change_include_matting(chk_state):
-    return gr.Box.update(visible=chk_state), (0, 0, 0), 0, 0, 0
-
-
-def change_foreground_threshold(fg_value, value):
-    fg, bg, erode = value
-    return fg_value, bg, erode
-
-
-def change_background_threshold(bg_value, value):
-    fg, bg, erode = value
-    return fg, bg_value, erode
-
-
-def change_erode_size(erode_value, value):
-    fg, bg, erode = value
-    return fg, bg, erode_value
-
-
-def set_dominant_color(chk_state):
-    return chk_state, gr.ColorPicker.update(value=False, visible=not chk_state)
-
-
-def change_picker_color(picker, dominant):
-    if not dominant:
-        return picker
-    return dominant
-
-
-def change_background_mode(chk_state):
-    return gr.ColorPicker.update(value=False, visible=chk_state), \
-        gr.Checkbox.update(value=False, visible=chk_state)
-
-
-footer = r"""
-
-"""
-
 with gr.Blocks(title="Remove background") as app:
-    color_state = gr.State(value=False)
-    matting_state = gr.State(value=(0, 0, 0))
-
     gr.HTML("<center><h1>Background Remover</h1></center>")
     with gr.Row(equal_height=False):
         with gr.Column():
             input_img = gr.Image(type="pil", label="Input image")
-            drp_models = gr.Dropdown(choices=model_choices, label="Model Segment", value="U2NET")
-            with gr.Row():
-                chk_include_matting = gr.Checkbox(label="Matting", value=False)
-                chk_smoot_mask = gr.Checkbox(label="Smoot Mask", value=False)
-                chk_show_mask = gr.Checkbox(label="Show Mask", value=False)
-            with gr.Box(visible=False) as slider_matting:
-                slr_fg_threshold = gr.Slider(0, 300, value=270, step=1, label="Alpha matting foreground threshold")
-                slr_bg_threshold = gr.Slider(0, 50, value=20, step=1, label="Alpha matting background threshold")
-                slr_erode_size = gr.Slider(0, 20, value=11, step=1, label="Alpha matting erode size")
-            with gr.Box():
-                with gr.Row():
-                    chk_change_color = gr.Checkbox(label="Change background color", value=False)
-                    pkr_color = gr.ColorPicker(label="Pick a new color", visible=False)
-                    chk_dominant = gr.Checkbox(label="Use dominant color", value=False, visible=False)
             run_btn = gr.Button(value="Remove background", variant="primary")
+            clear_btn = gr.Button(value="Clear", variant="secondary")
         with gr.Column():
-            output_img = gr.Image(type="pil", label="Image Result")
-            mask_img = gr.Image(type="pil", label="Image Mask", visible=False)
-            gr.ClearButton(components=[input_img, output_img, mask_img])
-
-    chk_include_matting.change(change_include_matting, inputs=[chk_include_matting],
-                               outputs=[slider_matting, matting_state,
-                                        slr_fg_threshold, slr_bg_threshold, slr_erode_size])
-
-    slr_fg_threshold.change(change_foreground_threshold, inputs=[slr_fg_threshold, matting_state],
-                            outputs=[matting_state])
-
-    slr_bg_threshold.change(change_background_threshold, inputs=[slr_bg_threshold, matting_state],
-                            outputs=[matting_state])
-
-    slr_erode_size.change(change_erode_size, inputs=[slr_erode_size, matting_state],
-                          outputs=[matting_state])
-
-    chk_show_mask.change(change_show_mask, inputs=[chk_show_mask], outputs=[mask_img])
-
-    chk_change_color.change(change_background_mode, inputs=[chk_change_color],
-                            outputs=[pkr_color, chk_dominant])
-
-    pkr_color.change(change_picker_color, inputs=[pkr_color, chk_dominant], outputs=[color_state])
-
-    chk_dominant.change(set_dominant_color, inputs=[chk_dominant], outputs=[color_state, pkr_color])
-
-    run_btn.click(predict, inputs=[input_img, drp_models, chk_smoot_mask, matting_state, color_state],
-                 outputs=[output_img, mask_img])
+            output_img = gr.Image(type="pil", label="Result image")
 
-    with gr.Row():
-        gr.HTML(footer)
+    run_btn.click(predict, inputs=[input_img, gr.Dropdown(choices=model_choices, value="U2NET"), False, None, None],
+                  outputs=[output_img])
+    clear_btn.click(lambda: (None, None), inputs=None, outputs=[input_img, output_img])
 
 app.launch(share=False, debug=True, enable_queue=True, show_error=True)