UPDATE: activity log

app.py

@@ -252,6 +252,7 @@ async def newChatbot(chatbotName: str, username: str):
 
 @app.post("/loadPDF")
 async def loadPDF(vectorstore: str, pdf: UploadFile = File(...)):
+    trackUsage(vectorstore=vectorstore, endpoint="/loadPDF")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     source = pdf.filename
     pdf = await pdf.read()
@@ -285,6 +286,7 @@ async def loadPDF(vectorstore: str, pdf: UploadFile = File(...)):
 
 @app.post("/loadImagePDF")
 async def loadImagePDF(vectorstore: str, pdf: UploadFile = File(...)):
+    trackUsage(vectorstore=vectorstore, endpoint="/loadImagePDF")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     source = pdf.filename
     pdf = await pdf.read()
@@ -319,6 +321,7 @@ class AddText(BaseModel):
 
 @app.post("/loadText")
 async def loadText(addTextConfig: AddText):
+    trackUsage(vectorstore=vectorstore, endpoint="/loadText")
     vectorstore, text = addTextConfig.vectorstore, addTextConfig.text
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     text = cleanText(text = text)
@@ -379,6 +382,7 @@ class LoadWebsite(BaseModel):
 @app.post("/loadWebURLs")
 async def loadWebURLs(loadWebsite: LoadWebsite):
     vectorstore, urls, source = loadWebsite.vectorstore, loadWebsite.urls, loadWebsite.source
+    trackUsage(vectorstore=vectorstore, endpoint="/loadWebURLs")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     text = extractTextFromUrlList(urls=urls)
     dct = {
@@ -406,6 +410,7 @@ async def loadWebURLs(loadWebsite: LoadWebsite):
 
 @app.post("/answerQuery")
 async def answerQuestion(request: Request, query: str, vectorstore: str, llmModel: str = "llama3-70b-8192"):
+    trackUsage(vectorstore=vectorstore, endpoint="/answerQuery")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     output = answerQuery(query=query, vectorstore=vectorstore, llmModel=llmModel)
     ip_address = request.client.host
@@ -424,6 +429,7 @@ async def answerQuestion(request: Request, query: str, vectorstore: str, llmMode
 
 @app.post("/deleteChatbot")
 async def deleteChatbot(vectorstore: str):
+    trackUsage(vectorstore=vectorstore, endpoint="/deleteChatbot")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     supabase.table('ConversAI_ChatbotInfo').delete().eq('user_id', username).eq('chatbotname', chatbotName).execute()
     return deleteTable(tableName=vectorstore)
@@ -459,6 +465,7 @@ class YtTranscript(BaseModel):
 @app.post("/loadYoutubeTranscript")
 async def loadYoutubeTranscript(ytTranscript: YtTranscript):
     vectorstore, urls = ytTranscript.vectorstore, ytTranscript.urls
+    trackUsage(vectorstore=vectorstore, endpoint="/loadYoutubeTranscript")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     text = getTranscript(urls=urls)
     dct = {
@@ -507,6 +514,7 @@ async def analyzeAndAnswer(query: str, file: UploadFile = File(...)):
 
 @app.post("/getChatHistory")
 async def chatHistory(vectorstore: str):
+    trackUsage(vectorstore=vectorstore, endpoint="/getChatHistory")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     response = supabase.table("ConversAI_ChatHistory").select("timestamp", "question", "response").eq("username",
                                                                                                       username).eq(
@@ -516,6 +524,7 @@ async def chatHistory(vectorstore: str):
 
 @app.post("/listChatbotSources")
 async def listChatbotSources(vectorstore: str):
+    trackUsage(vectorstore=vectorstore, endpoint="/listChatbotSources")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     result = supabase.table("ConversAI_ChatbotDataSources").select("*").eq("username", username).eq("chatbotName",
                                                                                                     chatbotName).execute().data
@@ -523,7 +532,8 @@ async def listChatbotSources(vectorstore: str):
 
 
 @app.post("/deleteChatbotSource")
-async def deleteChatbotSource(dataSourceName: str):
+async def deleteChatbotSource(vectorstore: str, dataSourceName: str):
+    trackUsage(vectorstore=vectorstore, endpoint="/deleteChatbotSource")
     response = supabase.table("ConversAI_ChatbotDataSources").delete().eq("dataSourceName", dataSourceName).execute()
     response = supabase.storage.from_('ConversAI').remove(f"{dataSourceName}_data.json")
     return {
@@ -541,6 +551,7 @@ class LoadEditedJson(BaseModel):
 @app.post("/loadEditedJson")
 async def loadEditedJson(loadEditedJsonConfig: LoadEditedJson):
     username, chatbotName = loadEditedJsonConfig.vectorstore.split("$")[1], loadEditedJsonConfig.vectorstore.split("$")[2]
+    trackUsage(vectorstore=loadEditedJsonConfig.vectorstore, endpoint="/loadEditedJson")
     jsonData = json.dumps(loadEditedJsonConfig.jsonData, indent = 1).encode("utf-8")
     fileName = createDataSourceName(loadEditedJsonConfig.dataSourceName)
     response = supabase.storage.from_("ConversAI").upload(file=jsonData, path=f"{fileName}_data.json")
@@ -561,6 +572,7 @@ async def loadEditedJson(loadEditedJsonConfig: LoadEditedJson):
 
 @app.post("/publicOrPrivate")
 async def publicOrPrivate(vectorstore: str, mode: str = "public"):
+    trackUsage(vectorstore=vectorstore, endpoint="/publicOrPrivate")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
     response = (
         supabase.table("ConversAI_ChatbotInfo")
@@ -583,6 +595,7 @@ class TrainChatbot(BaseModel):
 @app.post("/trainChatbot")
 async def trainChatbot(trainChatbotConfig: TrainChatbot):
     vectorstore, UrlSources = trainChatbotConfig.vectorstore, trainChatbotConfig.urls
+    trackUsage(vectorstore=vectorstore, endpoint="/trainChatbot")
     texts = []
     sources = []
     fileTypes = [supabase.table("ConversAI_ChatbotDataSources").select("sourceEndpoint").eq("sourceContentURL",

functions.py

@@ -368,4 +368,8 @@ def createDataSourceName(sourceName):
     while True:
       sourceName = sourceName + "-" + str(i)
       return createDataSourceName(sourceName)
-    
+    
+
+def trackUsage(vectorstore: str, endpoint: str):
+    username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
+    client.table("ConversAI_ActivityLog").insert({"username": username, "chatbotName": chatbotName, "endpointUsed": endpoint}).execute()