update gradio api

.gitignore

@@ -1,3 +1,4 @@
 .idea/
+.gradio/
 __pycache__/
 playground.py

app.py

@@ -27,11 +27,11 @@ def predict(image, session, smoot, matting, bg_color):
 
 
 def change_show_mask(chk_state):
-    return gr.Image.update(visible=chk_state)
+    return gr.Image(visible=chk_state)
 
 
 def change_include_matting(chk_state):
-    return gr.Box.update(visible=chk_state), (0, 0, 0), 0, 0, 0
+    return gr.Group(visible=chk_state), (0, 0, 0), 0, 0, 0
 
 
 def change_foreground_threshold(fg_value, value):
@@ -50,7 +50,7 @@ def change_erode_size(erode_value, value):
 
 
 def set_dominant_color(chk_state):
-    return chk_state, gr.ColorPicker.update(value=False, visible=not chk_state)
+    return chk_state, gr.ColorPicker(visible=not chk_state)
 
 
 def change_picker_color(picker, dominant):
@@ -60,8 +60,8 @@ def change_picker_color(picker, dominant):
 
 
 def change_background_mode(chk_state):
-    return gr.ColorPicker.update(value=False, visible=chk_state), \
-        gr.Checkbox.update(value=False, visible=chk_state)
+    return gr.ColorPicker(visible=chk_state), \
+        gr.Checkbox(value=False, visible=chk_state)
 
 
 footer = r"""
@@ -85,11 +85,11 @@ with gr.Blocks(title="Remove background") as app:
                 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:
+            with gr.Group(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.Group():
                 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)
@@ -128,4 +128,5 @@ with gr.Blocks(title="Remove background") as app:
     with gr.Row():
         gr.HTML(footer)
 
-app.launch(share=False, debug=True, enable_queue=True, show_error=True)
+app.queue()
+app.launch(share=False, debug=True, show_error=True)

cutter.py

@@ -1,6 +1,6 @@
 import PIL
 import numpy as np
-from PIL import Image, ImageColor, ImageDraw
+from PIL import Image, ImageDraw
 from PIL.Image import Image as PILImage
 from pymatting.alpha.estimate_alpha_cf import estimate_alpha_cf
 from pymatting.foreground.estimate_foreground_ml import estimate_foreground_ml
@@ -84,15 +84,25 @@ def remove(session, img: PILImage, smoot: bool, matting: tuple, color) -> (PILIm
         color = get_background_dominant_color(img, mask)
         cutout = apply_background_color(cutout, color)
     elif isinstance(color, str):
-        r, g, b = ImageColor.getcolor(color, "RGB")
-        cutout = apply_background_color(cutout, (r, g, b, 255))
+        cutout = apply_background_color(cutout, parse_rgba(color))
 
     return cutout, mask
 
+def parse_rgba(color_str):
+    color_values = color_str[5:-1].split(',')
+    r = int(float(color_values[0].strip()))
+    g = int(float(color_values[1].strip()))
+    b = int(float(color_values[2].strip()))
+    a = int(float(color_values[3].strip()) * 255)  # Alpha scaled to 0-255
+    return r, g, b, a
+
+def text_size(draw, text):
+    _, _, width, height = draw.textbbox((0, 0), text=text)
+    return width, height
 
 def make_label(text, width=600, height=200, color="black") -> PILImage:
     image = Image.new("RGB", (width, height), color)
     draw = ImageDraw.Draw(image)
-    text_width, text_height = draw.textsize(text)
+    text_width, text_height = text_size(draw, text)
     draw.text(((width-text_width)/2, height/2), text)
     return image

requirements.txt

@@ -1,4 +1,7 @@
 rembg~=2.0.47
-pillow~=9.5.0
-pymatting
+pillow~=10.4.0
+pymatting~=1.1.12
 opencv-python-headless
+gradio~=5.1.0
+numpy~=1.26.4
+scipy~=1.14.1