PrashanthB461 commited on
Commit
9f9b910
Β·
verified Β·
1 Parent(s): 9f61583

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -12
app.py CHANGED
@@ -141,7 +141,7 @@ class AttendanceSystem:
141
  self.next_worker_id += 1
142
 
143
  # Save face image
144
- face_pil = Image.fromarray(cv2.cvtColor(face_image, cv2.COLOR_BGR2RGB))
145
  face_pil.save(f"data/faces/{worker_id}_{worker_name}.jpg")
146
 
147
  self.save_data()
@@ -371,11 +371,12 @@ class AttendanceSystem:
371
  if not start_date or not end_date:
372
  return "Please select both start and end dates."
373
 
374
- # Convert to string format if needed
375
- if hasattr(start_date, 'strftime'):
376
- start_date = start_date.strftime('%Y-%m-%d')
377
- if hasattr(end_date, 'strftime'):
378
- end_date = end_date.strftime('%Y-%m-%d')
 
379
 
380
  # Filter records by date range
381
  filtered_records = [
@@ -569,13 +570,13 @@ def create_interface():
569
  with gr.Row():
570
  with gr.Column():
571
  gr.Markdown("#### πŸ“… Generate Report")
572
- start_date = gr.DatePicker(
573
- label="Start Date",
574
- value=date.today().replace(day=1)
575
  )
576
- end_date = gr.DatePicker(
577
- label="End Date",
578
- value=date.today()
579
  )
580
  generate_report_btn = gr.Button(
581
  "πŸ“Š Generate Report",
 
141
  self.next_worker_id += 1
142
 
143
  # Save face image
144
+ face_pil = Image.fromarray(cv2.cvtColor(face_image, cv2.COLOR_BGR2RGB)
145
  face_pil.save(f"data/faces/{worker_id}_{worker_name}.jpg")
146
 
147
  self.save_data()
 
371
  if not start_date or not end_date:
372
  return "Please select both start and end dates."
373
 
374
+ try:
375
+ # Validate date format
376
+ datetime.strptime(start_date, '%Y-%m-%d')
377
+ datetime.strptime(end_date, '%Y-%m-%d')
378
+ except ValueError:
379
+ return "Invalid date format. Please use YYYY-MM-DD."
380
 
381
  # Filter records by date range
382
  filtered_records = [
 
570
  with gr.Row():
571
  with gr.Column():
572
  gr.Markdown("#### πŸ“… Generate Report")
573
+ start_date = gr.Textbox(
574
+ label="Start Date (YYYY-MM-DD)",
575
+ value=date.today().replace(day=1).strftime('%Y-%m-%d')
576
  )
577
+ end_date = gr.Textbox(
578
+ label="End Date (YYYY-MM-DD)",
579
+ value=date.today().strftime('%Y-%m-%d')
580
  )
581
  generate_report_btn = gr.Button(
582
  "πŸ“Š Generate Report",