Upload webapp.py

webapp.py

@@ -3,16 +3,10 @@ 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
 
@@ -137,10 +131,14 @@ def main():
         # 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)
+
+        if ret:
+            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:
+            print("there is no frame detected")
+            continue
 
         print(score)
         if len(score) > 20: