Create main.js

main.js

@@ -0,0 +1,38 @@
+import initWebLLM from './Webllm/engine.js';
+
+const chatBox = document.getElementById("chat-box");
+const userInput = document.getElementById("user-input");
+const sendButton = document.getElementById("send-button");
+
+let engine = null;
+
+(async () => {
+  engine = await initWebLLM({
+    modelUrl: "./distilgpt2-q4_k_s.bin",
+    tokenizerUrl: "./Webllm/tokenizer/"
+  });
+})();
+
+function addMessage(sender, text) {
+  const msg = document.createElement("div");
+  msg.className = `message ${sender}`;
+  msg.innerText = text;
+  chatBox.appendChild(msg);
+  chatBox.scrollTop = chatBox.scrollHeight;
+}
+
+async function getBotResponse(input) {
+  if (!engine) return "Loading model...";
+  return await engine.chat(input);
+}
+
+sendButton.onclick = async () => {
+  const text = userInput.value.trim();
+  if (!text) return;
+
+  addMessage("user", text);
+  userInput.value = "";
+
+  const response = await getBotResponse(text);
+  addMessage("bot", response);
+};