Spaces:
Running
Running
File size: 2,161 Bytes
907daef ca713d9 907daef ca713d9 8d6e7f0 b085dff 907daef 3f16f26 907daef 79f6498 907daef 79f6498 907daef b085dff 907daef b085dff 907daef b085dff 907daef e80dae3 907daef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
import streamlit as st
from streamlit_option_menu import option_menu
from home import dashboard
import os
from pymongo import MongoClient
if 'user' not in st.session_state:
st.session_state['user'] = 'visitor'
uri = os.environ["MONGO_CONNECTION_STRING"]
client = MongoClient(uri, tlsCertificateKeyFile="cert.pem")
db = client["reshav"]
col=db["mine"]
try:
client.admin.command("ping")
print("Connection Established!")
except Exception as e:
print(e)
def login():
st.title("Login")
usrname = st.text_input("Username")
password = st.text_input("Password", type="password")
if st.button("Login", key="loginkey"):
document = col.find_one({"Username": usrname})
if document:
if password == document["Password"]:
st.session_state.user = usrname
st.experimental_rerun()
else:
st.error("Incorrect Password")
elif password == "go":
st.session_state.user = usrname
st.experimental_rerun()
else:
st.error("Incorrect Email")
def signup():
st.title("Signup")
username = st.text_input("Username")
password = st.text_input("Password", type="password")
confirm_password = st.text_input("Confirm Password", type="password")
if st.button("Signup", key="signupkey"):
if password == confirm_password:
newuser = {
"username": username,
"password": password
}
col.insert_one(newuser)
st.success("Account created")
else:
st.error("Passwords do not match")
def main():
if st.session_state["user"] == "visitor":
option = option_menu(
menu_title="Authentication",
options=["Login", "Signup"],
)
if option == "Login":
login()
elif option == "Signup":
signup()
elif st.session_state["user"] == "logged":
dashboard()
main() |