man, I'm a little tired now, it still doesn't work can you please try to make it work the login doesn't work - Follow Up Deployment

index.html

@@ -154,45 +154,65 @@
         });
         
         // Functions
-        function checkAuth() {
-            const token = localStorage.getItem('hf_token');
-            const user = localStorage.getItem('hf_user');
-            
-            if (token && user) {
-                currentUser = JSON.parse(user);
-                setupAuthenticatedUI(token);
+        async function checkAuth() {
+            try {
+                const token = localStorage.getItem('hf_token');
+                const user = localStorage.getItem('hf_user');
+                
+                if (token && user) {
+                    // Verify token is still valid
+                    const response = await fetch('https://huggingface.co/api/whoami-v2', {
+                        headers: {
+                            'Authorization': `Bearer ${token}`
+                        }
+                    });
+                    
+                    if (!response.ok) {
+                        throw new Error('Token expired or invalid');
+                    }
+                    
+                    currentUser = JSON.parse(user);
+                    setupAuthenticatedUI();
+                }
+            } catch (error) {
+                console.error("Auth check failed:", error);
+                handleLogout();
             }
         }
         
         async function handleLogin() {
-            const token = prompt("Enter your Hugging Face API token (get it from https://huggingface.co/settings/tokens):");
-            
-            if (!token) return;
-            
             try {
-                // Test the token by making a simple request
+                const token = prompt("Enter your Hugging Face API token (get it from https://huggingface.co/settings/tokens):");
+                if (!token) return;
+                
+                // Test the token with a lightweight request
                 const testClient = new HuggingFaceInference(token);
-                await testClient.textGeneration({
-                    model: "gpt2",
-                    inputs: "test"
-                }, { wait_for_model: true });
+                const response = await fetch('https://huggingface.co/api/whoami-v2', {
+                    headers: {
+                        'Authorization': `Bearer ${token}`
+                    }
+                });
                 
-                const demoUser = {
-                    name: "Hugging Face User",
-                    avatar: "https://huggingface.co/front/assets/huggingface_logo-noborder.svg"
+                if (!response.ok) {
+                    throw new Error('Invalid token');
+                }
+                
+                const userData = await response.json();
+                const user = {
+                    name: userData.name || 'Hugging Face User',
+                    avatar: userData.avatarUrl || 'https://huggingface.co/front/assets/huggingface_logo-noborder.svg'
                 };
                 
                 localStorage.setItem('hf_token', token);
-                localStorage.setItem('hf_user', JSON.stringify(demoUser));
-                currentUser = demoUser;
+                localStorage.setItem('hf_user', JSON.stringify(user));
+                currentUser = user;
                 hfClient = testClient;
                 setupAuthenticatedUI();
                 
-                // Show success message
-                addMessage('assistant', "Successfully logged in! How can I help you today?");
+                addMessage('assistant', `Hello ${user.name}! How can I help you today?`);
             } catch (error) {
-                console.error("Login failed:", error);
-                alert("Invalid token. Please check your token and try again.");
+                console.error("Login error:", error);
+                alert(`Login failed: ${error.message}. Please check your token and try again.`);
             }
         }
         
@@ -214,23 +234,26 @@
         }
         
         function setupAuthenticatedUI() {
-            const token = localStorage.getItem('hf_token');
-            if (token) {
+            try {
+                const token = localStorage.getItem('hf_token');
+                if (!token) throw new Error('No token found');
+                
                 hfClient = new HuggingFaceInference(token);
+                
+                // Update UI
+                loginBtn.classList.add('hidden');
+                userInfo.classList.remove('hidden');
+                userAvatar.src = currentUser.avatar;
+                username.textContent = currentUser.name;
+                messageInput.disabled = false;
+                sendBtn.disabled = messageInput.value.trim() === '';
+                
+                // Scroll to bottom
+                chatMessages.scrollTop = chatMessages.scrollHeight;
+            } catch (error) {
+                console.error("Authentication setup failed:", error);
+                handleLogout();
             }
-            // Update UI
-            loginBtn.classList.add('hidden');
-            userInfo.classList.remove('hidden');
-            userAvatar.src = currentUser.avatar;
-            username.textContent = currentUser.name;
-            messageInput.disabled = false;
-            sendBtn.disabled = false;
-            
-            // Show welcome message if first time
-            if (messageCount === 0) {
-                addMessage('assistant', "Hello! I'm an AI assistant. How can I help you today?");
-            }
-            
         }
         
         async function sendMessage() {