Update app.py

app.py

@@ -4,6 +4,10 @@ import numpy as np
 import gradio as gr
 import base64
 import time
+import os
+
+# Set MPLCONFIGDIR to avoid Matplotlib cache permission issues
+os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib-cache"
 
 # Initialize MediaPipe Selfie Segmentation
 mp_selfie_segmentation = mp.solutions.selfie_segmentation
@@ -34,6 +38,9 @@ def process_frame(frame, seg_enabled, blur_bg, set_bg, set_color, bg_color, blur
     if custom_image is not None and set_bg:
         bg_image = custom_image
 
+    if frame is None:
+        return None, "No video frame received"
+
     process_start = time.time()
     frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
     result = segmentation.process(frame_rgb)
@@ -74,7 +81,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# AI Background Remover")
     with gr.Row():
         with gr.Column():
-            webcam = gr.Webcam(label="Live Video")
+            webcam = gr.Image(sources="webcam", label="Live Video", streaming=True)
             output_image = gr.Image(label="Processed Output")
             seg_enabled = gr.Checkbox(label="Enable Background Removal", value=True)
             blur_bg = gr.Checkbox(label="Blur Background")