update functions.py

app.py

@@ -265,7 +265,7 @@ async def loadPDF(vectorstore: str, pdf: UploadFile = File(...)):
         "source": source
     }
     numTokens = len(" ".join([text[x] for x in text]).translate(str.maketrans('', '', string.punctuation)).split(" "))
-    dct = json.dumps(dct, indent=1).encode("utf-8")
+    dct = json.dumps(dct, indent=1).encode("utf-8", errors = "replace")
     fileName = createDataSourceName(sourceName=source)
     response = supabase.storage.from_("ConversAI").upload(file=dct, path=f"{fileName}_data.json")
     response = (
@@ -559,7 +559,7 @@ async def loadEditedJson(loadEditedJsonConfig: LoadEditedJson):
     username, chatbotName = loadEditedJsonConfig.vectorstore.split("$")[1], loadEditedJsonConfig.vectorstore.split("$")[2]
     trackUsage(vectorstore=loadEditedJsonConfig.vectorstore, endpoint="/loadEditedJson")
     jsonData = decodeBase64(loadEditedJsonConfig.jsonData)
-    jsonData = json.dumps(jsonData, indent = 1).encode("utf-8")
+    jsonData = json.dumps(jsonData, indent = 1).encode("utf-8", errors = "replace")
     fileName = createDataSourceName(loadEditedJsonConfig.dataSourceName)
     response = supabase.storage.from_("ConversAI").upload(file=jsonData, path=f"{fileName}_data.json")
     response = (
@@ -578,19 +578,32 @@ async def loadEditedJson(loadEditedJsonConfig: LoadEditedJson):
 
 
 @app.post("/publicOrPrivate")
-async def publicOrPrivate(vectorstore: str, mode: str = "public"):
+async def publicOrPrivate(vectorstore: str, mode: str | None = None):
     trackUsage(vectorstore=vectorstore, endpoint="/publicOrPrivate")
     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
-    response = (
+    if mode == None:
+        value = (
         supabase.table("ConversAI_ChatbotInfo")
-        .update({"public/private": mode})
+        .select("isPrivate")
         .eq("user_id", username)
         .eq("chatbotname", chatbotName)
         .execute()
-    )
-    return {
-        "output": "SUCCESS"
-    }
+        ) 
+        value = value.data[0]["isPrivate"]
+        return {
+            "output": value
+        }
+    else:
+        response = (
+            supabase.table("ConversAI_ChatbotInfo")
+            .update({"isPrivate": mode})
+            .eq("user_id", username)
+            .eq("chatbotname", chatbotName)
+            .execute()
+        )
+        return {
+            "output": "SUCCESS"
+        }
 
 
 

functions.py

@@ -359,18 +359,18 @@ def extractTextFromUrlList(urls):
 def encodeToBase64(dct: dict):
     for key in dct:
         if type(dct[key]) == str:
-            dct[key] = base64.b64encode(dct[key].encode("utf-8")).decode("utf-8")
+            dct[key] = base64.b64encode(dct[key].encode("utf-8")).decode("utf-8", errors = "replace")
         elif type(dct[key]) == dict:
             dct[key] = encodeToBase64(dct[key])
     return dct
 
 
 def decodeBase64(dct: dict):
-    for key in dct:
-        if type(dct[key]) == str:
-            dct[key] = base64.b64decode(dct[key].encode("utf-8")).decode("utf-8")
-        elif type(dct[key]) == dict:
-            dct[key] = decodeBase64(dct[key])
+    if type(dct["output"]) == str:
+        dct["output"] = base64.b64decode(dct["output"].encode("utf-8")).decode("utf-8", errors = "replace")
+    else:
+        for key in dct["output"]:
+            dct["output"][key] = base64.b64decode(dct["output"][key].encode("utf-8")).decode("utf-8", errors = "replace")
     return dct