Update dashboard.py

dashboard.py

@@ -5,14 +5,17 @@ import numpy as np
 import cv2
 from PIL import Image
 import tensorflow as tf
+
 import os
 import warnings
 import requests
 import json
 
-# --- Supabase Configuration ---
+from pages import about, community, user_guide
+
+# --- Config ---
 SUPABASE_URL = "https://fpbuhzbdtzwomjwytqul.supabase.co"
-SUPABASE_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImZwYnVoemJkdHp3b21qd3l0cXVsIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTE5NDk3NzYsImV4cCI6MjA2NzUyNTc3Nn0.oAa2TNNPQMyOGk63AOMZ7XKcwYvy5m-xoSWyvMZd6FY"
+SUPABASE_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9..."
 SUPABASE_TABLE = "user_details"
 
 headers = {
@@ -21,58 +24,67 @@ headers = {
     "Content-Type": "application/json"
 }
 
+# --- Setup ---
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 warnings.filterwarnings("ignore")
 np.seterr(all='ignore')
 
-# --- Model Loading ---
 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')
     ])
-    dummy_model.save(MODEL_PATH)
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore")
+        dummy_model.save(MODEL_PATH)
 
-deepfake_model = tf.keras.models.load_model(MODEL_PATH)
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore")
+    deepfake_model = tf.keras.models.load_model(MODEL_PATH)
 
-# --- Utils ---
+# --- Helpers ---
 def is_valid_email(email): return re.match(r"[^@]+@[^@]+\.[^@]+", email)
 def is_valid_phone(phone): return re.match(r"^[0-9]{10}$", phone)
 
 def preprocess_image(image):
-    if image.mode != 'RGB':
-        image = image.convert('RGB')
-    img = cv2.resize(np.array(image), (128, 128)).astype(np.float32) / 255.0
-    return np.expand_dims(img, axis=0)
+    if image.mode != 'RGB': image = image.convert('RGB')
+    image_arr = np.array(image)
+    image_arr = cv2.resize(image_arr, (128, 128))
+    image_arr = image_arr.astype(np.float32) / 255.0
+    return np.expand_dims(image_arr, axis=0)
 
 def predict_image(image):
-    if image is None: return "❌ Please upload an image."
-    processed = preprocess_image(image)
-    prediction = deepfake_model.predict(processed)[0][0]
+    if image is None: return "Please upload an image first."
+    preprocessed = preprocess_image(image)
+    prediction = deepfake_model.predict(preprocessed)[0][0]
     confidence = prediction if prediction >= 0.5 else 1 - prediction
     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."
-    if not is_valid_email(email): return "❌ Invalid email."
+        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 = f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}?email=eq.{email}"
-    r = requests.get(query, headers=headers)
-    if r.status_code == 200 and r.json(): return "⚠️ Email already 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(), bcrypt.gensalt()).decode()
+    hashed_pw = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode()
     data = {
-        "name": name, "phone": phone, "email": email,
-        "gender": gender, "password": hashed_pw
+        "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!" if r.status_code == 201 else "❌ Registration failed."
+    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}"
@@ -82,98 +94,118 @@ def login_user(email, password):
         return bcrypt.checkpw(password.encode(), stored_hash.encode())
     return False
 
-# --- Tab Labels ---
+# --- UI ---
 HOME_TAB_NAME = "🏠 Home"
 LOGIN_TAB_NAME = "🔐 Login"
-DETECT_TAB_NAME = "🧪 Detect"
+DETECT_TAB_NAME = "🧪 Detect Deepfake"
 ABOUT_TAB_NAME = "ℹ️ About"
 COMMUNITY_TAB_NAME = "🌐 Community"
 GUIDE_TAB_NAME = "📘 User Guide"
 
-# --- App UI ---
-with gr.Blocks(title="VerifiAI") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as demo:
     is_logged_in = gr.State(False)
-    selected_tab = gr.State(HOME_TAB_NAME)
-
-    # Each tab is a virtual view
-    with gr.Column(visible=True) as home_view:
-        gr.Markdown("# 👁️‍🗨️ Welcome to VerifiAI")
-        gr.Markdown("Upload images to detect deepfakes using AI.")
-        gr.Button("Go to Login").click(fn=lambda: LOGIN_TAB_NAME, outputs=selected_tab)
-
-    with gr.Column(visible=False) as login_view:
-        gr.Markdown("## Login")
-        message_output = gr.Markdown(visible=False)
-        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? Sign Up", open=False) as signup_accordion:
-            name_signup = gr.Textbox(label="Name")
-            phone_signup = gr.Textbox(label="Phone")
-            email_signup = gr.Textbox(label="Email")
-            gender_signup = gr.Dropdown(label="Gender", choices=["Male", "Female", "Other"])
-            password_signup = gr.Textbox(label="Password", type="password")
-            signup_btn = gr.Button("Sign Up")
-
-    with gr.Column(visible=False) as detect_view:
-        gr.Markdown("## Deepfake Detector")
-        logout_btn = gr.Button("Logout")
-        image_input = gr.Image(type="pil", label="Upload Image")
-        result = gr.Textbox(label="Result", interactive=False)
-        predict_btn = gr.Button("Predict")
-
-    with gr.Column(visible=False) as about_view:
-        gr.Markdown("## ℹ️ About VerifiAI")
-        gr.Markdown("VerifiAI helps detect deepfake images using AI models.")
-
-    with gr.Column(visible=False) as community_view:
-        gr.Markdown("## 🌐 Community")
-        gr.Markdown("Join our research and discussions on AI and ethics.")
-
-    with gr.Column(visible=False) as guide_view:
-        gr.Markdown("## 📘 User Guide")
-        gr.Markdown("1. Login or Sign Up\n2. Upload an image\n3. Click Predict")
-
-    # --- Tab Switching Logic
-    def switch_tab(tab):
-        return [gr.update(visible=(tab == t)) for t in [
-            HOME_TAB_NAME, LOGIN_TAB_NAME, DETECT_TAB_NAME,
-            ABOUT_TAB_NAME, COMMUNITY_TAB_NAME, GUIDE_TAB_NAME
-        ]]
-
-    selected_tab.change(fn=switch_tab, inputs=selected_tab,
-                        outputs=[home_view, login_view, detect_view,
-                                 about_view, community_view, guide_view])
-
-    # --- Login & Logout
+
+    with gr.Tabs() as tabs:
+        with gr.Tab(HOME_TAB_NAME, visible=True) as home_tab:
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("""
+                    <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(LOGIN_TAB_NAME, visible=True) 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.")
+                with gr.Column(scale=2):
+                    gr.Markdown("### Login or Sign Up")
+                    message_output = gr.Markdown(visible=False)
+                    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")
+                        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")
+                logout_btn = gr.Button("Logout")
+            with gr.Row():
+                image_input = gr.Image(type="pil", label="Upload Image", scale=1)
+                with gr.Column(scale=1):
+                    result = gr.Textbox(label="Prediction Result", interactive=False)
+                    predict_btn = gr.Button("Predict", variant="primary")
+
+        with gr.Tab(ABOUT_TAB_NAME): about.layout()
+        with gr.Tab(COMMUNITY_TAB_NAME): community.layout()
+        with gr.Tab(GUIDE_TAB_NAME): user_guide.layout()
+
+    gr.HTML("""
+    <style>
+    #home-markdown {
+        display: flex;
+        justify-content: center;
+        align-items: center;
+        height: 70vh;
+        text-align: center;
+    }
+    </style>
+    """)
+
+    def update_ui_on_auth_change(logged_in_status):
+        if logged_in_status:
+            return (
+                gr.update(visible=False),  # login_tab
+                gr.update(visible=True),   # detect_tab
+                gr.update(visible=False),  # home_tab
+                gr.update(value="✅ Login successful!", visible=True)
+            )
+        else:
+            return (
+                gr.update(visible=True),   # login_tab
+                gr.update(visible=False),  # detect_tab
+                gr.update(visible=True),   # home_tab
+                gr.update(value="", visible=False)
+            )
+
     def handle_login(email, password):
         if login_user(email, password):
-            return True, gr.update(value="✅ Login successful!", visible=True), DETECT_TAB_NAME
-        return False, gr.update(value="❌ Invalid credentials", visible=True), LOGIN_TAB_NAME
-
-    login_btn.click(fn=handle_login, inputs=[email_login, password_login],
-                    outputs=[is_logged_in, message_output, selected_tab])
+            return True, gr.update(value="✅ Login successful!", visible=True)
+        else:
+            return False, gr.update(value="❌ Invalid email or password.", visible=True)
 
-    logout_btn.click(fn=lambda: (False, "", "", HOME_TAB_NAME),
-                     outputs=[is_logged_in, email_login, password_login, selected_tab])
+    def handle_logout():
+        return False, "", ""
 
-    # --- Signup
     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), name, phone, email, gender, password, gr.update(open=True)
-
-    signup_btn.click(fn=handle_signup,
-                     inputs=[name_signup, phone_signup, email_signup, gender_signup, password_signup],
+        else:
+            return gr.update(value=msg, visible=True), name, phone, email, gender, password, gr.update(open=True)
+
+    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, home_tab, 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])
-
-    # --- Prediction
     predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
 
-    # --- On load
-    demo.load(fn=lambda: (False, HOME_TAB_NAME), outputs=[is_logged_in, selected_tab])
+    demo.load(lambda: False, None, [is_logged_in])
 
 if __name__ == "__main__":
     demo.launch()