Shekarss commited on
Commit
e829773
·
verified ·
1 Parent(s): 30e115d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -22,6 +22,13 @@ def register_user(name, image):
22
  image.save(os.path.join(user_path, f"{name}_1.jpg"))
23
  return f"✅ User '{name}' registered successfully."
24
 
 
 
 
 
 
 
 
25
  # Real-time processing function for live webcam stream
26
  def process_image(frame):
27
  if frame is None:
@@ -41,6 +48,11 @@ with gr.Blocks() as demo:
41
  register_out = gr.Textbox(label="Status")
42
  register_btn.click(fn=register_user, inputs=[name, image], outputs=register_out)
43
 
 
 
 
 
 
44
  with gr.Tab("🔍 Live Verify"):
45
  with gr.Row():
46
  input_img = gr.Image(label="📸 Webcam", streaming=True, sources=["webcam"])
 
22
  image.save(os.path.join(user_path, f"{name}_1.jpg"))
23
  return f"✅ User '{name}' registered successfully."
24
 
25
+ def delete_all_users():
26
+ if os.path.exists(registration_dir):
27
+ shutil.rmtree(registration_dir)
28
+ os.makedirs(registration_dir)
29
+ return "✅ All registered users deleted."
30
+ return "⚠️ No users to delete."
31
+
32
  # Real-time processing function for live webcam stream
33
  def process_image(frame):
34
  if frame is None:
 
48
  register_out = gr.Textbox(label="Status")
49
  register_btn.click(fn=register_user, inputs=[name, image], outputs=register_out)
50
 
51
+ with gr.Tab("🗑️ Admin"):
52
+ delete_btn = gr.Button("Delete All Registered Users")
53
+ delete_out = gr.Textbox(label="Cleanup Status")
54
+ delete_btn.click(fn=delete_all_users, outputs=delete_out)
55
+
56
  with gr.Tab("🔍 Live Verify"):
57
  with gr.Row():
58
  input_img = gr.Image(label="📸 Webcam", streaming=True, sources=["webcam"])