Update dashboard.py

dashboard.py

@@ -14,7 +14,7 @@ from pages import about, community, user_guide
 
 # --- Config ---
 SUPABASE_URL = "https://fpbuhzbdtzwomjwytqul.supabase.co"
-SUPABASE_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImZwYnVoemJkdHp3b21qd3l0cXVsIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTE5NDk3NzYsImV4cCI6MjA2NzUyNTc3Nn0.oAa2TNNPQMyOGk63AOMZ7XKcwYvy5m-xoSWyvMZd6FY"
+SUPABASE_API_KEY = "your-api-key"
 SUPABASE_TABLE = "user_details"
 
 headers = {
@@ -30,19 +30,14 @@ np.seterr(all='ignore')
 
 MODEL_PATH = "model_15_64.h5"
 if not os.path.exists(MODEL_PATH):
-    print(f"Model file '{MODEL_PATH}' not found. Creating a dummy model for testing.")
     dummy_model = tf.keras.Sequential([
         tf.keras.layers.Input(shape=(128, 128, 3)),
         tf.keras.layers.Flatten(),
         tf.keras.layers.Dense(1, activation='sigmoid')
     ])
-    with warnings.catch_warnings():
-        warnings.simplefilter("ignore")
-        dummy_model.save(MODEL_PATH)
+    dummy_model.save(MODEL_PATH)
 
-with warnings.catch_warnings():
-    warnings.simplefilter("ignore")
-    deepfake_model = tf.keras.models.load_model(MODEL_PATH)
+deepfake_model = tf.keras.models.load_model(MODEL_PATH)
 
 # --- Helpers ---
 def is_valid_email(email): return re.match(r"[^@]+@[^@]+\.[^@]+", email)
@@ -63,17 +58,39 @@ 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, 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."
+
+    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):
+    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:
-    # Custom CSS for centering content
-    gr.HTML("""
-    <style>
-        .home-content {
-            text-align: center;
-            margin-top: 100px;
-        }
-    </style>
-    """)
+    is_logged_in = gr.State(False)
 
     HOME_TAB_NAME = "🏠 Home"
     DETECT_TAB_NAME = "🧪 Detect Deepfake"
@@ -81,22 +98,21 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
     COMMUNITY_TAB_NAME = "🌐 Community"
     GUIDE_TAB_NAME = "📘 User Guide"
 
-    tabs = gr.Tabs()
-
-    with tabs:
+    with gr.Tabs(selected=HOME_TAB_NAME) as tabs:
+        # --- Home Tab (Intro Page) ---
         with gr.Tab(HOME_TAB_NAME) as home_tab:
-            with gr.Row(align="center", justify="center"):
+            with gr.Row():
                 with gr.Column():
                     gr.Markdown("""
-                    # 👁️‍🗨️ Welcome to VerifiAI  
-                    Your trusted assistant for detecting deepfakes in images using AI.
-
-                    🔍 Upload images, analyze authenticity, and learn how deepfakes work.
-
-                    👉 Use the tabs above to get started.
-                    """, elem_classes="home-content")
-
-        with gr.Tab(DETECT_TAB_NAME) as detect_tab:
+                    <div class="home-content">
+                        <h1>👁️‍🗨️ Welcome to VerifiAI</h1>
+                        <p>Your trusted assistant for detecting deepfakes in images using AI.</p>
+                        <p>🔍 Upload images, analyze authenticity, and learn how deepfakes work.</p>
+                        <p>👉 Use the tabs above to get started.</p>
+                    </div>
+                    """, elem_id="home-markdown")
+
+        with gr.Tab(DETECT_TAB_NAME, visible=True) as detect_tab:
             with gr.Row():
                 gr.Markdown("## Deepfake Detector")
                 logout_btn = gr.Button("Logout")
@@ -110,11 +126,26 @@ 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()
 
-    # Ensure Home tab is selected on load
-    demo.load(fn=lambda: gr.update(selected=HOME_TAB_NAME), outputs=tabs)
+    # --- CSS Styling ---
+    gr.HTML("""
+    <style>
+    #home-markdown {
+        display: flex;
+        justify-content: center;
+        align-items: center;
+        height: 70vh;
+        text-align: center;
+    }
+    </style>
+    """)
+
+    # --- Callbacks ---
+    def handle_logout():
+        return False, "", ""
 
     predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
-
+    logout_btn.click(fn=handle_logout, inputs=[], outputs=[is_logged_in, image_input, result])
+    demo.load(fn=lambda: False, outputs=is_logged_in)
 
 if __name__ == "__main__":
     demo.launch()