upgraded

app.py

@@ -66,6 +66,7 @@ class UI():
             pass
 
         with gr.Blocks() as demo:
+            self.faces = gr.State([])
             gr.Markdown("## FaceAnalyzer face recognition test")
             with gr.Tabs():
                 with gr.TabItem('Realtime Recognize'):
@@ -109,7 +110,7 @@ class UI():
                                 self.btn_clear = gr.Button("Clear Gallery")
 
                                 self.add_file = gr.Files(label="Files",file_types=["image"])
-                                self.add_file.change(self.add_files, self.add_file, self.gallery)
+                                self.add_file.change(self.add_files, self.add_file, [self.gallery, self.faces])
                                 self.txtFace_name2 = gr.Textbox(label="face_name")
                                 self.btn_start = gr.Button("Build face embeddings")
                                 self.status = gr.Label(label="Status")
@@ -156,11 +157,15 @@ class UI():
 
 
     def add_files(self, files):
-        for file in files:
-            img = cv2.cvtColor(cv2.imread(file.name), cv2.COLOR_BGR2RGB)
-            self.current_face_files.append(img)
-        return self.current_face_files
-    
+        current_face_files = []
+        if files is not None:
+            for file in files:
+                img = cv2.cvtColor(cv2.imread(file.name), cv2.COLOR_BGR2RGB)
+                current_face_files.append(img)
+            return current_face_files, current_face_files
+        else:
+            return []
+            
     def set_th(self, value):
         self.threshold=value
 
@@ -192,6 +197,7 @@ class UI():
 
     def upgrade_faces(self):
         # Load faces
+        print("Reloading faces")
         self.known_faces=[]
         self.known_faces_names=[]
         face_files = [f for f in faces_path.iterdir() if f.name.endswith("pkl")]