Upload 2 files

app.py

@@ -1,39 +1,86 @@
 import streamlit as st
+from streamlit_option_menu import  option_menu
 import pymongo
 
+class ErrorMSG:
+    ERROR_MSG_NONESTR = 0
+    ERROR_MSG_NOFOUND = 1
+    ERROR_MSG_NONESTREDIT = 2
+    
+    msg = [
+        'Warning: You should input a Task above and select the right Category & Priority',
+        'Warning: No such task found, Please check your input agian',
+        'Warning: You should input a Task you want to edit and input the right information you want to edit'
+    ]
+
 uri = "mongodb+srv://123123:mb123123@cluster.c0bshep.mongodb.net/?retryWrites=true&w=majority"
 
 client = pymongo.MongoClient(uri)
 db = client["demo"]
 collection = db["class"]
 
+st.set_page_config(page_title="Marble's ToDo List",layout='wide')
+
 st.title('TODO LIST')
 st.write('This is a TODO List which allows you to insert, update, delete your task')
 
-page = st.sidebar.selectbox('Menu', ['Home', 'Insert', 'Edit', 'Delete'])
+with st.sidebar:
+    page = option_menu("ToDo List",['Home','Insert','Edit', 'Delete'],
+                         icons=['house','box-arrow-in-up','brush', 'backspace-reverse'])
 
 if page == 'Insert':
     st.header('Insert Task')
     task = st.text_input('Task')
     cate = st.selectbox('Category', ['Work', 'Study', 'Entertainment'])
+    pri = st.slider('Priority', 1, 5)
     
     if st.button('Insert'):
-        collection.insert_one({'Task':task, 'Category': cate})
-        st.success('Successfully Inserted')
+        if len(task):
+            collection.insert_one({'Task': task, 'Category': cate, 'Priority': pri})
+            st.success('Successfully Inserted')
+        else:
+            st.warning(ErrorMSG.msg[ErrorMSG.ERROR_MSG_NONESTR], icon="⚠️")
         
 elif page == 'Edit':
     st.header('Edit Task')
+    task = st.text_input('Find the Task & Category & Priority you want to Edit')
+    cate = st.selectbox('Category', ['Work', 'Study', 'Entertainment'])
+    pri = st.slider('Priority', 1, 5)
+    
+    st.header('Input your Edit')
+    taskEdit = st.text_input('Input the Task & Category & Priority you want to Edit')
+    cateEdit = st.selectbox('Edit Category', ['Work', 'Study', 'Entertainment'])
+    priEdit = st.slider('Edit Priority', 1, 5)
+    
+    if st.button('Edit'):
+        if len(task) and len(taskEdit):
+            result = collection.find_one({'Task': task, 'Category': cate, 'Priority': pri})
+            if (result):
+                collection.update_one({'Task': task, 'Category': cate, 'Priority': pri}, {'$set':{'Task': taskEdit, 'Category': cateEdit, 'Priority': priEdit}})
+                st.success('Successfully Edited')
+            else:
+                st.warning(ErrorMSG.msg[ErrorMSG.ERROR_MSG_NOFOUND], icon="⚠️")
+        else:
+            st.warning(ErrorMSG.msg[ErrorMSG.ERROR_MSG_NONESTREDIT], icon="⚠️")
     
 elif page == 'Delete':
     st.header('Delete Task')
-    task = st.text_input('Input the Task&Category you want to delete')
+    task = st.text_input('Input the Task & Category you want to Delete')
     cate = st.selectbox('Category', ['Work', 'Study', 'Entertainment'])
+    pri = st.slider('Priority', 1, 5)
     
     if st.button('Delete'):
-        collection.delete_one({'Task':task, 'Category':cate})
-        st.success('Successfully Deleted')
+        if (len(task)):
+            result = collection.find_one({'Task': task, 'Category': cate, 'Priority': pri})
+            if (result):
+                collection.delete_one({'Task': task, 'Category': cate, 'Priority': pri})
+                st.success('Successfully Deleted')
+            else:
+                st.warning(ErrorMSG.msg[ErrorMSG.ERROR_MSG_NOFOUND], icon="⚠️")
+        else:
+            st.warning(ErrorMSG.msg[ErrorMSG.ERROR_MSG_NONESTR], icon="⚠️")
 
 elif page == 'Home':
     st.header('Home Page')
-    tasks = collection.find({}, {'_id':0, })
+    tasks = collection.find({}, {'_id':0, }).sort([('Priority', -1), ('Task', 1)])
     st.table(tasks)