Update dashboard.py

dashboard.py

@@ -30,14 +30,19 @@ 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')
     ])
-    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)
 
 # --- Helpers ---
 def is_valid_email(email): return re.match(r"[^@]+@[^@]+\.[^@]+", email)
@@ -58,35 +63,17 @@ def predict_image(image):
     label = "✅ Real Image" if prediction >= 0.5 else "⚠️ Fake Image"
     return f"{label} (Confidence: {confidence:.2%})"
 
-def fetch_dashboard_stats():
-    url = f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}?select=gender"
-    res = requests.get(url, headers=headers)
-    if res.status_code != 200:
-        return "❌ Unable to fetch stats."
-    users = res.json()
-    total = len(users)
-    gender_counts = {"M": 0, "F": 0, "O": 0, "U": 0}
-    for user in users:
-        gender = user.get("gender", "U")
-        gender_counts[gender] += 1
-    return (
-        f"👥 **Total Users:** {total}\n\n"
-        f"📊 **Gender Breakdown:**\n"
-        f"- Male: {gender_counts['M']}\n"
-        f"- Female: {gender_counts['F']}\n"
-        f"- Other: {gender_counts['O']}\n"
-        f"- Unknown: {gender_counts['U']}"
-    )
-
 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,
@@ -107,71 +94,89 @@ def login_user(email, password):
     return False
 
 # --- UI ---
-with gr.Blocks(title="VerifiAI - Deepfake Detector") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as demo:
     is_logged_in = gr.State(False)
 
-    with gr.Tabs() as tabs:
-        with gr.Tab("🔐 Login") as login_tab:
-            gr.Markdown("## Welcome to VerifiAI!")
-            stats_output = gr.Markdown(visible=False)
-            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):
-                name_signup = gr.Textbox(label="Name")
-                phone_signup = gr.Textbox(label="Phone (10 digits)")
-                email_signup = gr.Textbox(label="Email")
-                gender_signup = gr.Radio(label="Gender", choices=["M", "F", "O"], value="M")
-                password_signup = gr.Textbox(label="Create Password", type="password")
-                signup_btn = gr.Button("Sign Up")
-
-        with gr.Tab("🧪 Detect Deepfake", visible=False) as detect_tab:
-            gr.Markdown("## Deepfake Detector")
-            logout_btn = gr.Button("Logout")
-            image_input = gr.Image(type="pil", label="Upload Image")
-            result = gr.Textbox(label="Prediction Result", interactive=False)
-            predict_btn = gr.Button("Predict", variant="primary")
-
-        with gr.Tab("ℹ️ About"): about.layout()
-        with gr.Tab("🌐 Community"): community.layout()
-        with gr.Tab("📘 User Guide"): user_guide.layout()
-
-    # --- Functions ---
+    LOGIN_TAB_NAME = "🔐 Login"
+    DETECT_TAB_NAME = "🧪 Detect Deepfake"
+    ABOUT_TAB_NAME = "ℹ️ About"
+    COMMUNITY_TAB_NAME = "🌐 Community"
+    GUIDE_TAB_NAME = "📘 User Guide"
+
+    with gr.Tabs(selected=LOGIN_TAB_NAME) as tabs:
+        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.")
+                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()
+
+    def update_ui_on_auth_change(logged_in_status):
+        if logged_in_status:
+            return (
+                gr.update(visible=False),
+                gr.update(visible=True),
+                gr.update(selected=DETECT_TAB_NAME),
+                gr.update(value="✅ Login successful!", visible=True)
+            )
+        else:
+            return (
+                gr.update(visible=True),
+                gr.update(visible=False),
+                gr.update(selected=LOGIN_TAB_NAME),
+                gr.update(value="", visible=False)
+            )
+
     def handle_login(email, password):
         if login_user(email, password):
-            stats = fetch_dashboard_stats()
-            return True, stats, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+            return True, gr.update(value="✅ Login successful!", visible=True)
         else:
-            return False, "", gr.update(value="❌ Invalid credentials", visible=True), gr.update(visible=False), gr.update(visible=False)
+            return False, gr.update(value="❌ Invalid email or password.", visible=True)
 
     def handle_logout():
-        return False, "", "", gr.update(visible=False), gr.update(visible=False)
+        return False, "", ""
 
     def handle_signup(name, phone, email, gender, password):
         msg = register_user(name, phone, email, gender, password)
-        return gr.update(value=msg, visible=True)
-
-    login_btn.click(
-        fn=handle_login,
-        inputs=[email_login, password_login],
-        outputs=[is_logged_in, stats_output, stats_output, detect_tab, message_output]
-    )
-
-    logout_btn.click(
-        fn=handle_logout,
-        inputs=[],
-        outputs=[is_logged_in, email_login, password_login, detect_tab, stats_output]
-    )
+        if msg.startswith("✅"):
+            return gr.update(value=msg, visible=True), "", "", "", "", "", gr.update(open=False)
+        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, tabs, message_output])
     signup_btn.click(
         fn=handle_signup,
         inputs=[name_signup, phone_signup, email_signup, gender_signup, password_signup],
-        outputs=[message_output]
+        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__":
     demo.launch()
-