UPDATE: QnA Functionality

app.py

@@ -3,6 +3,7 @@ from functions import *
 from PyPDF2 import PdfReader
 import pandas as pd
 from fastapi import FastAPI, File, UploadFile
+from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 from langchain_community.document_loaders import UnstructuredURLLoader
 
@@ -71,6 +72,29 @@ async def addText(vectorstore: str, text: str):
         }
 
 
+
+class AddQAPair(BaseModel):
+    vectorstore: str
+    question: str
+    answer: str
+
+
+@app.post("/addQAPair")
+async def addText(addQaPair: AddQAPair):
+    username, chatbotname = addQaPair.vectorstore.split("-")[1], addQaPair.vectorstore.split("-")[2]
+    df = pd.DataFrame(client.table("ConversAI_ChatbotInfo").select("*").execute().data)
+    currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
+    qa = f"QUESTION: {addQaPair.question}\tANSWER: {addQaPair.answer}"
+    newCount = currentCount + len(qa)
+    if newCount < 400000:
+        client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
+        return addDocuments(text = qa, vectorstore = addQaPair.vectorstore)
+    else:
+        return {
+            "output": "WEBSITE EXCEEDING LIMITS, PLEASE TRY WITH A SMALLER DOCUMENT."
+        }
+
+
 @app.post("/addWebsite")
 async def addWebsite(vectorstore: str, websiteUrls: list[str]):
     urls = websiteUrls

requirements.txt

@@ -14,6 +14,7 @@ langsmith
 lxml
 PyPDF2
 python-dotenv
+pydantic
 pandas
 sentence-transformers
 supabase