adding new modification method

pages/provide_sentences_mongo.py

@@ -166,39 +166,13 @@ def modify_line():
         st.warning("The line that you provided does not exist !")
     
     else:
+        
         st.session_state.left_sentence = sentences.loc[number, sentences.columns.tolist()[1]]
         
         st.session_state.right_sentence = sentences.loc[number, sentences.columns.tolist()[2]]
         
-        # recuperate the already saved sentences (for the moment french/wolof)
-        sentences = db_manager.load_data_frames()
-
-def update_sentences():
-    
-    global sentences
-    
-    number = st.session_state.line2
-    
-    sentence_1 = st.session_state.left_sentence.strip()
-    
-    sentence_2 = st.session_state.right_sentence.strip()
-    
-    if number > len(sentences) - 1:
-        
-        st.warning("The line that you provided does not exist !")
-    
-    elif sentence_1 == '' or sentence_2 == '':
-        
-        st.warning("You didn't provide a sentence ! Please provide before submitting.", icon= "🚨")
-    
-    else:
-        st.write(sentence_1)
-        st.write(sentence_2)
-        db_manager.update_document(number, 
-                                   {
-                                       'french': sentence_1,
-                                       'wolof': sentence_2
-                                   })
+        # add the document to update to the updated documents
+        db_manager.update_document(number)
         
         # recuperate the already saved sentences (for the moment french/wolof)
         sentences = db_manager.load_data_frames()
@@ -231,7 +205,6 @@ st.sidebar.selectbox("Wolof sentences", key="wolof_sentence", options = wolof_ex
 # add a submit button to add new sentences
 st.button('Submit', 'submit_button', on_click=add_new_sentences, help="Place the new sentences without deleting them.")
 st.button('Place', 'submit_button2', on_click=add_new_sentences2, help="Remove sentences when placing them inside the data set.")
-st.button('Update', 'submit_button3', on_click=update_sentences, help="Modify the sentences at the index indicated to 'Provide line to modify'.")
 
 st.markdown("""---""")
 

send_and_retrieve_data.ipynb

@@ -658,7 +658,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
     {
@@ -708,7 +708,7 @@
     "        \n",
     "        return results\n",
     "    \n",
-    "    def update_document(self, id: int, document: dict, collection: str = \"sentences\", update_collection: str = \"updated\"):\n",
+    "    def update_document(self, id: int, collection: str = \"sentences\", update_collection: str = \"updated\"):\n",
     "        \n",
     "        # recuperate the document to update\n",
     "        upd_sent = self.db[collection].find_one(\n",
@@ -719,15 +719,10 @@
     "            }\n",
     "        )\n",
     "        \n",
-    "        # update the document\n",
-    "        results = self.db[collection].update_one(\n",
-    "            {\n",
-    "                '_id': {\n",
-    "                    '$eq': id\n",
-    "                }\n",
-    "            },\n",
+    "        # delete the document\n",
+    "        self.db[collection].delete_one(\n",
     "            {\n",
-    "                '$set': document\n",
+    "                '_id': {'$eq': del_sent['_id']}\n",
     "            }\n",
     "        )\n",
     "        \n",

wolof_translate/utils/database_manager.py

@@ -35,7 +35,7 @@ class TranslationMongoDBManager:
         
         return results
     
-    def update_document(self, id: int, document: dict, collection: str = "sentences", update_collection: str = "updated"):
+    def update_document(self, id: int, collection: str = "sentences", update_collection: str = "updated"):
         
         # recuperate the document to update
         upd_sent = self.db[collection].find_one(
@@ -46,15 +46,10 @@ class TranslationMongoDBManager:
             }
         )
         
-        # update the document
-        results = self.db[collection].update_one(
-            {
-                '_id': {
-                    '$eq': id
-                }
-            },
+        # delete the document
+        self.db[collection].delete_one(
             {
-                '$set': document
+                '_id': {'$eq': del_sent['_id']}
             }
         )