Trying a different approach

chat_application/datasetHandler.py

@@ -0,0 +1,27 @@
+from huggingface_hub import upload_folder
+from huggingface_hub import HfApi
+from huggingface_hub import login
+import os
+
+class datasetHandler():
+
+  def __init__(self,hf_token,mongoDB_name):
+    login(hf_token)
+    self.api = HfApi(token = hf_token)
+    self.DB_name = mongoDB_name
+
+  def make_dump(self):
+    try:
+      os.system(f"mongodump --db {self.DB_name} --out {/tmp/mongoDBContents}")
+      return 0
+    except Exception as e:
+      print(e)
+      return 1
+    
+  def upload_dump(self):
+    try:
+      upload_folder(folder_path="/tmp/mongoDbContents",path_in_repo="mongoDump", repo_id="ProjectFrozone/MongoDBDumps", repo_type="dataset")
+      return 0
+    except Exception as e:
+      print(e)
+      return 1

chat_application/main.py

@@ -22,6 +22,7 @@ from humanizing import humanize
 from quote_removal import remove_quotes
 from weird_char_removal import remove_weird_characters
 from duplicate_detection import duplicate_check
+from datasetHandler import datasetHandler
 
 #controls
 CHAT_CONTEXT = 20 #how many messages from chat history to append to inference prompt