Upload 2 files

server.py

@@ -0,0 +1,45 @@
+import cv2
+import socket
+import pickle
+import numpy as np
+import demo
+
+s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+ip = "127.0.0.1"
+port = 6666
+s.bind((ip,port))
+
+face_locations = []
+# face_encodings = []
+face_names = []
+process_this_frame = True
+
+score = []
+
+faces = 0
+
+while True:
+    print("Hello")
+    x = s.recvfrom(1000000)
+    clientip = x[1][0]
+    data = x[0]
+
+    data = pickle.loads(data)
+
+    frame = cv2.imdecode(data, cv2.IMREAD_COLOR)
+
+    result, process_this_frame, face_locations, faces, face_names, score = demo.process_frame(frame, process_this_frame, face_locations, faces, face_names, score)
+    
+    result = cv2.cvtColor(result, cv2.COLOR_BGR2RGB)
+
+    cv2.imshow('server side', result)
+
+    print(score)
+    if len(score) > 20:
+        avg_score =  sum(score) / len(score)
+        break
+    
+    if cv2.waitKey(5) & 0xFF == 27:
+        break
+
+cv2.destroyAllWindows()

webapp.py

@@ -3,13 +3,13 @@ import similarity_check as sc
 import cv2
 from PIL import Image
 import numpy as np
-from streamlit_webrtc import VideoTransformerBase, webrtc_streamer
 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
 
 def main():
 
@@ -122,38 +122,53 @@ def main():
     faces = 0
 
     FRAME_WINDOW = st.image([])
-    # camera = cv2.VideoCapture(0)
 
+    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
+
+    camera = cv2.VideoCapture(0)
 
     while run:
 
         # Capture frame-by-frame
         # Grab a single frame of video
+        ret, frame = camera.read()
 
-        frame_buffer = st.camera_input("Take a picture")
+        if ret is not None:
+            ret, buffer = cv2.imencode(".jpg", frame, [int(cv2.IMWRITE_JPEG_QUALITY),30])
 
-        if frame_buffer is not None:
-            bytes_data = frame_buffer.getvalue()
-            frame = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), cv2.IMREAD_COLOR)
-            result, process_this_frame, face_locations, faces, face_names, score = demo.process_frame(frame, process_this_frame, face_locations, faces, face_names, score)
-            # Display the resulting image
-            FRAME_WINDOW.image(result)
-        else:
-            st.write("No frames.")
-            print("there is no frame detected")
-            run = not run
-            break
-
-        print(score)
-        if len(score) > 20:
-            avg_score =  sum(score) / len(score)
-            st.write(avg_score)
-            # st.write(f'{demo.convert_distance_to_percentage(avg_score, 0.45)}')
-            # camera.release()
-            run = not run
-            st.session_state['data']['avg_score'] = str(avg_score)
-    else:
-        st.write('Stopped')
+        x_as_bytes = pickle.dumps(buffer)
+
+        s.sendto((x_as_bytes),(server_ip, server_port))
+
+        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+
+        FRAME_WINDOW.image(frame)
+
+        # camera.release()
+        # if ret:
+        #     # ret, buffer = cv2.imencode(".jpg", frame, [int(cv2.IMWRITE_JPEG_QUALITY)])
+        #     # result, process_this_frame, face_locations, faces, face_names, score = demo.process_frame(frame, process_this_frame, face_locations, faces, face_names, score)
+        #     # Display the resulting image
+        #     FRAME_WINDOW.image(frame)
+        # else:
+        #     print("there is no frame detected")
+        #     continue
+
+        # print(score)
+        # if len(score) > 20:
+        #     avg_score =  sum(score) / len(score)
+        #     st.write(avg_score)
+        #     # st.write(f'{demo.convert_distance_to_percentage(avg_score, 0.45)}')
+        #     # camera.release()
+        #     run = not run
+        #     st.session_state['data']['avg_score'] = str(avg_score)
+
+
+    ## unrelated
     
     if st.button("Confirm"):
         st.experimental_set_query_params(