Update app.py

app.py

@@ -132,29 +132,31 @@ def main():
             app = FaceAnalysis(name='buffalo_l')
             app.prepare(ctx_id=0, det_size=(640, 640))
 
-            # Start capturing video from webcam
-            cap = cv2.VideoCapture(0)
-
-            # Process each frame in real-time
-            while True:
-                # Capture frame-by-frame
-                ret, frame = cap.read()
-                if not ret:
-                    break
-
-                # Perform face recognition
-                frame = recognize_faces(frame, names, embeddings, app)
-
-                # Display the resulting frame
-                st.image(frame, channels="BGR", use_column_width=True)
-
-                # Break the loop if 'q' is pressed
-                if cv2.waitKey(1) & 0xFF == ord('q'):
-                    break
-
-            # Release the capture
-            cap.release()
-            cv2.destroyAllWindows()
+            # Display a button to start webcam
+            if st.button("Start Webcam"):
+                # Start capturing video from webcam
+                cap = cv2.VideoCapture(0)
+
+                # Process each frame in real-time
+                while True:
+                    # Capture frame-by-frame
+                    ret, frame = cap.read()
+                    if not ret:
+                        break
+
+                    # Perform face recognition
+                    frame = recognize_faces(frame, names, embeddings, app)
+
+                    # Display the resulting frame
+                    st.image(frame, channels="BGR", use_column_width=True)
+
+                    # Break the loop if 'q' is pressed or the user closes the Streamlit app
+                    if st.button("Stop"):
+                        break
+
+                # Release the capture
+                cap.release()
+                cv2.destroyAllWindows()