Update main.py

main.py

@@ -120,8 +120,14 @@ def login():
         # Check if Google user exists in the database
         user = c.execute("SELECT * FROM users WHERE google_id=?", (google_id,)).fetchone()
         if user:
+            # Retrieve data_filled from user_details table
+            user_details = c.execute("SELECT data_filled FROM user_details WHERE user_id=?", (user[0],)).fetchone()
+            if user_details:
+                data_filled = user_details[0]
+            else:
+                data_filled = False  # Default value if data not found
             access_token = create_access_token(identity=user[1], expires_delta=False)
-            return jsonify({"access_token": access_token, "data_filled": user[2]}), 200
+            return jsonify({"access_token": access_token, "data_filled": data_filled}), 200
         else:
             return jsonify({"message": "User not found"}), 404
 
@@ -130,13 +136,21 @@ def login():
         return jsonify({"message": "Missing username or password"}), 400
 
     user = c.execute("SELECT * FROM users WHERE username=?", (username,)).fetchone()
+    print(user)
     if user and check_password_hash(user[2], password):
+        # Retrieve data_filled from user_details table
+        user_details = c.execute("SELECT data_filled FROM user_details WHERE user_id=?", (user[0],)).fetchone()
+        if user_details:
+            data_filled = user_details[0]
+        else:
+            data_filled = False  # Default value if data not found
         access_token = create_access_token(identity=username, expires_delta=False)
-        return jsonify({"access_token": access_token, "data_filled": user[2]}), 200
+        return jsonify({"access_token": access_token, "data_filled": data_filled}), 200
     else:
         return jsonify({"message": "Invalid username or password"}), 401
 
 
+
 @app.route('/user_details', methods=['POST'])
 @jwt_required()
 def add_user_details():