Update app.py

app.py

@@ -289,6 +289,13 @@ def badge(text: str, colour: str) -> str:
 # Face comparison
 # ───────────────────────────────
 def compare(img_left, img_right, variant):
+    if img_left is None and img_right is None:
+        return None, None, badge("Please upload/select two face images", "#DC2626")
+    if img_left is None:
+        return None, None, badge("Please upload/select a face image for Image A (left)", "#DC2626")
+    if img_right is None:
+        return None, None, badge("Please upload/select a face image for Image B (right)", "#DC2626")
+
     img_left = Image.fromarray(img_left).convert('RGB')
     img_right = Image.fromarray(img_right).convert('RGB')
 
@@ -296,9 +303,9 @@ def compare(img_left, img_right, variant):
     if crop_a is None and crop_b is None:
         return None, None, badge("No face detected", "#DC2626")
     if crop_a is None:
-        return None, None, badge("No face in A", "#DC2626")
+        return None, None, badge("No face was detected in Image A (left)", "#DC2626")
     if crop_b is None:
-        return None, None, badge("No face in B", "#DC2626")
+        return None, None, badge("No face was detected in Image B (right)", "#DC2626")
     mdl = get_face_rec_model(variant)
     dev = next(mdl.parameters()).device
     with torch.no_grad():
@@ -326,7 +333,7 @@ HERO_HTML = f"""
     <a href="https://arxiv.org/abs/2411.08470v2">arXiv</a>&nbsp;•&nbsp;
     <a href="https://gitlab.idiap.ch/biometric/code.iclr2025_hyperface">Code</a>&nbsp;•&nbsp;
     <a href="https://huggingface.co/collections/Idiap/hyperface-682485119ccbd3ba5c42bde1">Models</a>&nbsp;•&nbsp;
-    <a href="https://zenodo.org/records/15087238">Dataset</a>&nbsp;•&nbsp;
+    <a href="https://zenodo.org/records/15087238">Dataset</a>
   </div>
 </div>
 """