Upload webapp.py

webapp.py

@@ -10,6 +10,15 @@ import check_hkid_validity as chv
 import search_engine as se
 import socket
 import pickle
+from streamlit_webrtc import webrtc_streamer, VideoTransformerBase, RTCConfiguration, WebRtcMode
+
+class WebcamTransformer(VideoTransformerBase):
+    def __init__(self):
+        self.frame_count = 0
+
+    def transform(self, frame):
+        self.frame_count += 1
+        return frame
 
 def main():
 
@@ -123,30 +132,48 @@ def main():
 
     FRAME_WINDOW = st.image([])
 
-    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-    s.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 1000000)
 
-    server_ip = "127.0.0.1"
-    server_port = 6666
+    # server_ip = "127.0.0.1"
+    # server_port = 6666
 
-    camera = cv2.VideoCapture(1)
+    # camera = cv2.VideoCapture(1)
+    # s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    # s.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 1000000)
 
     while run:
 
+        rtc_configuration = RTCConfiguration({"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]})
+
         # Capture frame-by-frame
         # Grab a single frame of video
-        ret, frame = camera.read()
+        # ret, frame = camera.read()
+
+        # Initialize the WebRTC streaming
+        webrtc_ctx = webrtc_streamer(
+            key="face_rec",
+            mode=WebRtcMode.SENDRECV,
+            rtc_configuration=rtc_configuration,
+            video_transformer_factory=WebcamTransformer,
+            media_stream_constraints={"video": True, "audio": False},
+            async_processing=True,
+        )
+
+        if webrtc_ctx.video_transformer:
+            st.header("Webcam Preview")
+            frame = webrtc_ctx.video_transformer.frame
+            result, process_this_frame, face_locations, faces, face_names, score = demo.process_frame(frame, process_this_frame, face_locations, faces, face_names, score)
+            st.video(result)
 
-        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        # frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
 
-        FRAME_WINDOW.image(frame)
+        # FRAME_WINDOW.image(frame)
 
-        if ret is not None:
-            ret, buffer = cv2.imencode(".jpg", frame, [int(cv2.IMWRITE_JPEG_QUALITY),30])
+        # if ret is not None:
+        #     ret, buffer = cv2.imencode(".jpg", frame, [int(cv2.IMWRITE_JPEG_QUALITY),30])
 
-        x_as_bytes = pickle.dumps(buffer)
+        # x_as_bytes = pickle.dumps(buffer)
 
-        s.sendto((x_as_bytes),(server_ip, server_port))
+        # s.sendto((x_as_bytes),(server_ip, server_port))
 
         # camera.release()
         # if ret: