Upload webapp.py

webapp.py

@@ -7,18 +7,26 @@ import demo
 import streamlit as st
 import request_json.sbt_request_generator as sbt
 import check_hkid_validity as chv
-import search_engine as se
-import socket
-import pickle
+import av
 from streamlit_webrtc import webrtc_streamer, VideoTransformerBase, RTCConfiguration, WebRtcMode
 
-class WebcamTransformer(VideoTransformerBase):
-    def __init__(self):
-        self.frame_count = 0
+def init():
+    face_locations = []
+    # face_encodings = []
+    face_names = []
+    process_this_frame = True
 
-    def transform(self, frame):
-        self.frame_count += 1
-        return frame
+    score = []
+
+    faces = 0
+
+def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
+    image = frame.to_ndarray(format="bgr24")
+    
+
+
+
+    return av.VideoFrame.from_ndarray(image, format="bgr24")
 
 def main():
 
@@ -139,8 +147,11 @@ def main():
     # camera = cv2.VideoCapture(1)
     # s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
     # s.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 1000000)
+    
+    # if "face_rec" not in st.session_state:
+    #     st.session_state.face_rec = []
 
-    while run:
+    if run:
 
         rtc_configuration = RTCConfiguration({"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]})
 
@@ -153,11 +164,16 @@ def main():
             key="face_rec",
             mode=WebRtcMode.SENDRECV,
             rtc_configuration=rtc_configuration,
-            video_transformer_factory=WebcamTransformer,
+            # video_transformer_factory=WebcamTransformer,
+            video_frame_callback=video_frame_callback,
             media_stream_constraints={"video": True, "audio": False},
             async_processing=True,
         )
 
+        # print(f'xd: look here {type(webrtc_ctx)}')
+
+        # st.session_state.face_rec = webrtc_ctx
+
         if webrtc_ctx.video_transformer:
             st.header("Webcam Preview")
             frame = webrtc_ctx.video_transformer.frame