Upload index.js with huggingface_hub

index.js

@@ -141,25 +141,38 @@ class Chatbot {
             const responseContainer = this.addMessage('', 'assistant', true);
             const messageContent = responseContainer.querySelector('.message-content');
             
+            // Build the conversation context
+            let conversationText = "";
+            for (const msg of this.messages) {
+                if (msg.role === "user") {
+                    conversationText += `User: ${msg.content}\n`;
+                } else if (msg.role === "assistant") {
+                    conversationText += `Assistant: ${msg.content}\n`;
+                }
+            }
+            conversationText += "Assistant: ";
+
             // Custom streamer to update UI
+            let fullResponse = "";
             const streamer = new TextStreamer(this.generator.tokenizer, {
                 skip_prompt: true,
                 skip_special_tokens: true,
                 callback_function: (text) => {
-                    messageContent.textContent += text;
+                    fullResponse += text;
+                    messageContent.textContent = fullResponse;
                     this.scrollToBottom();
                 }
             });
 
             // Generate response
-            const output = await this.generator(this.messages, {
+            await this.generator(conversationText, {
                 max_new_tokens: 500,
                 do_sample: false,
                 streamer: streamer,
             });
 
-            const assistantMessage = output[0].generated_text.at(-1).content;
-            this.messages.push({ role: "assistant", content: assistantMessage });
+            // Add the complete response to messages
+            this.messages.push({ role: "assistant", content: fullResponse });
             
         } catch (error) {
             console.error('Error generating response:', error);