Update app.py

app.py

@@ -2,9 +2,8 @@ import cv2
 import mediapipe as mp
 import numpy as np
 import gradio as gr
-import base64
-import time
 import os
+import time
 
 # Set MPLCONFIGDIR to avoid Matplotlib cache permission issues
 os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib-cache"
@@ -31,7 +30,7 @@ def process_frame(frame, seg_enabled, blur_bg, set_bg, set_color, bg_color, blur
         "blur_bg": blur_bg,
         "set_bg": set_bg,
         "set_color": set_color,
-        "bg_color": tuple(map(int, bg_color.split(","))) if set_color else (0, 0, 0),
+        "bg_color": tuple(map(int, bg_color.split(","))) if set_color and bg_color else (0, 0, 0),
         "blur_intensity": blur_intensity
     })
 
@@ -81,12 +80,12 @@ with gr.Blocks() as demo:
     gr.Markdown("# AI Background Remover")
     with gr.Row():
         with gr.Column():
-            webcam = gr.Image(sources="webcam", label="Live Video", streaming=True)
+            webcam = gr.Image(sources=["webcam"], streaming=True, label="Live Video")
             output_image = gr.Image(label="Processed Output")
             seg_enabled = gr.Checkbox(label="Enable Background Removal", value=True)
             blur_bg = gr.Checkbox(label="Blur Background")
             set_bg = gr.Checkbox(label="Custom Image Background")
-            custom_image = gr.Image(label="Upload Custom Background")
+            custom_image = gr.Image(label="Upload Custom Background", type="numpy")
             set_color = gr.Checkbox(label="Solid Color Background")
             bg_color = gr.Textbox(label="Background Color (R,G,B)", value="0,0,0")
             blur_intensity = gr.Slider(label="Blur Intensity", minimum=5, maximum=25, value=15, step=2)
@@ -95,8 +94,13 @@ with gr.Blocks() as demo:
     webcam.stream(
         fn=process_frame,
         inputs=[webcam, seg_enabled, blur_bg, set_bg, set_color, bg_color, blur_intensity, custom_image],
-        outputs=[output_image, processing_time]
+        outputs=[output_image, processing_time],
+        _js="""async () => {
+            // Ensure webcam permissions are requested
+            await navigator.mediaDevices.getUserMedia({ video: true });
+            return true;
+        }"""
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)