Update dashboard.py

dashboard.py

@@ -7,14 +7,21 @@ from PIL import Image
 import tensorflow as tf
 import os
 import warnings
-
+import requests
+import json
 
 from pages import about, community, user_guide
 
-# --- Supabase Config ---
+# --- Config ---
 SUPABASE_URL = "https://fpbuhzbdtzwomjwytqul.supabase.co"
-SUPABASE_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImZwYnVoemJkdHp3b21qd3l0cXVsIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTE5NDk3NzYsImV4cCI6MjA2NzUyNTc3Nn0.oAa2TNNPQMyOGk63AOMZ7XKcwYvy5m-xoSWyvMZd6FY"
-supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+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'
@@ -62,37 +69,29 @@ def register_user(name, phone, email, gender, password):
     if not is_valid_email(email): return "❌ Invalid email format."
     if not is_valid_phone(phone): return "❌ Phone must be 10 digits."
 
-    try:
-        existing = supabase.table("user_details").select("email").eq("email", email).execute()
-        if existing.data:
-            return "⚠️ Email already registered."
-
-        hashed_pw = bcrypt.hashpw(password.encode("utf-8"), bcrypt.gensalt()).decode("utf-8")
-
-        supabase.table("user_details").insert({
-            "name": name,
-            "phone": phone,
-            "email": email,
-            "gender": gender,
-            "password": hashed_pw
-        }).execute()
-
-        return "✅ Registration successful! Please log in."
-
-    except Exception as e:
-        print(f"Supabase error: {e}")
-        return "❌ A Supabase error occurred."
+    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": gender,
+        "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."
 
 def login_user(email, password):
-    if not email or not password: return False
-    try:
-        result = supabase.table("user_details").select("password").eq("email", email).execute()
-        if result.data and bcrypt.checkpw(password.encode('utf-8'), result.data[0]["password"].encode("utf-8")):
-            return True
-        return False
-    except Exception as e:
-        print(f"Supabase login error: {e}")
-        return False
+    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
 
 # --- UI ---
 with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as demo:
@@ -108,12 +107,12 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
         with gr.Tab(LOGIN_TAB_NAME) as login_tab:
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("## Welcome to VerifiAI!", "Login to access the detector, or sign up for a new account.")
+                    gr.Markdown("## Welcome!", "Login to access the detector, or sign up for a new account.")
                 with gr.Column(scale=2):
                     gr.Markdown("### Login or Sign Up")
                     message_output = gr.Markdown(visible=False)
-                    email_login = gr.Textbox(label="Email", elem_id="login_email")
-                    password_login = gr.Textbox(label="Password", type="password", elem_id="login_pass")
+                    email_login = gr.Textbox(label="Email")
+                    password_login = gr.Textbox(label="Password", type="password")
                     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")
@@ -137,8 +136,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
         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 (
@@ -174,8 +171,11 @@ 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])
-    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])
+    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]
+    )
     predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
 
 if __name__ == "__main__":