chat endpoint

app.py

@@ -18,6 +18,14 @@ async def receive_embeddings(request: Request):
     # Process the embeddings as needed
     return {"status": "OK"}
 
+@app.post("/chat")
+async def chat(request: Request):
+    data = await request.json()
+    message = data.get("message")
+    # Process the message and generate a reply
+    reply = f"Received your message: {message}"
+    return {"reply": reply}
+
 @app.get("/embeddings")
 def get_embeddings(input: str):
     result = embeddings.embed_query(input)

static/index.html

@@ -31,6 +31,14 @@
                 <p class="text-gen-output"></p>
             </form>
         </section>
+        <section id="chat">
+            <h2>Chat with the Model</h2>
+            <div id="chat-box"></div>
+            <form id="chat-form">
+                <input id="chat-input" type="text" placeholder="Type your message here..." />
+                <button type="submit">Send</button>
+            </form>
+        </section>
     </main>
 </body>
 

static/script.js

@@ -25,7 +25,9 @@ textGenForm.addEventListener('submit', async (event) => {
 });
 
 const downloadButton = document.getElementById('download-embeddings');
+
 const uploadButton = document.getElementById('upload-embeddings');
+
 const fileInput = document.getElementById('file-input');
 
 const updateDownloadButtonState = () => {
@@ -54,6 +56,10 @@ fileInput.addEventListener('change', async (event) => {
             try {
                 const embeddings = JSON.parse(e.target.result);
                 embeddingsList = embeddings; // Store uploaded embeddings in the variable
+
+                const textGenParagraph = document.querySelector('.text-gen-output');
+                textGenParagraph.textContent = JSON.stringify(embeddingsList);
+
                 updateDownloadButtonState(); // Update button state
                 // Optionally, you can send the embeddings to the server
                 await fetch('/receive-embeddings', {
@@ -75,4 +81,41 @@ downloadButton.addEventListener('click', downloadEmbeddings);
 uploadButton.addEventListener('click', uploadEmbeddings);
 
 // Initialize button state
-updateDownloadButtonState();
+updateDownloadButtonState();
+
+const chatForm = document.getElementById('chat-form');
+const chatInput = document.getElementById('chat-input');
+const chatBox = document.getElementById('chat-box');
+
+const sendMessage = async (message) => {
+    const response = await fetch('/chat', {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({ message })
+    });
+    const data = await response.json();
+    return data.reply;
+};
+
+chatForm.addEventListener('submit', async (event) => {
+    event.preventDefault();
+    const userMessage = chatInput.value;
+    if (userMessage.trim() === '') return;
+
+    const userMessageElement = document.createElement('div');
+    userMessageElement.textContent = `You: ${userMessage}`;
+    chatBox.appendChild(userMessageElement);
+
+    chatInput.value = '';
+
+    try {
+        const reply = await sendMessage(userMessage);
+        const replyMessageElement = document.createElement('div');
+        replyMessageElement.textContent = `Bot: ${reply}`;
+        chatBox.appendChild(replyMessageElement);
+    } catch (err) {
+        console.error('Error sending message:', err);
+    }
+});