test

app.py

@@ -70,15 +70,21 @@ def edit_bar(photo):
     sleep(wait_time)
     photo.save("image.png")
     if current_perc > 30 and not check_id:
-      # if photo.size != (1920, 1080):
-      #   error_text = '''<div style="border: 3px solid red;"><div style="font-size:3em;text-align:center; color:red">Ошибка!</div>
-      #   <div style="text-align:center; font-size:2em">Судя по всему вы используете неправильную вебкамеру! Попытайтесь снова.</div></div>'''
-      #   yield gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True, value=error_text)gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
-      #   return
-      embedding_objs = DeepFace.represent(
-        img_path = "image.png",
-        model_name = "Facenet512",
-      )
+      if photo.size != (1920, 1080):
+        error_text = '''<div style="border: 3px solid red;"><div style="font-size:3em;text-align:center; color:red">Ошибка!</div>
+        <div style="text-align:center; font-size:2em">Судя по всему вы используете неправильную вебкамеру! Попытайтесь снова.</div></div>'''
+        yield gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True, value=error_text), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return
+      try:
+        embedding_objs = DeepFace.represent(
+          img_path = "image.png",
+          model_name = "Facenet512",
+        )
+      except ValueError:
+        error_text = '''<div style="border: 3px solid red;"><div style="font-size:3em;text-align:center; color:red">Ошибка!</div>
+        <div style="text-align:center; font-size:2em">Не можем найти лицо на фото! Попытайтесь сделать снимок снова.</div></div>'''
+        yield gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True, value=error_text), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return
       embed = torch.Tensor(embedding_objs[0]['embedding'])
       alex_embed = torch.load("Alex_embed.pt")
       print(embed.shape, alex_embed.shape)