The login button doesn't work anymore - Follow Up Deployment

index.html

@@ -128,7 +128,10 @@
         let messageCount = 0;
         
         // Initialize
-        checkAuth();
+        document.addEventListener('DOMContentLoaded', () => {
+            checkAuth();
+            messageInput.disabled = false; // Always enable input
+        });
         
         // Event Listeners
         loginBtn.addEventListener('click', handleLogin);
@@ -161,23 +164,35 @@
             }
         }
         
-        function handleLogin() {
-            // In a real app, you would use OAuth flow
-            // For demo purposes, we'll simulate it with a prompt
-            const token = prompt("Enter your Hugging Face token:");
+        async function handleLogin() {
+            const token = prompt("Enter your Hugging Face API token (get it from https://huggingface.co/settings/tokens):");
             
-            if (token) {
-                // Simulate getting user info
+            if (!token) return;
+            
+            try {
+                // Test the token by making a simple request
+                const testClient = new HuggingFaceInference(token);
+                await testClient.textGeneration({
+                    model: "gpt2",
+                    inputs: "test"
+                }, { wait_for_model: true });
+                
                 const demoUser = {
-                    name: "Demo User",
-                    avatar: "https://ui-avatars.com/api/?name=Demo+User&background=3b82f6&color=fff"
+                    name: "Hugging Face User",
+                    avatar: "https://huggingface.co/front/assets/huggingface_logo-noborder.svg"
                 };
                 
                 localStorage.setItem('hf_token', token);
                 localStorage.setItem('hf_user', JSON.stringify(demoUser));
                 currentUser = demoUser;
+                hfClient = testClient;
+                setupAuthenticatedUI();
                 
-                setupAuthenticatedUI(token);
+                // Show success message
+                addMessage('assistant', "Successfully logged in! How can I help you today?");
+            } catch (error) {
+                console.error("Login failed:", error);
+                alert("Invalid token. Please check your token and try again.");
             }
         }
         
@@ -198,21 +213,24 @@
             messageCount = 0;
         }
         
-        function setupAuthenticatedUI(token) {
-            // Initialize HF client
-            hfClient = new HuggingFace.inference.InferenceApi(token);
-            
+        function setupAuthenticatedUI() {
+            const token = localStorage.getItem('hf_token');
+            if (token) {
+                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 = 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() {
@@ -231,39 +249,36 @@
             try {
                 isGenerating = true;
                 
-                // Get selected model
-                const model = modelSelect.value;
+                const token = localStorage.getItem('hf_token');
+                if (!token || !hfClient) {
+                    throw new Error("Not authenticated. Please login first.");
+                }
                 
-                // Prepare messages for API
-                const messages = getConversationHistory();
+                // Initialize client if not already done
+                if (!hfClient) {
+                    hfClient = new HuggingFaceInference(token);
+                }
                 
-                // Stream the response
+                const model = modelSelect.value;
                 const response = await hfClient.conversational({
                     model: model,
                     inputs: {
-                        past_user_inputs: messages.filter(m => m.role === 'user').map(m => m.content),
-                        generated_responses: messages.filter(m => m.role === 'assistant').map(m => m.content),
-                        text: message
+                        text: message,
+                        past_user_inputs: [],
+                        generated_responses: []
                     }
                 });
                 
                 updateMessage(loadingId, response.generated_text);
                 chatMessages.scrollTop = chatMessages.scrollHeight;
-                
-                // Update token count (approximate)
-                const tokens = Math.ceil(response.generated_text.length / 4);
-                tokenCounter.textContent = `${tokens} tokens`;
-                
-                // Update token count (approximate)
-                const tokens = Math.ceil(fullResponse.length / 4);
-                tokenCounter.textContent = `${tokens} tokens`;
+                tokenCounter.textContent = `${response.conversation.generated_responses[0].length} tokens`;
                 
             } catch (error) {
                 console.error("Error:", error);
-                updateMessage(loadingId, "Sorry, I encountered an error. Please try again.");
+                updateMessage(loadingId, `Error: ${error.message}`);
             } finally {
                 isGenerating = false;
-                currentStream = null;
+                sendBtn.disabled = false;
             }
         }