Update dashboard.py

dashboard.py

@@ -35,39 +35,33 @@ with warnings.catch_warnings():
 
 # --- DB ---
 DB_PATH = os.path.abspath("users.db")
-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,
-    NAME TEXT NOT NULL,
-    PHONE TEXT NOT NULL,
-    EMAIL TEXT UNIQUE NOT NULL,
-    GENDER TEXT,
-    PASSWORD BLOB NOT NULL
-)
-''')
-conn.commit()
-conn.close()
+with sqlite3.connect(DB_PATH) as conn:
+    cursor = conn.cursor()
+    cursor.execute('''
+    CREATE TABLE IF NOT EXISTS user_details (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        NAME TEXT NOT NULL,
+        PHONE TEXT NOT NULL,
+        EMAIL TEXT UNIQUE NOT NULL,
+        GENDER TEXT,
+        PASSWORD BLOB NOT NULL
+    )
+    ''')
+    conn.commit()
 
 # --- 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 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')
+    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 first."
+    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
@@ -75,51 +69,34 @@ def predict_image(image):
     return f"{label} (Confidence: {confidence:.2%})"
 
 def register_user(name, phone, email, password):
-    if not all([name, phone, email, 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."
-
+    if not all([name, phone, email, 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."
     try:
-        conn = sqlite3.connect(DB_PATH)
-        cursor = conn.cursor()
-        cursor.execute("SELECT * FROM user_details WHERE EMAIL = ?", (email,))
-        if cursor.fetchone():
-            return "⚠️ Email already registered."
-
-        hashed_pw = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
-        cursor.execute(
-            "INSERT INTO user_details (NAME, PHONE, EMAIL, GENDER, PASSWORD) VALUES (?, ?, ?, ?, ?)",
-            (name, phone, email, "U", hashed_pw)
-        )
-        conn.commit()
-        return "✅ Registration successful! Please log in."
+        with sqlite3.connect(DB_PATH) as conn:
+            cursor = conn.cursor()
+            cursor.execute("SELECT * FROM user_details WHERE EMAIL = ?", (email,))
+            if cursor.fetchone(): return "⚠️ Email already registered."
+            hashed_pw = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+            cursor.execute("INSERT INTO user_details (NAME, PHONE, EMAIL, GENDER, PASSWORD) VALUES (?, ?, ?, ?, ?)", (name, phone, email, "U", hashed_pw))
+            conn.commit()
+            return "✅ Registration successful! Please log in."
     except sqlite3.Error as e:
         print(f"Database error: {e}")
-        return "❌ A database error occurred. Please try again."
-    finally:
-        if conn:
-            conn.close()
+        return "❌ A database error occurred."
 
 def login_user(email, password):
-    if not email or not password:
-        return False
+    if not email or not password: return False
     try:
-        conn = sqlite3.connect(DB_PATH)
-        cursor = conn.cursor()
-        cursor.execute("SELECT PASSWORD FROM user_details WHERE EMAIL = ?", (email,))
-        result = cursor.fetchone()
-        if result and bcrypt.checkpw(password.encode('utf-8'), result[0]):
-            return True
-        return False
+        with sqlite3.connect(DB_PATH) as conn:
+            cursor = conn.cursor()
+            cursor.execute("SELECT PASSWORD FROM user_details WHERE EMAIL = ?", (email,))
+            result = cursor.fetchone()
+            if result and bcrypt.checkpw(password.encode('utf-8'), result[0]): return True
+            return False
     except sqlite3.Error as e:
         print(f"Database error during login: {e}")
         return False
-    finally:
-        if conn:
-            conn.close()
 
 # --- UI ---
 with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
@@ -136,107 +113,105 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
             with gr.Row():
                 with gr.Column(scale=1):
                     gr.Markdown("## Welcome!", "Login to access the detector, or sign up for a new account.")
-                    gr.Markdown("Our tool uses advanced AI to analyze images and determine their authenticity, helping you combat misinformation.")
                 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")
                     login_btn = gr.Button("Login", variant="primary")
-
-                    with gr.Accordion("New User? Click here to Sign Up", open=False):
-                        with gr.Group(visible=True) as signup_group:
-                            name_signup = gr.Textbox(label="Name")
-                            phone_signup = gr.Textbox(label="Phone (10 digits)")
-                            email_signup = gr.Textbox(label="Email")
-                            password_signup = gr.Textbox(label="Create Password", type="password")
-                            signup_btn = gr.Button("Sign Up")
+                    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")
+                        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")
-            gr.Markdown("Upload an image to check if it's a real photograph or an AI-generated deepfake.")
             with gr.Row():
-                with gr.Column():
-                    image_input = gr.Image(type="pil", label="Upload Image")
-                    predict_btn = gr.Button("Predict", variant="primary")
-                with gr.Column():
+                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()
 
     # --- Events ---
-    def handle_login(email, password):
-        if login_user(email, password):
+
+    # This function is now the single point for UI changes based on login state
+    def update_ui_on_auth_change(logged_in_status):
+        if logged_in_status:
+            # User has logged in
             return (
-                gr.update(value="✅ Login successful!", visible=True),
-                True,                                 # is_logged_in
-                gr.update(visible=False),             # hide login tab
-                gr.update(visible=True),              # show detect tab
-                gr.update(selected=DETECT_TAB_NAME)   # switch to detect tab
+                gr.update(visible=False),              # Hide login tab
+                gr.update(visible=True),               # Show detect tab
+                gr.update(selected=DETECT_TAB_NAME),   # Switch to detect tab
+                gr.update(value="✅ Login successful!", visible=True)
             )
         else:
+            # User has logged out
             return (
-                gr.update(value="❌ Invalid email or password.", visible=True),
-                False,
-                gr.update(), gr.update(), gr.update()
-        )
+                gr.update(visible=True),               # Show login tab
+                gr.update(visible=False),              # Hide detect tab
+                gr.update(selected=LOGIN_TAB_NAME),    # Go back to login tab
+                gr.update(value="", visible=False)
+            )
 
+    # Login button now only updates the login state and provides a message
+    def handle_login(email, password):
+        if login_user(email, password):
+            return True, gr.update(value="✅ Login successful!", visible=True)
+        else:
+            return False, gr.update(value="❌ Invalid email or password.", visible=True)
+
+    # Logout button only needs to set the state to False
+    def handle_logout():
+        return False, "", "" # logged_in=False, clear email/pass fields
 
+    # Signup remains mostly the same, but can clear inputs and close accordion
     def handle_signup(name, phone, email, password):
         msg = register_user(name, phone, email, password)
         if msg.startswith("✅"):
-            return (
-                gr.update(value=msg, visible=True),
-                "", "", "", "",  # clear inputs
-                gr.update(visible=False)  # hide signup form
-            )
+            # On success, clear inputs and close the accordion
+            return gr.update(value=msg, visible=True), "", "", "", "", gr.update(open=False)
         else:
-            return (
-                gr.update(value=msg, visible=True),
-                name, phone, email, password,
-                gr.update(visible=True)
-            )
+            # On failure, show message but don't clear or close
+            return gr.update(value=msg, visible=True), name, phone, email, password, gr.update(open=True)
 
-    def handle_logout():
-        return (
-            False,
-            gr.update(visible=True), gr.update(visible=False),
-            gr.update(selected=LOGIN_TAB_NAME),
-            "", ""
-        )
 
+    # --- Event Listeners Wiring ---
+
+    # 1. Login/Logout buttons update the `is_logged_in` state.
     login_btn.click(
         fn=handle_login,
         inputs=[email_login, password_login],
-        outputs=[message_output, is_logged_in, login_tab, detect_tab, tabs]
-    )
-
-
-    signup_btn.click(
-        fn=handle_signup,
-        inputs=[name_signup, phone_signup, email_signup, password_signup],
-        outputs=[
-            message_output,
-            name_signup, phone_signup, email_signup, password_signup,
-            signup_group
-        ]
+        outputs=[is_logged_in, message_output]
     )
 
     logout_btn.click(
         fn=handle_logout,
         inputs=[],
-        outputs=[is_logged_in, login_tab, detect_tab, tabs, email_login, password_login]
+        outputs=[is_logged_in, email_login, password_login]
     )
 
-    predict_btn.click(
-        fn=predict_image,
-        inputs=image_input,
-        outputs=result
+    # 2. A change in `is_logged_in` triggers the main UI update. THIS IS THE KEY.
+    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, password_signup],
+        outputs=[message_output, name_signup, phone_signup, email_signup, password_signup, signup_accordion]
+    )
+
+    predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()