Update index.js

index.js

@@ -141,38 +141,25 @@ 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) => {
-                    fullResponse += text;
-                    messageContent.textContent = fullResponse;
+                    messageContent.textContent += text;
                     this.scrollToBottom();
                 }
             });
 
             // Generate response
-            await this.generator(conversationText, {
+            const output = await this.generator(this.messages, {
                 max_new_tokens: 500,
                 do_sample: false,
                 streamer: streamer,
             });
 
-            // Add the complete response to messages
-            this.messages.push({ role: "assistant", content: fullResponse });
+            const assistantMessage = output[0].generated_text.at(-1).content;
+            this.messages.push({ role: "assistant", content: assistantMessage });
             
         } catch (error) {
             console.error('Error generating response:', error);