Update app.py

app.py

@@ -84,29 +84,22 @@ if upload_option == "Image Upload":
 
 elif upload_option == "Webcam":
     st.sidebar.write("Webcam Capture")
-    run_webcam = st.sidebar.button('Run Webcam')
-    stop_webcam = st.sidebar.button('Stop Webcam')
+    run = st.checkbox('Run Webcam')
     FRAME_WINDOW = st.image([])
 
-    if run_webcam:
-        camera = cv2.VideoCapture(0)
-        if not camera.isOpened():
-            st.error("Could not open webcam. Please check if it is connected and try again.")
-        else:
-            st.session_state['run'] = True
-
-    if 'run' in st.session_state and st.session_state['run']:
-        while True:
-            success, frame = camera.read()
-            if not success:
-                st.error("Unable to read from webcam. Please check your camera settings.")
-                break
-            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            processed_frame = video_frame_callback(frame)
-            FRAME_WINDOW.image(processed_frame)
-            if stop_webcam:
-                st.session_state['run'] = False
-                camera.release()
-                break
+    camera = cv2.VideoCapture(0)
+
+    while run:
+        success, frame = camera.read()
+        if not success:
+            st.error("Unable to read from webcam. Please check your camera settings.")
+            break
+        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        processed_frame = video_frame_callback(frame)
+        FRAME_WINDOW.image(processed_frame)
+
+    camera.release()
 else:
     st.write("Please select an option to start.")
+
+