Update app.py

app.py

@@ -1,20 +1,38 @@
 import streamlit as st
 import cv2
 import numpy as np
-from PIL import Image
+from ultralytics import YOLO  
 
-st.set_page_config(page_title="Webcam App", page_icon="📷")
+# Load YOLO model (Download "yolov8n.pt" first if not available)
+model = YOLO("yolov8n.pt")  
 
-st.title("📷 Streamlit Webcam App")
+st.title("🎥 YOLO Object Detection Live Stream")
 
-# Use Streamlit's built-in camera input
-image_file = st.camera_input("Take a picture")
+# Open Webcam
+cap = cv2.VideoCapture(0)
+if not cap.isOpened():
+    st.error("Error: Could not access the webcam.")
+    st.stop()
 
-if image_file:
-    # Convert image to OpenCV format
-    img = Image.open(image_file)
-    img = np.array(img)  # Convert to NumPy array
- 
-    
-    # Show the captured image
-    st.image(img, caption="Captured Image")
+stframe = st.empty()  # Streamlit placeholder for video frames
+
+while True:
+    ret, frame = cap.read()
+    if not ret:
+        st.error("Error: Failed to capture frame.")
+        break
+
+    # Convert frame to RGB (Streamlit expects RGB format)
+    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+
+    # Run YOLO object detection
+    results = model(frame)
+    img_pot = result[0].plot()
+
+  
+
+    # Show processed frame in Streamlit
+    stframe.image(img_pot, channels="RGB")
+
+# Release webcam
+cap.release()