Update app.py

app.py

@@ -53,12 +53,12 @@ def pretty_print_response(resp: dict):
 def process_webcam_stream(image):
     """Send webcam image to HF MCP Server and get result"""
     if image is None:
-        return "", "", "", ""
+        return None, None, None, None
 
-    # Convert to base64
+    # Convert PIL Image to base64
+    import io
     buffered = io.BytesIO()
-    img = Image.fromarray(image)
-    img.save(buffered, format="JPEG")
+    image.save(buffered, format="JPEG")  # <--- directly save PIL Image
     b64_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
     # Prepare payload
@@ -69,21 +69,23 @@ def process_webcam_stream(image):
         "hf_token": HF_TOKEN
     }
 
-    # Send to HF Space
+    # Send to HF Space using streaming-friendly predict
     client = Client(HF_SPACE)
     try:
         resp = client.predict(payload, api_name=API_NAME)
+        # Print table in console
         pretty_print_response(resp)
+
+        # Return selected fields for Gradio display
         return (
             resp.get("description", ""),
             resp.get("human", ""),
-            ", ".join(resp.get("objects", [])) if resp.get("objects") else "",
+            ", ".join(resp.get("objects", [])),
             resp.get("environment", "")
         )
     except Exception as e:
         console.print(f"[bold red]Error sending to HF:[/bold red] {e}")
-        return "", "", "", ""
-
+        return None, None, None, None
 
 with gr.Blocks() as demo:
     gr.Markdown("## 🎥 Robot Vision Webcam Stream")