Update app.py

app.py

@@ -1,14 +1,24 @@
 import cv2
 import streamlit as st
+import time
 
 st.title("webcam live feed")
-run = st.checkbox('Run')
+run = st.toggle('Start Webcam')
 FRAME_WINDOW = st.image([])
 camera = cv2.VideoCapture(0)
+time.sleep(2)
+
+if not camera.isOpened():
+        print("Cannot open camera")
+        exit()
 
 while run:
     _, frame = camera.read()
-    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-    FRAME_WINDOW.image(frame)
+    if not _:
+        print("Can't receive frame (steam end?).Exiting ...")
+        break
+    else:
+        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        FRAME_WINDOW.image(frame)
 else:
     st.write('Stopped')