UPDATE: New Endpoints

app.py

@@ -32,6 +32,12 @@ async def login(username: str, password: str):
 
 @app.post("/newChatbot")
 async def newChatbot(chatbotName: str, username: str):
+    currentBotCount = listTables(username = username)
+    limit = client.table("ConversAI_UserConfig").select("chatbotLimit").eq("username", "rauhan").execute().data[0]["chatbotLimit"]
+    if currentBotCount < limit:
+        return {
+            "output": "CHATBOT LIMIT EXCEEDED"
+        }
     client.table("ConversAI_ChatbotInfo").insert({"username": username, "chatbotname": chatbotName}).execute()
     chatbotName = f"convai-{username}-{chatbotName}"
     return createTable(tablename = chatbotName)
@@ -47,8 +53,9 @@ async def addPDFData(vectorstore: str, pdf: UploadFile = File(...)):
     username, chatbotname = vectorstore.split("-")[1], vectorstore.split("-")[2]
     df = pd.DataFrame(client.table("ConversAI_ChatbotInfo").select("*").execute().data)
     currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
+    limit = client.table("ConversAI_UserConfig").select("tokenLimit").eq("username", username).execute().data[0]["tokenLimit"]
     newCount = currentCount + len(text)
-    if newCount < 1000000:
+    if newCount < int(limit):
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:
@@ -63,7 +70,8 @@ async def addText(vectorstore: str, text: str):
     df = pd.DataFrame(client.table("ConversAI_ChatbotInfo").select("*").execute().data)
     currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
     newCount = currentCount + len(text)
-    if newCount < 1000000:
+    limit = client.table("ConversAI_UserConfig").select("tokenLimit").eq("username", username).execute().data[0]["tokenLimit"]
+    if newCount < int(limit):
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:
@@ -86,7 +94,8 @@ async def addText(addQaPair: AddQAPair):
     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 < 1000000:
+    limit = client.table("ConversAI_UserConfig").select("tokenLimit").eq("username", username).execute().data[0]["tokenLimit"]
+    if newCount < int(limit):
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = qa, vectorstore = addQaPair.vectorstore)
     else:
@@ -105,7 +114,8 @@ async def addWebsite(vectorstore: str, websiteUrls: list[str]):
     df = pd.DataFrame(client.table("ConversAI_ChatbotInfo").select("*").execute().data)
     currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
     newCount = currentCount + len(text)
-    if newCount < 1000000:
+    limit = client.table("ConversAI_UserConfig").select("tokenLimit").eq("username", username).execute().data[0]["tokenLimit"]
+    if newCount < int(limit):
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:

functions.py

@@ -78,11 +78,8 @@ def createUser(username: str, password: str) -> None:
     try:
         userData = client.table("ConversAI_UserInfo").select("*").execute().data
         if username not in [userData[x]["username"] for x in range(len(userData))]:
-            response = (
-            client.table("ConversAI_UserInfo")
-            .insert({"username": username, "password": password})
-            .execute()
-            )
+            client.table("ConversAI_UserInfo").insert({"username": username, "password": password}).execute()
+            client.table("ConversAI_UserConfig").insert({"username": username}).execute()      
             return {
                 "output": "SUCCESS"
             }