Update app.py

app.py

@@ -112,11 +112,13 @@ def process(image, bg):
  pred_pil = transforms.ToPILImage()(pred)
  mask = pred_pil.resize(image_size)
 
- if bg.startswith("#"):
+ if isinstance(bg, str) and bg.startswith("#"):
  color_rgb = tuple(int(bg[i:i+2], 16) for i in (1, 3, 5))
  background = Image.new("RGBA", image_size, color_rgb + (255,))
- else:
+ elif isinstance(bg, Image.Image):
  background = bg.convert("RGBA").resize(image_size)
+ else:
+ background = Image.open(bg).convert("RGBA").resize(image_size)
 
  # Composite the image onto the background using the mask
  image = Image.composite(image, background, mask)
@@ -126,10 +128,14 @@ def process(image, bg):
 
 with gr.Blocks(theme=gr.themes.Ocean()) as demo:
  with gr.Row():
- in_video = gr.Video(label="Input Video")
+ in_video = gr.Video(label="Input Video", interactive=True if isinstance(bg, str) and bg.startswith("#"):
+ color_rgb = tuple(int(bg[i:i+2], 16) for i in (1, 3, 5))
+ background = Image.new("RGBA", image_size, color_rgb + (255,))
+ elif isinstance(bg, Image.Image):
+ background = bg.convert("RGBA").resize(image_size))
  stream_image = gr.Image(label="Streaming Output", visible=False)
  out_video = gr.Video(label="Final Output Video")
- submit_button = gr.Button("Change Background")
+ submit_button = gr.Button("Change Background", interactive=True)
  with gr.Row():
  fps_slider = gr.Slider(
  minimum=0,
@@ -137,13 +143,14 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
  step=1,
  value=0,
  label="Output FPS (0 will inherit the original fps value)",
+ interactive=True
  )
- bg_type = gr.Radio(["Color", "Image", "Video"], label="Background Type", value="Color")
- color_picker = gr.ColorPicker(label="Background Color", value="#00FF00", visible=True)
- bg_image = gr.Image(label="Background Image", type="filepath", visible=False)
- bg_video = gr.Video(label="Background Video", visible=False)
+ bg_type = gr.Radio(["Color", "Image", "Video"], label="Background Type", value="Color", interactive=True)
+ color_picker = gr.ColorPicker(label="Background Color", value="#00FF00", visible=True, interactive=True)
+ bg_image = gr.Image(label="Background Image", type="filepath", visible=False, interactive=True)
+ bg_video = gr.Video(label="Background Video", visible=False, interactive=True)
  with gr.Column(visible=False) as video_handling_options:
- video_handling_radio = gr.Radio(["slow_down", "loop"], label="Video Handling", value="slow_down")
+ video_handling_radio = gr.Radio(["slow_down", "loop"], label="Video Handling", value="slow_down", interactive=True)
 
  def update_visibility(bg_type):
  if bg_type == "Color":