Added timestamps to mongo dumps

chat_application/main.py

@@ -28,7 +28,8 @@ from duplicate_detection import duplicate_check
 from huggingface_hub import upload_folder
 from huggingface_hub import HfApi
 from huggingface_hub import login
-import os
+from datetime import datetime
+
 
 class datasetHandler():
 
@@ -36,6 +37,7 @@ class datasetHandler():
     login(hf_token)
     self.api = HfApi(token = hf_token)
     self.DB_name = mongoDB_name
+    self.timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
 
   def make_dump(self):
     try:
@@ -47,16 +49,20 @@ class datasetHandler():
 
   def upload_dump(self):
     try:
-      upload_folder(folder_path="/tmp/mongoDBContents",path_in_repo="mongoDump", repo_id="ProjectFrozone/MongoDBDumps", repo_type="dataset")
+      upload_folder(folder_path="/tmp/mongoDBContents",path_in_repo=f"mongoDump_{self.timestamp}", repo_id="ProjectFrozone/MongoDBDumps", repo_type="dataset")
       return 0
     except Exception as e:
       print(e)
       return 1
 
+  def list_items(self):
+    all_files = self.api.list_repo_files(repo_id="ProjectFrozone/MongoDBDumps", repo_type="dataset") 
+    return all_files 
+
   def dump_db(self):
     self.make_dump()
     self.upload_dump()
-# End backup code
+# End database backup code
 
 #controls
 CHAT_CONTEXT = 20 #how many messages from chat history to append to inference prompt