bug fixed

pages/1_Technology Behind It.py

@@ -1,13 +1,26 @@
 import streamlit as st
 from utils.levels import complete_level, render_page, initialize_level
 from utils.login import initialize_login
+import cv2
 
 LEVEL = 1
 
 initialize_login()
 initialize_level()
 
-
+def returnCameraIndexes():
+    # checks the first 10 indexes.
+    index = 0
+    arr = []
+    i = 10
+    while i > 0:
+        cap = cv2.VideoCapture(index)
+        if cap.read()[0]:
+            arr.append(index)
+            cap.release()
+        index += 1
+        i -= 1
+    return arr
 
 def step1_page():
     st.header("Technology Behind It")
@@ -47,6 +60,29 @@ def step1_page():
 
     st.info("Click on the button below to continue!")
 
+    """TEST"""
+    input_type = st.radio("Select the Input Type", ["Image upload", "Camera", "Live Video"])
+    st.write(returnCameraIndexes())
+    if input_type == "Live Video":
+        cam = cv2.VideoCapture(0)
+        if not cam.isOpened():
+            print("Cannot open camera")
+            exit()
+        cam.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
+        cam.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)
+        FRAME_WINDOW = st.image([])
+
+        while True:
+            ret, frame = cam.read()
+            if not ret:
+                st.error("Failed to capture frame from camera")
+                st.info("Please turn off the other app that is using the camera and restart app")
+                st.stop()
+            # image, name, face_id = recognize(frame, tolerance)
+            image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            # Display name and ID of the person
+            FRAME_WINDOW.image(image)
+
     if st.button("Complete"):
         complete_level(LEVEL)