Update dashboard.py

dashboard.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import sqlite3
 import re
 import bcrypt
 import numpy as np
@@ -7,22 +8,9 @@ from PIL import Image
 import tensorflow as tf
 import os
 import warnings
-import requests
-import json
 
 from pages import about, community, user_guide
 
-# --- Config ---
-SUPABASE_URL = "https://fpbuhzbdtzwomjwytqul.supabase.co"
-SUPABASE_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImZwYnVoemJkdHp3b21qd3l0cXVsIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTE5NDk3NzYsImV4cCI6MjA2NzUyNTc3Nn0.oAa2TNNPQMyOGk63AOMZ7XKcwYvy5m-xoSWyvMZd6FY"
-SUPABASE_TABLE = "user_details"
-
-headers = {
-    "apikey": SUPABASE_API_KEY,
-    "Authorization": f"Bearer {SUPABASE_API_KEY}",
-    "Content-Type": "application/json"
-}
-
 # --- Setup ---
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 warnings.filterwarnings("ignore")
@@ -44,6 +32,22 @@ with warnings.catch_warnings():
     warnings.simplefilter("ignore")
     deepfake_model = tf.keras.models.load_model(MODEL_PATH)
 
+# --- DB ---
+DB_PATH = os.path.abspath("users.db")
+with sqlite3.connect(DB_PATH) as conn:
+    cursor = conn.cursor()
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS user_details (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        NAME TEXT NOT NULL,
+        PHONE TEXT NOT NULL,
+        EMAIL TEXT UNIQUE NOT NULL,
+        GENDER TEXT,
+        PASSWORD BLOB NOT NULL
+    )
+    ''')
+    conn.commit()
+
 # --- Helpers ---
 def is_valid_email(email): return re.match(r"[^@]+@[^@]+\.[^@]+", email)
 def is_valid_phone(phone): return re.match(r"^[0-9]{10}$", phone)
@@ -63,43 +67,44 @@ def predict_image(image):
     label = "✅ Real Image" if prediction >= 0.5 else "⚠️ Fake Image"
     return f"{label} (Confidence: {confidence:.2%})"
 
-def register_user(name, phone, email, password):
-    if not all([name, phone, email, password]): return "❌ All fields are required for signup."
+def register_user(name, phone, email, gender, password):
+    if not all([name, phone, email, gender, password]): return "❌ All fields are required for signup."
     if not is_valid_email(email): return "❌ Invalid email format."
     if not is_valid_phone(phone): return "❌ Phone must be 10 digits."
-
-    # Check if user exists
-    query_url = f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}?email=eq.{email}"
-    r = requests.get(query_url, headers=headers)
-    if r.status_code == 200 and len(r.json()) > 0:
-        return "⚠️ Email already registered."
-
-    hashed_pw = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode()
-    data = {
-        "name": name,
-        "phone": phone,
-        "email": email,
-        "gender": "U",
-        "password": hashed_pw
-    }
-    r = requests.post(f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}", headers=headers, data=json.dumps(data))
-    return "✅ Registration successful! Please log in." if r.status_code == 201 else "❌ Error during registration."
+    try:
+        with sqlite3.connect(DB_PATH) as conn:
+            cursor = conn.cursor()
+            cursor.execute("SELECT * FROM user_details WHERE EMAIL = ?", (email,))
+            if cursor.fetchone(): return "⚠️ Email already registered."
+            hashed_pw = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+            cursor.execute("INSERT INTO user_details (NAME, PHONE, EMAIL, GENDER, PASSWORD) VALUES (?, ?, ?, ?, ?)",
+                           (name, phone, email, gender, hashed_pw))
+            conn.commit()
+            return "✅ Registration successful! Please log in."
+    except sqlite3.Error as e:
+        print(f"Database error: {e}")
+        return "❌ A database error occurred."
 
 def login_user(email, password):
-    url = f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}?email=eq.{email}"
-    r = requests.get(url, headers=headers)
-    if r.status_code == 200 and r.json():
-        stored_hash = r.json()[0]["password"]
-        return bcrypt.checkpw(password.encode(), stored_hash.encode())
-    return False
+    if not email or not password: return False
+    try:
+        with sqlite3.connect(DB_PATH) as conn:
+            cursor = conn.cursor()
+            cursor.execute("SELECT PASSWORD FROM user_details WHERE EMAIL = ?", (email,))
+            result = cursor.fetchone()
+            if result and bcrypt.checkpw(password.encode('utf-8'), result[0]): return True
+            return False
+    except sqlite3.Error as e:
+        print(f"Database error during login: {e}")
+        return False
 
 # --- UI ---
-with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detection") as demo:
     is_logged_in = gr.State(False)
 
-    LOGIN_TAB_NAME = "🔐 Login"
-    DETECT_TAB_NAME = "🧪 Detect Deepfake"
-    ABOUT_TAB_NAME = "ℹ️ About"
+    LOGIN_TAB_NAME = "🔐 VerifiAI Login"
+    DETECT_TAB_NAME = "🧪 Run Detection"
+    ABOUT_TAB_NAME = "ℹ️ About VerifiAI"
     COMMUNITY_TAB_NAME = "🌐 Community"
     GUIDE_TAB_NAME = "📘 User Guide"
 
@@ -107,23 +112,25 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
         with gr.Tab(LOGIN_TAB_NAME) as login_tab:
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("## Welcome!", "Login to access the detector, or sign up for a new account.")
+                    gr.Markdown("## Welcome to VerifiAI!")
+                    gr.Markdown("*AI-powered truth detection for the digital age.*")
                 with gr.Column(scale=2):
-                    gr.Markdown("### Login or Sign Up")
+                    gr.Markdown("### Access VerifiAI")
                     message_output = gr.Markdown(visible=False)
-                    email_login = gr.Textbox(label="Email")
-                    password_login = gr.Textbox(label="Password", type="password")
+                    email_login = gr.Textbox(label="Email", elem_id="login_email")
+                    password_login = gr.Textbox(label="Password", type="password", elem_id="login_pass")
                     login_btn = gr.Button("Login", variant="primary")
                     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)")
                         email_signup = gr.Textbox(label="Email")
+                        gender_signup = gr.Dropdown(label="Gender", choices=["Male", "Female", "Other"])
                         password_signup = gr.Textbox(label="Create Password", type="password")
                         signup_btn = gr.Button("Sign Up")
 
         with gr.Tab(DETECT_TAB_NAME, visible=False) as detect_tab:
             with gr.Row():
-                gr.Markdown("## Deepfake Detector")
+                gr.Markdown("## VerifiAI Deepfake Detector")
                 logout_btn = gr.Button("Logout")
             with gr.Row():
                 image_input = gr.Image(type="pil", label="Upload Image", scale=1)
@@ -135,6 +142,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
         with gr.Tab(COMMUNITY_TAB_NAME): community.layout()
         with gr.Tab(GUIDE_TAB_NAME): user_guide.layout()
 
+    # --- Events ---
     def update_ui_on_auth_change(logged_in_status):
         if logged_in_status:
             return (
@@ -158,20 +166,40 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
             return False, gr.update(value="❌ Invalid email or password.", visible=True)
 
     def handle_logout():
-        return False, "", ""
+        return False, "", ""  # logged_in=False, clear email/pass fields
 
-    def handle_signup(name, phone, email, password):
-        msg = register_user(name, phone, email, password)
+    def handle_signup(name, phone, email, gender, password):
+        msg = register_user(name, phone, email, gender, password)
         if msg.startswith("✅"):
-            return gr.update(value=msg, visible=True), "", "", "", "", gr.update(open=False)
+            return gr.update(value=msg, visible=True), "", "", "", "", "", gr.update(open=False)
         else:
-            return gr.update(value=msg, visible=True), name, phone, email, password, gr.update(open=True)
+            return gr.update(value=msg, visible=True), name, phone, email, gender, password, gr.update(open=True)
+
+    # --- Event Listeners Wiring ---
+    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]
+    )
+
+    signup_btn.click(
+        fn=handle_signup,
+        inputs=[name_signup, phone_signup, email_signup, gender_signup, password_signup],
+        outputs=[message_output, name_signup, phone_signup, email_signup, gender_signup, password_signup, signup_accordion]
+    )
 
-    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])
-    signup_btn.click(fn=handle_signup, inputs=[name_signup, phone_signup, email_signup, password_signup],
-                     outputs=[message_output, name_signup, phone_signup, email_signup, password_signup, signup_accordion])
     predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
 
 if __name__ == "__main__":