bugfix

frontend/src/AuthContext.jsx

@@ -6,10 +6,12 @@ const AuthContext = createContext(null);
 export const AuthProvider = ({ children }) => {
   const [user, setUser] = useState(null);
   const [loading, setLoading] = useState(true);
+  const [token, setToken] = useState(localStorage.getItem('token'));
 
   useEffect(() => {
-    const token = localStorage.getItem('token');
-    if (token) {
+    const storedToken = localStorage.getItem('token');
+    if (storedToken) {
+      setToken(storedToken);
       api.getMe()
         .then(data => setUser(data.user))
         .catch(() => {
@@ -25,12 +27,14 @@ export const AuthProvider = ({ children }) => {
   const login = async (email, password) => {
     const data = await api.login(email, password);
     setUser(data.user);
+    setToken(data.token);
     return data;
   };
 
   const register = async (email, username, password) => {
     const data = await api.register(email, username, password);
     setUser(data.user);
+    setToken(data.token);
     return data;
   };
 
@@ -38,10 +42,11 @@ export const AuthProvider = ({ children }) => {
     localStorage.removeItem('token');
     api.setToken(null);
     setUser(null);
+    setToken(null);
   };
 
   return (
-    <AuthContext.Provider value={{ user, login, register, logout, loading }}>
+    <AuthContext.Provider value={{ user, token, login, register, logout, loading }}>
       {children}
     </AuthContext.Provider>
   );