Update app.py

app.py

@@ -1,13 +1,73 @@
-
 import streamlit as st
-import time
-from Auth import main
+from streamlit_option_menu import option_menu
+import json
+from Home import dashboard
+import pymongo
+
+
+
+from pymongo.mongo_client import MongoClient
+
+uri = "mongodb+srv://new-userr:[email protected]/?retryWrites=true&w=majority"
+
+# Create a new client and connect to the server
+client = MongoClient(uri)
+
+db = client["Cosmo"]
+
+col = db["Users"]
+
+# Send a ping to confirm a successful connection
+try:
+    client.admin.command('ping')
+    print("Pinged your deployment. You successfully connected to MongoDB!")
+except Exception as e:
+    print(e)
+
+
+
+
+def login():
+    st.write("Login")
+    username = st.text_input("Username")
+    password = st.text_input("Password", type="password")
+    if st.button("Login"):
+        if username in col.find():
+            if password in col.find():
+                st.session_state.userr = username
+                st.experimental_rerun()
+            else:
+                st.error("Incorrect password")
+            
+            
+def register():
+    st.write("Register")
+    username = st.text_input("Username")
+    password = st.text_input("Password", type="password")
+    data = {
+        "Username": username,
+        "Password": password
+    }
+    if st.button("Register"):
+        col.insert_one(data)
+        st.success("User created!")
+
+
+
 
-progress_text = "Operation in progress. Please wait."
-my_bar = st.progress(0, text=progress_text)
+def main():
+    if 'user' not in st.session_state:
+        st.session_state.user = None
+        
+    if st.session_state.user is None:
+        with st.sidebar:
+            selected = option_menu(None, ['Login', 'Register'])
+            if selected == 'Login':
+                login()
+            elif selected == 'Register':
+                register()
+    else:
+        dashboard()
 
-for percent_complete in range(100):
-    time.sleep(0.1)
-    my_bar.progress(percent_complete + 1, text=progress_text)
 
 main()