Update app.py

app.py

@@ -1,8 +1,6 @@
 import streamlit as st
 from home import dashboard
 from streamlit_option_menu import option_menu
-import json
-import uuid
 import pymongo
 from dotenv import load_dotenv
 import os
@@ -16,9 +14,9 @@ from pymongo.mongo_client import MongoClient
 uri = os.environ["MONGO_CONNECTION_STRING"]
 
 # Create a new client and connect to the server
-client = MongoClient(uri)
+client = MongoClient(uri, tlsCertificateKeyFile="cert.pem")
 
-db = client["Cosmo"]
+db = client["mydata"]
 
 col = db["Users"]
 
@@ -36,20 +34,6 @@ except Exception as e:
 # st.title("Authentication")
 
 
-def load_json():
-   with open("database/data.json") as file:
-        data = json.load(file)
-        return data
-        
-
-
-    
-
-def save_json():
- with open("database/data.json", "w") as file:
-        json.dump(data, file, indent=4)
-
-
 
 def login():
     st.title("Login")
@@ -57,7 +41,8 @@ def login():
     usrname = st.text_input("Username")
     password = st.text_input("Password", type="password")
     if st.button("Login", key="loginkey"):
-        for user in data["users"]:
+        users = list(col.find())
+        for user in users:
             if usrname == user["username"] and password == user["password"]:
                 st.success("Logged in as {}".format(usrname))
                 st.session_state["user"] = "logged"
@@ -79,14 +64,11 @@ def signup():
     confirm_password = st.text_input("Confirm Password", type="password")
     if st.button("Signup", key="signupkey"):
         if password == confirm_password:
-            data = json.load(open("database/data.json"))
             newuser = {
                 "username": username,
-                "password": password,
-                "id": str(uuid.uuid4())
+                "password": password
             }
-            data["users"].append(newuser)
-            json.dump(data, open("database/data.json", "w"), indent=4)
+            col.insert_one(newuser)
             st.success("Account created! You can now login.")  
             st.snow() 
             st.cache_data.clear()
@@ -100,10 +82,7 @@ def main():
 
 
 
-    if st.session_state["user"] == "logged":
-        dashboard()
-
-    elif st.session_state["user"] == "visitor":
+    if st.session_state["user"] == "visitor":
    
         option = option_menu(
             menu_title="Authentication",
@@ -118,7 +97,10 @@ def main():
             login()
         elif option == "Signup":
             signup()
-            
+    elif st.session_state["user"] == "logged":
+        dashboard()
+
+    
 
 
 main()