Update dashboard.py

dashboard.py

@@ -9,10 +9,10 @@ import tensorflow as tf
 import os
 import warnings
 
-# Import pages
+# Import content pages
 from pages import about, community, user_guide
 
-# Suppress logs and warnings
+# Suppress TensorFlow and warning logs
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 warnings.filterwarnings("ignore")
 np.seterr(all='ignore')
@@ -20,12 +20,11 @@ np.seterr(all='ignore')
 # Load deepfake model
 deepfake_model = tf.keras.models.load_model("model_15_64.h5")
 
-# SQLite setup
+# Database setup
 db_path = os.path.abspath("users.db")
 print(f"✅ Using database at: {db_path}")
 conn = sqlite3.connect(db_path, check_same_thread=False)
 cursor = conn.cursor()
-
 cursor.execute('''
 CREATE TABLE IF NOT EXISTS user_details (
     id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -38,13 +37,14 @@ CREATE TABLE IF NOT EXISTS user_details (
 ''')
 conn.commit()
 
-# Utilities
+# Validators
 def is_valid_email(email):
     return re.match(r"[^@]+@[^@]+\.[^@]+", email)
 
 def is_valid_phone(phone):
     return re.match(r"^[0-9]{10}$", phone)
 
+# Image preprocessing
 def preprocess_image(image):
     image = np.array(image)
     image = cv2.resize(image, (128, 128))
@@ -56,78 +56,55 @@ def predict_image(image):
     prediction = deepfake_model.predict(preprocessed)[0][0]
     return "✅ Real Image" if prediction >= 0.5 else "⚠️ Fake Image"
 
+# Auth logic
 def register_user(name, phone, email, password):
     if not is_valid_email(email):
-        return "❌ Invalid email", False
+        return "❌ Invalid email"
     if not is_valid_phone(phone):
-        return "❌ Phone must be 10 digits", False
-
+        return "❌ Phone must be 10 digits"
     cursor.execute("SELECT * FROM user_details WHERE EMAIL = ?", (email,))
     if cursor.fetchone():
-        return "⚠️ Email already registered", False
-
+        return "⚠️ Email already registered"
     hashed_pw = bcrypt.hashpw(password.encode(), bcrypt.gensalt())
     cursor.execute("INSERT INTO user_details (NAME, PHONE, EMAIL, GENDER, PASSWORD) VALUES (?, ?, ?, ?, ?)",
                    (name, phone, email, "U", hashed_pw))
     conn.commit()
     print(f"✅ Registered new user: {email}")
-    return "✅ Registration successful! Please log in.", False
+    return "✅ Registration successful! Please log in."
 
 def login_user(email, password):
     cursor.execute("SELECT PASSWORD FROM user_details WHERE EMAIL = ?", (email,))
     result = cursor.fetchone()
-    if result and bcrypt.checkpw(password.encode(), result[0]):
-        return "✅ Login successful!", True
-    return "❌ Invalid credentials", False
+    if result and bcrypt.checkpw(password.encode(), result[0] if isinstance(result[0], bytes) else result[0].encode()):
+        return True
+    return False
 
-
-# Gradio App
+# Gradio UI
 with gr.Blocks() as demo:
-    session = gr.State(value=False)  # Stores login state (True/False)
+    is_logged_in = gr.State(False)
+    tab_index = gr.State(0)
 
     with gr.Tabs() as tabs:
         with gr.Tab("🔐 Login"):
             gr.Markdown("### Login or Sign Up")
 
-            status = gr.Textbox(label="Status", interactive=False)
             name = gr.Textbox(label="Name (Sign Up Only)")
             phone = gr.Textbox(label="Phone (Sign Up Only)")
             email = gr.Textbox(label="Email")
             password = gr.Textbox(label="Password", type="password")
             login_btn = gr.Button("Login")
             signup_btn = gr.Button("Sign Up")
+            message_output = gr.Markdown("", visible=False)
 
-            def handle_login(e, p):
-                msg, ok = login_user(e, p)
-                return msg, ok
-
-            def handle_signup(n, ph, e, p):
-                msg, ok = register_user(n, ph, e, p)
-                return msg, ok
-
-            login_btn.click(handle_login, [email, password], [status, session])
-            signup_btn.click(handle_signup, [name, phone, email, password], [status, session])
-
-        with gr.Tab("🧪 Detect Deepfake") as detect_tab:
-            with gr.Column(visible=False) as detection_content:
+        with gr.Tab("🧪 Detect Deepfake", visible=False) as detect_tab:
+            detect_area = gr.Column(visible=True)
+            with detect_area:
                 gr.Markdown("### Upload an Image to Detect Deepfake")
                 image_input = gr.Image(type="pil")
                 result = gr.Textbox(label="Prediction Result")
                 predict_btn = gr.Button("Predict")
                 predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
-
-            # Show warning if not logged in
-            with gr.Column(visible=True) as login_prompt:
-                warning_text = gr.Markdown("⚠️ Please login or sign up to access deepfake detection.")
-
-        def toggle_tab(logged_in):
-            return (
-                gr.update(visible=logged_in),  # detection_content
-                gr.update(visible=not logged_in),  # login_prompt
-            )
-
-        # Toggle detection tab visibility based on login state
-        session.change(fn=toggle_tab, inputs=session, outputs=[detection_content, login_prompt])
+            detect_warning = gr.Markdown("❌ Please log in to use this feature.", visible=False)
 
         with gr.Tab("ℹ️ About"):
             about.layout()
@@ -138,6 +115,54 @@ with gr.Blocks() as demo:
         with gr.Tab("📘 User Guide"):
             user_guide.layout()
 
-# Launch App
+        with gr.Tab("🚪 Logout", visible=False) as logout_tab:
+            gr.Markdown("You are logged in.")
+            logout_btn = gr.Button("Logout")
+
+    # Handlers
+    def handle_login(email, password):
+        success = login_user(email, password)
+        return (
+            "✅ Login successful!" if success else "❌ Invalid credentials",
+            success,
+            1 if success else 0,
+            gr.update(visible=success),
+            gr.update(visible=not success),
+            gr.update(visible=success),
+            gr.update(visible=True)
+        )
+
+    def handle_signup(name, phone, email, password):
+        msg = register_user(name, phone, email, password)
+        return gr.update(value=msg, visible=True)
+
+    def handle_logout():
+        return (
+            False,
+            0,
+            gr.update(visible=False),
+            gr.update(visible=True),
+            gr.update(visible=False),
+            gr.update(visible=False)
+        )
+
+    login_btn.click(
+        fn=handle_login,
+        inputs=[email, password],
+        outputs=[message_output, is_logged_in, tab_index, detect_area, detect_warning, logout_tab, message_output]
+    )
+
+    signup_btn.click(
+        fn=handle_signup,
+        inputs=[name, phone, email, password],
+        outputs=[message_output]
+    )
+
+    logout_btn.click(
+        fn=handle_logout,
+        inputs=[],
+        outputs=[is_logged_in, tab_index, detect_area, detect_warning, logout_tab, message_output]
+    )
+
 if __name__ == "__main__":
     demo.launch()