Update dashboard.py

dashboard.py

@@ -17,88 +17,52 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 warnings.filterwarnings("ignore")
 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)
-
-with warnings.catch_warnings():
-    warnings.simplefilter("ignore")
-    deepfake_model = tf.keras.models.load_model(MODEL_PATH)
-
-# --- DB ---
-DB_PATH = os.path.abspath("users.db")
-with sqlite3.connect(DB_PATH) as conn:
+# Database path and setup
+DB_PATH = os.path.abspath("deepfake_users.db")
+
+def init_db():
+    conn = sqlite3.connect(DB_PATH)
     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
-    )
+        CREATE TABLE IF NOT EXISTS accounts (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            name TEXT NOT NULL,
+            phone TEXT NOT NULL,
+            email TEXT UNIQUE NOT NULL,
+            gender TEXT DEFAULT 'U',
+            password TEXT NOT NULL
+        )
     ''')
     conn.commit()
+    conn.close()
+
+init_db()
 
-# --- 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')
-    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."
-    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, 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."
+# --- Authentication Functions ---
+def signup_user(name, phone, email, password):
     try:
-        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."
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        cursor.execute("INSERT INTO accounts (name, phone, email, password) VALUES (?, ?, ?, ?)",
+                       (name, phone, email, password))
+        conn.commit()
+        conn.close()
+        return "✅ Signup successful. Please log in.", False
+    except sqlite3.IntegrityError:
+        return "❌ Email already registered.", False
 
 def login_user(email, password):
-    if not email or not password: return False
-    try:
-        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
-
-# --- UI ---
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM accounts WHERE email=? AND password=?", (email, password))
+    user = cursor.fetchone()
+    conn.close()
+    if user:
+        return "✅ Login successful!", True, gr.Tabs.update(selected="🧪 Detect Deepfake"), True
+    else:
+        return "❌ Invalid credentials.", False, gr.Tabs.update(selected="🔐 Login"), False
+
+# --- Layout ---
 with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
     is_logged_in = gr.State(False)
 
@@ -113,105 +77,61 @@ 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")
+                    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")
-                        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):
+                        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.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():
-                image_input = gr.Image(type="pil", label="Upload Image", scale=1)
-                with gr.Column(scale=1):
-                    result = gr.Textbox(label="Prediction Result", interactive=False)
+                with gr.Column():
+                    image_input = gr.Image(type="pil", label="Upload Image")
                     predict_btn = gr.Button("Predict", variant="primary")
+                with gr.Column():
+                    result = gr.Textbox(label="Prediction Result", interactive=False)
 
-        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 ---
-
-    # 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(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(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("✅"):
-            # On success, clear inputs and close the accordion
-            return gr.update(value=msg, visible=True), "", "", "", "", gr.update(open=False)
-        else:
-            # 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)
-
-
-    # --- 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=[is_logged_in, message_output]
-    )
+        with gr.Tab(ABOUT_TAB_NAME):
+            gr.Markdown("### About\nThis tool detects deepfakes using AI models. Built for research and awareness.")
 
-    logout_btn.click(
-        fn=handle_logout,
-        inputs=[],
-        outputs=[is_logged_in, email_login, password_login]
-    )
+        with gr.Tab(COMMUNITY_TAB_NAME):
+            gr.Markdown("### Community\nJoin our community forum to discuss and share insights.")
+
+        with gr.Tab(GUIDE_TAB_NAME):
+            gr.Markdown("### User Guide\nStep-by-step instructions for using the deepfake detector.")
 
-    # 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]
+    # --- Function Bindings ---
+    login_btn.click(
+        fn=login_user,
+        inputs=[email_login, password_login],
+        outputs=[message_output, is_logged_in, tabs, detect_tab]
     )
 
     signup_btn.click(
-        fn=handle_signup,
+        fn=signup_user,
         inputs=[name_signup, phone_signup, email_signup, password_signup],
-        outputs=[message_output, name_signup, phone_signup, email_signup, password_signup, signup_accordion]
+        outputs=[message_output, signup_group]
     )
 
-    predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
+    logout_btn.click(
+        fn=lambda: (False, gr.Tabs.update(selected=LOGIN_TAB_NAME)),
+        inputs=None,
+        outputs=[is_logged_in, tabs]
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()