Update app.py

app.py

@@ -1,35 +1,71 @@
 import gradio as gr
 import os
-import cv2
+from PIL import Image
 import numpy as np
 from utils import load_model, preprocess, verify_identity
 
+# Load the Siamese model
 model = load_model("siamese_model.h5")
 registration_dir = "registered_users"
+os.makedirs(registration_dir, exist_ok=True)
+
+# Registration function
+from PIL import Image
 
 def register_user(name, image):
+    if image is None or not name.strip():
+        return "❌ Please provide a name and photo."
+    
+    # Convert NumPy array to PIL Image
+    image = Image.fromarray(image.astype("uint8"))
+
     user_path = os.path.join(registration_dir, name)
     os.makedirs(user_path, exist_ok=True)
     image.save(os.path.join(user_path, f"{name}_1.jpg"))
-    return f"User {name} registered successfully."
+    return f"✅ User '{name}' registered successfully."
 
-def verify(image):
-    result = verify_identity(model, image, registration_dir)
-    return result
+# Real-time processing function for live webcam stream
+def process_image(frame):
+    if frame is None:
+        return None, "No frame captured."
 
+    result = verify_identity(model, frame, registration_dir)
+    return frame, result
+
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 Siamese Face Verification")
-    with gr.Tab("Register"):
+    gr.Markdown("## 🧠 Real-Time Siamese Face Verification")
+
+    with gr.Tab("👤 Register"):
         name = gr.Textbox(label="Enter Your Name")
-        reg_img = gr.Camera(label="Click Your Photo", show_label=False)
-        reg_btn = gr.Button("Register")
-        reg_output = gr.Textbox()
-        reg_btn.click(fn=register_user, inputs=[name, reg_img], outputs=reg_output)
-    
-    with gr.Tab("Verify"):
-        ver_img = gr.Camera(label="Click Your Photo", show_label=False)
-        ver_btn = gr.Button("Verify")
-        ver_output = gr.Textbox()
-        ver_btn.click(fn=verify, inputs=ver_img, outputs=ver_output)
+        image = gr.Image(label="Capture Image")
+        register_btn = gr.Button("Register")
+        register_out = gr.Textbox(label="Status")
+        register_btn.click(fn=register_user, inputs=[name, image], outputs=register_out)
+
+    with gr.Tab("🔍 Live Verify"):
+        with gr.Row():
+            input_img = gr.Image(label="📸 Webcam", streaming=True, sources=["webcam"])
+            output_img = gr.Image(label="🧠 Output Frame")
+        result_text = gr.Textbox(label="Prediction", interactive=False)
+
+        input_img.stream(fn=process_image, inputs=input_img, outputs=[output_img, result_text])
+
+import threading
+import time
+import shutil
+
+def auto_cleanup(interval=600):  # 600 seconds = 10 minutes
+    while True:
+        if os.path.exists(registration_dir):
+            shutil.rmtree(registration_dir)
+            os.makedirs(registration_dir)
+            print("🧹 Cleaned up registered_users/")
+        time.sleep(interval)
+
+# Start cleanup thread
+threading.Thread(target=auto_cleanup, daemon=True).start()
 
-demo.launch()
+# Launch app
+if __name__ == "__main__":
+    demo.launch()