AhmedAdamu commited on
Commit
7999b95
·
verified ·
1 Parent(s): ba21e2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -186,27 +186,29 @@ with gr.Blocks(title="SecureFace ID – Privacy-First Recognition") as demo:
186
  status = gr.Markdown()
187
  btn.click(enroll_person, [name_in, img_in], status)
188
 
189
- with gr.Tab("Database"):
190
- gr.Markdown("### Registered People (live view)")
191
 
192
- # This component will be refreshed every time someone enrolls
193
  people_list = gr.Markdown()
194
 
195
  def refresh_database():
196
  if not os.path.exists(KNOWN_EMBS_PATH):
197
- return "### Empty database\nAdd someone using the Enroll tab"
 
198
  current_names = np.load(KNOWN_NAMES_PATH, allow_pickle=True).tolist()
199
  if len(current_names) == 0:
200
  return "### Empty database"
 
201
  text = f"**{len(current_names)} people in database:**\n\n"
202
  for i, name in enumerate(sorted(current_names), 1):
203
  text += f"{i}. **{name}**\n"
204
  return text
205
 
206
- # Show list on first load
207
  demo.load(refresh_database, outputs=people_list)
208
 
209
- # Refresh list every time someone clicks Enroll
210
  btn.click(refresh_database, outputs=people_list)
211
 
212
  demo.launch()
 
186
  status = gr.Markdown()
187
  btn.click(enroll_person, [name_in, img_in], status)
188
 
189
+ with gr.Tab("Database"):
190
+ gr.Markdown("### Registered People (live view")
191
 
192
+ # This component will refresh every time someone enrolls
193
  people_list = gr.Markdown()
194
 
195
  def refresh_database():
196
  if not os.path.exists(KNOWN_EMBS_PATH):
197
+ return "### Empty database\nAdd someone using the Enroll tab"
198
+
199
  current_names = np.load(KNOWN_NAMES_PATH, allow_pickle=True).tolist()
200
  if len(current_names) == 0:
201
  return "### Empty database"
202
+
203
  text = f"**{len(current_names)} people in database:**\n\n"
204
  for i, name in enumerate(sorted(current_names), 1):
205
  text += f"{i}. **{name}**\n"
206
  return text
207
 
208
+ # Show list on first app load
209
  demo.load(refresh_database, outputs=people_list)
210
 
211
+ # Refresh list every time someone clicks the Enroll button
212
  btn.click(refresh_database, outputs=people_list)
213
 
214
  demo.launch()