app.py

import streamlit as st
from home import dashboard
from streamlit_option_menu import option_menu
import pymongo
from dotenv import load_dotenv
import os
import re


load_dotenv()

from pymongo.mongo_client import MongoClient

uri = os.environ["MONGO_CONNECTION_STRING"]

# Create a new client and connect to the server
client = MongoClient(uri, tlsCertificateKeyFile="cert.pem")

db = client["mydata"]

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)



# st.set_page_config(page_title="Authentication", page_icon=":guardsman:", layout="wide")

# st.title("Authentication")



def login():
    st.title("Login")
    data = json.load(open("database/data.json"))
    usrname = st.text_input("Username")
    password = st.text_input("Password", type="password")
    if st.button("Login", key="loginkey"):
        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"
                flag = True
                st.experimental_rerun()
            else:
                flag = False 
        if flag == False:
            st.error("Invalid username or password")
            st.stop()
    st.balloons()


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! You can now login.")  
            st.snow() 
            st.cache_data.clear()
        else:
            st.error("Passwords do not match")
        
def main():
    # st.title("Authentication")
    if "user" not in st.session_state:
        st.session_state["user"] = "visitor"



    if st.session_state["user"] == "visitor":
   
        option = option_menu(
            menu_title="Authentication",
            options=["Login", "Signup"],
            icons=["house", "activity"],
            menu_icon="cast",
            default_index=0,
            orientation="horizontal",

        )
        if option == "Login":
            login()
        elif option == "Signup":
            signup()
    elif st.session_state["user"] == "logged":
        dashboard()

    


main()