Update app.py

app.py

@@ -1,27 +1,27 @@
 import gradio as gr
 import os
+import shutil
+import threading
+import time
 from PIL import Image
 import numpy as np
-from utils import load_model, preprocess, verify_identity
+from utils import 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
-
+# Register user
 def register_user(name, image):
     if image is None or not name.strip():
         return "❌ Please provide a name and webcam photo."
     
-    image = Image.fromarray(image.astype("uint8"))  # Ensure PIL format
+    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."
 
+# Delete all
 def delete_all_users():
     if os.path.exists(registration_dir):
         shutil.rmtree(registration_dir)
@@ -29,17 +29,16 @@ def delete_all_users():
         return "✅ All registered users deleted."
     return "⚠️ No users to delete."
 
-# Real-time processing function for live webcam stream
+# Webcam verify
 def process_image(frame):
     if frame is None:
         return None, "No frame captured."
-
-    result = verify_identity(model, frame, registration_dir)
+    result = verify_identity(None, frame, registration_dir)
     return frame, result
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 Real-Time Siamese Face Verification")
+    gr.Markdown("## 🧠 Real-Time Face Verification (FaceNet)")
 
     with gr.Tab("👤 Register"):
         name = gr.Textbox(label="Enter Your Name")
@@ -58,14 +57,9 @@ with gr.Blocks() as demo:
             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
+def auto_cleanup(interval=600):
     while True:
         if os.path.exists(registration_dir):
             shutil.rmtree(registration_dir)
@@ -73,9 +67,7 @@ def auto_cleanup(interval=600):  # 600 seconds = 10 minutes
             print("🧹 Cleaned up registered_users/")
         time.sleep(interval)
 
-# Start cleanup thread
 threading.Thread(target=auto_cleanup, daemon=True).start()
 
-# Launch app
 if __name__ == "__main__":
     demo.launch()