sending insertion method

pages/provide_sentences_mongo.py

@@ -90,13 +90,14 @@ def add_new_sentences():
     
     else:
         
-        
-        
-        ## save the result
-        sentences.to_csv('wolof_translate/data/sentences/wolof_french.csv', index=False)
+        # insert the new sentences
+        db_manager.insert_document({
+            'french': sentence_1,
+            'wolof': sentence_2
+        })
         
         # recuperate the already saved sentences (for the moment french/wolof)
-        sentences = pd.read_csv('wolof_translate/data/sentences/wolof_french.csv')
+        sentences = db_manager.load_data_frames()
         
         # add the last position to delete and modify
         st.session_state.line1 = len(sentences) - 1

send_and_retrieve_data.ipynb

@@ -658,18 +658,19 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 132,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Overwriting wolof-translate/wolof_translate/utils/database_manager.py\n"
+      "Overwriting wolof_translate/utils/database_manager.py\n"
      ]
     }
    ],
    "source": [
+    "%%writefile wolof_translate/utils/database_manager.py\n",
     "from pymongo.mongo_client import MongoClient\n",
     "from pymongo.server_api import ServerApi\n",
     "import pandas as pd\n",
@@ -693,7 +694,7 @@
     "    \n",
     "    def insert_document(self, document: dict, collection: str = \"sentences\"):\n",
     "        \n",
-    "        assert '_id' in document\n",
+    "        assert not '_id' in document\n",
     "        \n",
     "        # get the id of the last sentence (recuperate the max id and add 1 to it)\n",
     "        max_id = self.get_max_id(collection)\n",

wolof_translate/utils/database_manager.py

@@ -21,7 +21,7 @@ class TranslationMongoDBManager:
     
     def insert_document(self, document: dict, collection: str = "sentences"):
         
-        assert '_id' in document
+        assert not '_id' in document
         
         # get the id of the last sentence (recuperate the max id and add 1 to it)
         max_id = self.get_max_id(collection)
@@ -35,7 +35,16 @@ class TranslationMongoDBManager:
         
         return results
     
-    def update_document(self, id: int, document: dict, collection: str = "sentences"):
+    def update_document(self, id: int, document: dict, collection: str = "sentences", update_collection: str = "updated"):
+        
+        # recuperate the document to update
+        upd_sent = self.db[collection].find_one(
+            {
+                '_id': {
+                    '$eq': id
+                }
+            }
+        )
         
         # update the document
         results = self.db[collection].update_one(
@@ -49,9 +58,16 @@ class TranslationMongoDBManager:
             }
         )
         
+        # add the sentences to the deleted sentences
+        upd_sent['_id'] = len(list(self.db[update_collection].find()))
+        
+        results = self.db[update_collection].insert_one(
+            upd_sent
+        )
+        
         return results
     
-    def delete_document(self, id: int, document: dict, collection: str = "sentences", del_collection: str = "deleted"):
+    def delete_document(self, id: int, collection: str = "sentences", del_collection: str = "deleted"):
         
         # recuperate the document to delete
         del_sent = self.db[collection].find_one(
@@ -70,7 +86,7 @@ class TranslationMongoDBManager:
         )
         
         # add the sentences to the deleted sentences
-        del_sent['_id'] = len(list(deleted.find()))
+        del_sent['_id'] = len(list(self.db[del_collection].find()))
         
         results = self.db[del_collection].insert_one(
             del_sent