Update dashboard.py

dashboard.py

@@ -93,6 +93,28 @@ def login_user(email, password):
         return bcrypt.checkpw(password.encode(), stored_hash.encode())
     return False
 
+def get_user_stats():
+    url = f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}?select=gender"
+    r = requests.get(url, headers=headers)
+    if r.status_code != 200:
+        return "Failed to fetch user stats."
+
+    users = r.json()
+    total = len(users)
+    gender_counts = {"Male": 0, "Female": 0, "Other": 0}
+    for u in users:
+        g = u.get("gender", "Other")
+        gender_counts[g] = gender_counts.get(g, 0) + 1
+
+    stats = f"""
+### 📊 Dashboard  
+- 👥 Total Users: **{total}**  
+- 🚹 Male: **{gender_counts['Male']}**  
+- 🚺 Female: **{gender_counts['Female']}**  
+- 🌈 Other: **{gender_counts['Other']}**
+"""
+    return stats
+
 # --- UI ---
 with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as demo:
     is_logged_in = gr.State(False)
@@ -114,6 +136,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
                     email_login = gr.Textbox(label="Email")
                     password_login = gr.Textbox(label="Password", type="password")
                     login_btn = gr.Button("Login", variant="primary")
+                    dashboard_stats = gr.Markdown(visible=False)
                     with gr.Accordion("New User? Click here to Sign Up", open=False) as signup_accordion:
                         name_signup = gr.Textbox(label="Name")
                         phone_signup = gr.Textbox(label="Phone (10 digits)")
@@ -138,18 +161,21 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
 
     def update_ui_on_auth_change(logged_in_status):
         if logged_in_status:
+            stats = get_user_stats()
             return (
                 gr.update(visible=False),
                 gr.update(visible=True),
                 gr.update(selected=DETECT_TAB_NAME),
-                gr.update(value="✅ Login successful!", visible=True)
+                gr.update(value="✅ Login successful!", visible=True),
+                gr.update(value=stats, visible=True)
             )
         else:
             return (
                 gr.update(visible=True),
                 gr.update(visible=False),
                 gr.update(selected=LOGIN_TAB_NAME),
-                gr.update(value="", visible=False)
+                gr.update(value="", visible=False),
+                gr.update(visible=False)
             )
 
     def handle_login(email, password):
@@ -170,7 +196,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
 
     login_btn.click(fn=handle_login, inputs=[email_login, password_login], outputs=[is_logged_in, message_output])
     logout_btn.click(fn=handle_logout, inputs=[], outputs=[is_logged_in, email_login, password_login])
-    is_logged_in.change(fn=update_ui_on_auth_change, inputs=is_logged_in, outputs=[login_tab, detect_tab, tabs, message_output])
+    is_logged_in.change(fn=update_ui_on_auth_change, inputs=is_logged_in, outputs=[login_tab, detect_tab, tabs, message_output, dashboard_stats])
     signup_btn.click(
         fn=handle_signup,
         inputs=[name_signup, phone_signup, email_signup, gender_signup, password_signup],