Update dashboard.py

dashboard.py

@@ -53,28 +53,41 @@ def predict_image(image):
     prediction = deepfake_model.predict(preprocessed)[0][0]
     return "✅ Real Image" if prediction >= 0.5 else "⚠️ Fake Image"
 
-# Auth logic (now using local DB connection inside each function)
 def register_user(name, phone, email, password):
+    print(f"Attempting to register: {name}, {phone}, {email}")  # ✅ Add this line
+
     if not is_valid_email(email):
+        print("Invalid email")
         return "❌ Invalid email"
     if not is_valid_phone(phone):
+        print("Invalid phone")
         return "❌ Phone must be 10 digits"
 
-    conn = sqlite3.connect(db_path)
-    cursor = conn.cursor()
-    cursor.execute("SELECT * FROM user_details WHERE EMAIL = ?", (email,))
-    if cursor.fetchone():
+    try:
+        conn = sqlite3.connect(db_path)
+        cursor = conn.cursor()
+
+        cursor.execute("SELECT * FROM user_details WHERE EMAIL = ?", (email,))
+        if cursor.fetchone():
+            print("Duplicate email")
+            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("✅ Registration successful")
+        return "✅ Registration successful! Please log in."
+
+    except sqlite3.Error as e:
+        print("❌ SQLite error:", e)
+        return f"❌ DB Error: {str(e)}"
+
+    finally:
         conn.close()
-        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()
-    conn.close()
-    return "✅ Registration successful! Please log in."
 
 def login_user(email, password):
     conn = sqlite3.connect(db_path)