Update app.py

app.py

@@ -5,19 +5,20 @@ import plotly.graph_objects as go
 from sklearn.metrics.pairwise import cosine_distances
 
 # =========================
-# FACE UTILITIES
+# FACE SETUP
 # =========================
 
 face_cascade = cv2.CascadeClassifier(
     cv2.data.haarcascades + "haarcascade_frontalface_default.xml"
 )
 
-def detect(image):
+def extract_face(image):
     if image is None:
-        return image, None
+        return None, None
 
     gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     faces = face_cascade.detectMultiScale(gray, 1.3, 5)
+
     if len(faces) == 0:
         return image, None
 
@@ -31,30 +32,37 @@ def embed(face):
     return vec / np.linalg.norm(vec)
 
 # =========================
-# LOGIC
+# STEP FUNCTIONS
 # =========================
 
-def enroll(img):
-    img, face = detect(img)
+def step_detect(img):
+    img, _ = extract_face(img)
+    return img
+
+def step_enroll(img):
+    img, face = extract_face(img)
     if face is None:
         return img, None, "❌ No face detected"
-    return img, embed(face), "✅ Face enrolled"
 
-def verify(img, stored):
-    img, face = detect(img)
+    emb = embed(face)
+    return img, emb, "✅ Face enrolled"
+
+def step_verify(img, stored):
+    img, face = extract_face(img)
     if face is None or stored is None:
         return img, "❌ Missing data", None
 
     live = embed(face)
-    dist = cosine_distances([stored],[live])[0][0]
+    dist = cosine_distances([stored], [live])[0][0]
     status = "🔓 UNLOCKED" if dist < 0.35 else "🔒 DENIED"
 
     fig = go.Figure(go.Indicator(
         mode="gauge+number",
         value=dist,
-        gauge={"axis":{"range":[0,1]}},
-        title={"text": status}
+        title={"text": status},
+        gauge={"axis": {"range": [0, 1]}}
     ))
+
     return img, f"Distance: {dist:.3f}", fig
 
 # =========================
@@ -63,9 +71,9 @@ def verify(img, stored):
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     page = gr.State(0)
-    emb_state = gr.State()
+    stored_embedding = gr.State()
 
-    gr.Markdown("# 🔐 Face Unlock – How It Works")
+    gr.Markdown("# 🔐 Face Unlock — How Mobile Face ID Works")
 
     with gr.Row():
         back = gr.Button("⬅ Back")
@@ -74,44 +82,52 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # ---------- PAGE 0 ----------
     page0 = gr.Column(visible=True)
     with page0:
-        gr.Markdown("## 📸 Capture Face")
+        gr.Markdown("## 📸 Face Detection")
         cam0 = gr.Image(sources=["webcam"], type="numpy")
         out0 = gr.Image()
-        gr.Button("Detect").click(detect, cam0, out0)
+        gr.Button("Detect Face").click(step_detect, cam0, out0)
 
     # ---------- PAGE 1 ----------
     page1 = gr.Column(visible=False)
     with page1:
-        gr.Markdown("## 🧠 Enroll Face")
+        gr.Markdown("## 🧠 Face Enrollment")
         cam1 = gr.Image(sources=["webcam"], type="numpy")
         out1 = gr.Image()
         msg1 = gr.Markdown()
-        gr.Button("Enroll").click(enroll, cam1, [out1, emb_state, msg1])
+        gr.Button("Enroll Face").click(
+            step_enroll,
+            cam1,
+            [out1, stored_embedding, msg1]
+        )
 
     # ---------- PAGE 2 ----------
     page2 = gr.Column(visible=False)
     with page2:
-        gr.Markdown("## 🔓 Verify Face")
+        gr.Markdown("## 🔓 Face Verification")
         cam2 = gr.Image(sources=["webcam"], type="numpy")
         out2 = gr.Image()
         msg2 = gr.Markdown()
         gauge = gr.Plot()
-        gr.Button("Verify").click(verify, [cam2, emb_state], [out2, msg2, gauge])
+        gr.Button("Verify").click(
+            step_verify,
+            [cam2, stored_embedding],
+            [out2, msg2, gauge]
+        )
 
     # =========================
     # NAVIGATION
     # =========================
 
-    def nav(p, step):
-        p = min(2, max(0, p+step))
+    def navigate(p, step):
+        p = min(2, max(0, p + step))
         return (
             p,
-            gr.update(visible=p==0),
-            gr.update(visible=p==1),
-            gr.update(visible=p==2)
+            gr.update(visible=p == 0),
+            gr.update(visible=p == 1),
+            gr.update(visible=p == 2)
         )
 
-    back.click(nav, [page, gr.State(-1)], [page, page0, page1, page2])
-    nextb.click(nav, [page, gr.State(1)], [page, page0, page1, page2])
+    back.click(navigate, [page, gr.State(-1)], [page, page0, page1, page2])
+    nextb.click(navigate, [page, gr.State(1)], [page, page0, page1, page2])
 
 demo.launch()