Update app.py

app.py

@@ -84,21 +84,26 @@ if upload_option == "Image Upload":
 
 elif upload_option == "Webcam":
     st.sidebar.write("Webcam Capture")
-    run = st.checkbox('Run Webcam')
+    run_webcam = st.sidebar.button('Run Webcam')
+    stop_webcam = st.sidebar.button('Stop Webcam')
     FRAME_WINDOW = st.image([])
 
-    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()
+    if run_webcam:
+        camera = cv2.VideoCapture(0)
+        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
 else:
     st.write("Please select an option to start.")
-