UPDATE: New Endpoints

app.py

@@ -45,7 +45,7 @@ async def addPDFData(vectorstore: str, pdf: UploadFile = File(...)):
         text += page.extract_text()
     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"][0]
+    currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
     newCount = currentCount + len(text)
     if newCount < 400000:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
@@ -60,7 +60,7 @@ async def addPDFData(vectorstore: str, pdf: UploadFile = File(...)):
 async def addText(vectorstore: str, text: str):
     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"][0]
+    currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
     newCount = currentCount + len(text)
     if newCount < 400000:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
@@ -79,7 +79,7 @@ async def addWebsite(vectorstore: str, websiteUrl: str):
     text = "\n\n\n\n".join([f"Metadata:\n{docs[doc].metadata} \nPage Content:\n {docs[doc].page_content}" for doc in range(len(docs))])
     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"][0]
+    currentCount = df[(df["username"] == username) & (df["chatbotname"] == chatbotname)]["charactercount"].iloc[0]
     newCount = currentCount + len(text)
     if newCount < 400000:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
@@ -115,5 +115,5 @@ async def getCount(vectorstore: str):
     username, chatbotName = chatbotName.split("-")[1], chatbotName.split("-")[2]
     df = pd.DataFrame(client.table("ConversAI_ChatbotInfo").select("*").execute().data)
     return {
-        "currentCount": df[(df['username'] == username) & (df['chatbotname'] == chatbotName)]['charactercount'][0]
+        "currentCount": df[(df['username'] == username) & (df['chatbotname'] == chatbotName)]['charactercount'].iloc[0]
         }