Upload webapp.py

webapp.py

@@ -9,25 +9,23 @@ import request_json.sbt_request_generator as sbt
 import check_hkid_validity as chv
 import av
 from streamlit_webrtc import webrtc_streamer, VideoTransformerBase, RTCConfiguration, WebRtcMode
+import search_engine as se
+import get_bank_statement as bs
 
-def init():
-    face_locations = []
-    # face_encodings = []
-    face_names = []
-    process_this_frame = True
+# def init():
+#     face_locations = []
+#     # face_encodings = []
+#     face_names = []
+#     process_this_frame = True
 
-    score = []
+#     score = []
 
-    faces = 0
+#     faces = 0
 
-def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
-    image = frame.to_ndarray(format="bgr24")
+# 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():
 
     # st.title("SBT Web Application")
@@ -91,7 +89,9 @@ def main():
             data = sc.get_data(image1, image2)
             # se.get_data_link(data['chi_name_id'], data["name_on_id"], data["address"])
         if 'data' in st.session_state:
+            data["nationality"] = 'N/A' # for hkid
             st.session_state['data'] = data
+            st.session_state['verified'] = "True"
         st.success('Done!')
         score = int(st.session_state['data']['similarity_score'])
         st.text(f'score: {score}')
@@ -101,13 +101,15 @@ def main():
             st.text(f'unmatched')
         
         data = st.session_state['data']
-        st.header("IIa. HKID Data Extraction")
+        st.header("Ia. HKID Data Extraction")
         st.text(f'English Name: {data["name_on_id"]}') # name is without space
         st.text(f'Chinese Name: {data["chi_name_id"]}') # name is without space
         st.text(f'HKID: {data["hkid"]} and validity: {data["validity"]}')
         st.text(f'Date of issue: {data["issue_date"]}')
+        st.text(f'Date of birth: {data["dateofbirth"]}')
+        st.text(f'nationality: {data["nationality"]}')
     
-        st.header("IIb. Bank Statement Data Extraction")
+        st.header("Ib. Bank Statement Data Extraction")
         st.text(f'Name: {data["nameStatement"]}')
         st.text(f'Address: {data["address"]}')
         st.text(f'Bank: {data["bank"]}')
@@ -117,11 +119,11 @@ def main():
 
     if 'data' in st.session_state:
         tempout = st.session_state['data']
-        print(f'hello: {tempout}')
+        print(f'data: {tempout}')
     
 
-    st.header("II. Facial Recognition")
-    run = st.checkbox('Run')
+    # st.header("II. Facial Recognition")
+    # run = st.checkbox('Run')
 
     # webrtc_streamer(key="example")
     # 1. Web Rtc
@@ -129,60 +131,61 @@ def main():
 
 
     # # init the camera
-    face_locations = []
+    # face_locations = []
     # face_encodings = []
-    face_names = []
-    process_this_frame = True
+    # face_names = []
+    # process_this_frame = True
 
-    score = []
+    # score = []
 
-    faces = 0
+    # faces = 0
 
-    FRAME_WINDOW = st.image([])
+    # FRAME_WINDOW = st.image([])
 
 
     # server_ip = "127.0.0.1"
     # server_port = 6666
 
-    # camera = cv2.VideoCapture(1)
+    # camera = cv2.VideoCapture(0)
     # 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 = []
 
-    if run:
+    # while run:
 
-        rtc_configuration = RTCConfiguration({"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]})
+        # rtc_configuration = RTCConfiguration({"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]})
 
-        # Capture frame-by-frame
-        # Grab a single frame of video
+        # # Capture frame-by-frame
+        # # Grab a single frame of video
         # 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,
-            video_frame_callback=video_frame_callback,
-            media_stream_constraints={"video": True, "audio": False},
-            async_processing=True,
-        )
+    
+        # result = frame
+        # # Initialize the WebRTC streaming
+        # webrtc_ctx = webrtc_streamer(
+        #     key="face_rec",
+        #     mode=WebRtcMode.SENDRECV,
+        #     rtc_configuration=rtc_configuration,
+        #     # 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
-            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)
+        # 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_WINDOW.image(frame)
+        # FRAME_WINDOW.image(result)
 
         # if ret is not None:
         #     ret, buffer = cv2.imencode(".jpg", frame, [int(cv2.IMWRITE_JPEG_QUALITY),30])
@@ -212,7 +215,17 @@ def main():
 
 
     ## unrelated
-    
+
+    st.header("III. Search Engine and Bank Statement")
+    user_input_id  = st.text_input("Enter the user ID here", " ")
+    if st.button("Search data"):
+        with st.spinner('Searching data...'):
+            se.get_data_link(user_input_id)
+        st.success('Done!')
+    if st.button("Fetch bank statement"):
+        with st.spinner('getting statements...'):
+            bs.get_bs(user_input_id)
+        st.success('Done!')
     if st.button("Confirm"):
         st.experimental_set_query_params(
             verified=True,
@@ -221,7 +234,7 @@ def main():
             print(st.session_state['data'])
             sbt.split_data(st.session_state['data'])
         st.success('Done!')
-
+    
 
 if __name__ == '__main__':
     main()