import streamlit as st import similarity_check as sc import cv2 from PIL import Image import numpy as np import tempfile from streamlit_webrtc import VideoTransformerBase, webrtc_streamer import demo import time import streamlit as st import requests import json import request_json.sbt_request_generator as sbt import pathlib import os import check_hkid_validity as chv import search_engine as se def main(): # st.title("SBT Web Application") # today's date = get_today_date # global data html_temp = """

SBT Web Application

""" st.markdown(html_temp, unsafe_allow_html=True) if 'hkid_image_validity' not in st.session_state: st.session_state.hkid_image_validity = False if 'data' not in st.session_state: st.session_state['data'] = {} st.header("I. Similarity Check") image_file = st.file_uploader("Upload Image", type=['jpg', 'png', 'jpeg', 'pdf'], accept_multiple_files=True) if len(image_file) == 1: image1 = Image.open(image_file[0]) st.text("HKID card") st.image(image1) image1.save('image/hkid.jpg', 'JPEG') if chv.check_hkid('image/hkid.jpg'): st.text("Valid HKID card.") st.session_state.hkid_image_validity = True else: st.text("Invalid HKID card. Please upload again!") st.session_state.hkid_image_validity = False elif len(image_file) == 2: image1 = Image.open(image_file[0]) st.text("HKID card") st.image(image1) image2 = Image.open(image_file[1]) # image2 = image_file[1] # image2.save('image/hkid.jpg', 'JPEG') # file_name = image_file[1].name st.text("Bank statement") st.image(image2) print(f"the id is: {st.session_state.hkid_image_validity}") # if image_file2 is not None: # image2 = Image.open(image_file) # st.text("Bank statement") # st.image(image2) # path1 = 'IMG_4495.jpg' # path2 = 'hangseng_page-0001.jpg' # image1 = save_image(image1) # image2 = save_image(image2) data = {} if st.button("Recognise"): with st.spinner('Wait for it...'): # global data 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: st.session_state['data'] = data st.success('Done!') score = int(st.session_state['data']['similarity_score']) st.text(f'score: {score}') if (score>85): st.text(f'matched') else: st.text(f'unmatched') data = st.session_state['data'] st.header("IIa. 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.header("IIb. Bank Statement Data Extraction") st.text(f'Name: {data["nameStatement"]}') st.text(f'Address: {data["address"]}') st.text(f'Bank: {data["bank"]}') st.text(f'Date: {data["statementDate"]}') st.text(f'Asset: {data["totalAsset"]} hkd') st.text(f'Liabilities: {data["totalLiability"]} hkd') if 'data' in st.session_state: tempout = st.session_state['data'] print(f'hello: {tempout}') st.header("II. Facial Recognition") run = st.checkbox('Run') # webrtc_streamer(key="example") # 1. Web Rtc # webrtc_streamer(key="jhv", video_frame_callback=video_frame_callback) # # init the camera face_locations = [] # face_encodings = [] face_names = [] process_this_frame = True score = [] faces = 0 FRAME_WINDOW = st.image([]) camera = cv2.VideoCapture(0) while run: # Capture frame-by-frame # Grab a single frame of video ret, frame = camera.read() 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) 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') if st.button("Confirm"): st.experimental_set_query_params( verified=True, ) with st.spinner('Sending data...'): print(st.session_state['data']) sbt.split_data(st.session_state['data']) st.success('Done!') if __name__ == '__main__': main()