Spaces:

techconspartners
/

ConversAI

Sleeping

App Files Files Community

Rauhan commited on Aug 5

Commit

2d84492

•

1 Parent(s): 7d859ca

UPDATE: QnA Functionality

Browse files

Files changed (1) hide show

app.py +4 -4

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ async def addPDFData(vectorstore: str, pdf: UploadFile = File(...)):
     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 < 400000:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:
@@ -63,7 +63,7 @@ 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 < 400000:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:
@@ -86,7 +86,7 @@ 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 < 400000:
         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 +105,7 @@ 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 < 400000:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:

     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:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:
     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:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else:
     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:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = qa, vectorstore = addQaPair.vectorstore)
     else:
     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:
         client.table("ConversAI_ChatbotInfo").update({"charactercount": str(newCount)}).eq("username", username).eq("chatbotname", chatbotname).execute()
         return addDocuments(text = text, vectorstore = vectorstore)
     else: