Update app.py

app.py

@@ -1,202 +1,249 @@
 import streamlit as st
 import mysql.connector
 import bcrypt
-import re
 import datetime
+import re
 import pytz
-import pandas as pd
-import plotly.express as px
 
 # MySQL Connection
-def get_database_connection():
-    connection = mysql.connector.connect(
-        host="gateway01.ap-southeast-1.prod.aws.tidbcloud.com",
-        port=4000,
-        user="37QUb7dvTn3P6E8.root",
-        password="MBAg14V0HaMdxwX0"
-    )
-    return connection
+connection = mysql.connector.connect(
+    host="gateway01.ap-southeast-1.prod.aws.tidbcloud.com",
+    port=4000,
+    user="37QUb7dvTn3P6E8.root",
+    password="MBAg14V0HaMdxwX0"
+)
 
-# Initialize database and tables
-def init_database():
-    connection = get_database_connection()
-    cursor = connection.cursor(buffered=True)
-    
-    cursor.execute("CREATE DATABASE IF NOT EXISTS EmployeePerformance")
-    cursor.execute('USE EmployeePerformance')
-    
-    # Create User_data table
-    cursor.execute('''CREATE TABLE IF NOT EXISTS User_data (
-        id INT AUTO_INCREMENT PRIMARY KEY,
-        username VARCHAR(50) UNIQUE NOT NULL,
-        password VARCHAR(255) NOT NULL,
-        email VARCHAR(255) UNIQUE NOT NULL,
-        registered_date TIMESTAMP,
-        last_login TIMESTAMP
-    )''')
-    
-    # Create Employee_data table
-    cursor.execute('''CREATE TABLE IF NOT EXISTS Employee_data (
-        id INT AUTO_INCREMENT PRIMARY KEY,
-        employee_id VARCHAR(50) UNIQUE NOT NULL,
-        name VARCHAR(100) NOT NULL,
-        age INT,
-        last_performance_score FLOAT,
-        current_performance_score FLOAT,
-        future_advancement TEXT,
-        employee_rank INT
-    )''')
-    
-    connection.commit()
-    cursor.close()
-    connection.close()
+mycursor = connection.cursor(buffered=True)
+
+mycursor.execute("CREATE DATABASE IF NOT EXISTS PerformancePrediction")
+mycursor.execute('USE PerformancePrediction')
+
+mycursor.execute('''CREATE TABLE IF NOT EXISTS User_data
+                    (id INT AUTO_INCREMENT PRIMARY KEY,
+                        username VARCHAR(50) UNIQUE NOT NULL,
+                        password VARCHAR(255) NOT NULL,
+                        email VARCHAR(255) UNIQUE NOT NULL,
+                        registered_date TIMESTAMP,
+                        last_login TIMESTAMP)''')
 
-# User authentication functions
 def username_exists(username):
-    connection = get_database_connection()
-    cursor = connection.cursor(buffered=True)
-    cursor.execute('USE EmployeePerformance')
-    cursor.execute("SELECT * FROM User_data WHERE username = %s", (username,))
-    result = cursor.fetchone() is not None
-    cursor.close()
-    connection.close()
-    return result
+    mycursor.execute("SELECT * FROM User_data WHERE username = %s", (username,))
+    return mycursor.fetchone() is not None
 
 def email_exists(email):
-    connection = get_database_connection()
-    cursor = connection.cursor(buffered=True)
-    cursor.execute('USE EmployeePerformance')
-    cursor.execute("SELECT * FROM User_data WHERE email = %s", (email,))
-    result = cursor.fetchone() is not None
-    cursor.close()
-    connection.close()
-    return result
+    mycursor.execute("SELECT * FROM User_data WHERE email = %s", (email,))
+    return mycursor.fetchone() is not None
 
 def is_valid_email(email):
     pattern = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
     return re.match(pattern, email) is not None
 
-def create_user(username, password, email):
+def create_user(username, password, email, registered_date):
     if username_exists(username):
         return 'username_exists'
+    
     if email_exists(email):
         return 'email_exists'
     
-    connection = get_database_connection()
-    cursor = connection.cursor(buffered=True)
-    cursor.execute('USE EmployeePerformance')
-    
     hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
-    registered_date = datetime.datetime.now(pytz.timezone('Asia/Kolkata'))
-    
-    cursor.execute(
+    mycursor.execute(
         "INSERT INTO User_data (username, password, email, registered_date) VALUES (%s, %s, %s, %s)",
         (username, hashed_password, email, registered_date)
     )
-    
     connection.commit()
-    cursor.close()
-    connection.close()
     return 'success'
 
 def verify_user(username, password):
-    connection = get_database_connection()
-    cursor = connection.cursor(buffered=True)
-    cursor.execute('USE EmployeePerformance')
-    
-    cursor.execute("SELECT password FROM User_data WHERE username = %s", (username,))
-    record = cursor.fetchone()
-    
+    mycursor.execute("SELECT password FROM User_data WHERE username = %s", (username,))
+    record = mycursor.fetchone()
     if record and bcrypt.checkpw(password.encode('utf-8'), record[0].encode('utf-8')):
-        cursor.execute("UPDATE User_data SET last_login = %s WHERE username = %s", 
-                       (datetime.datetime.now(pytz.timezone('Asia/Kolkata')), username))
+        mycursor.execute("UPDATE User_data SET last_login = %s WHERE username = %s", (datetime.datetime.now(pytz.timezone('Asia/Kolkata')), username))
         connection.commit()
-        cursor.close()
-        connection.close()
         return True
-    
-    cursor.close()
-    connection.close()
     return False
 
-# Employee data functions
-def get_employee_data():
-    connection = get_database_connection()
-    cursor = connection.cursor(buffered=True)
-    cursor.execute('USE EmployeePerformance')
-    
-    cursor.execute("SELECT * FROM Employee_data")
-    columns = [col[0] for col in cursor.description]
-    employees = [dict(zip(columns, row)) for row in cursor.fetchall()]
-    
-    cursor.close()
-    connection.close()
-    return employees
+def reset_password(username, new_password):
+    hashed_password = bcrypt.hashpw(new_password.encode('utf-8'), bcrypt.gensalt())
+    mycursor.execute(
+        "UPDATE User_data SET password = %s WHERE username = %s",
+        (hashed_password, username)
+    )
+    connection.commit()
 
-# Streamlit app
-def main():
-    st.set_page_config(page_title="Employee Performance Dashboard", layout="wide")
-    
-    init_database()
-    
-    if 'logged_in' not in st.session_state:
-        st.session_state.logged_in = False
-    
-    if not st.session_state.logged_in:
-        login_page()
-    else:
-        home_page()
+# Session state management
+if 'sign_up_successful' not in st.session_state:
+    st.session_state.sign_up_successful = False
+if 'login_successful' not in st.session_state:
+    st.session_state.login_successful = False
+if 'reset_password' not in st.session_state:
+    st.session_state.reset_password = False
+if 'username' not in st.session_state:
+    st.session_state.username = ''
+if 'current_page' not in st.session_state:
+    st.session_state.current_page = 'login'
 
-def login_page():
-    st.title("Login")
-    
-    username = st.text_input("Username")
-    password = st.text_input("Password", type="password")
-    
-    if st.button("Login"):
-        if verify_user(username, password):
-            st.session_state.logged_in = True
-            st.session_state.username = username
+# Page styling
+st.markdown("""
+    <style>
+    @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap');
+    body {
+        font-family: 'Roboto', sans-serif;
+        background-color: #f0f2f6;
+    }
+    .big-font {
+        font-size: 36px !important;
+        font-weight: bold;
+        color: #1E88E5;
+        margin-bottom: 30px;
+        text-align: center;
+    }
+    .stButton > button {
+        width: 100%;
+        background-color: #1E88E5;
+        color: white;
+        font-weight: bold;
+        border: none;
+        padding: 10px 0;
+        border-radius: 5px;
+        cursor: pointer;
+        transition: background-color 0.3s ease;
+    }
+    .stButton > button:hover {
+        background-color: #1565C0;
+    }
+    .stTextInput > div > div > input {
+        border-radius: 5px;
+        border: 2px solid #90CAF9;
+    }
+    .stTextInput > div > div > input:focus {
+        border-color: #1E88E5;
+        box-shadow: 0 0 0 1px #1E88E5;
+    }
+    .link-text {
+        color: #1E88E5;
+        text-align: center;
+        cursor: pointer;
+        transition: color 0.3s ease;
+    }
+    .link-text:hover {
+        color: #1565C0;
+        text-decoration: underline;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+# Login page
+def login():
+    st.markdown('<p class="big-font">Performance Prediction Login</p>', unsafe_allow_html=True)
+
+    col1, col2, col3 = st.columns([1, 2, 1])
+
+    with col2:
+        with st.form(key='login_form', clear_on_submit=True):
+            username = st.text_input(label='Username', placeholder='Enter Username')
+            password = st.text_input(label='Password', placeholder='Enter Password', type='password')
+
+            st.markdown("<br>", unsafe_allow_html=True)
+
+            if st.form_submit_button('Login'):
+                if not username or not password:
+                    st.error("Please enter all credentials")
+                elif verify_user(username, password):
+                    st.success(f"Welcome, {username}!")
+                    st.session_state.login_successful = True
+                    st.session_state.username = username
+                    st.session_state.current_page = 'home'
+                    st.experimental_rerun()
+                else:
+                    st.error("Incorrect username or password. Please try again or sign up if you don't have an account.")
+
+        if not st.session_state.get('login_successful', False):
+            st.markdown("<br>", unsafe_allow_html=True)
+            col_a, col_b = st.columns(2)
+            with col_a:
+                st.markdown("<div class='link-text'>New user?</div>", unsafe_allow_html=True)
+                if st.button('Sign Up'):
+                    st.session_state.current_page = 'sign_up'
+                    st.experimental_rerun()
+            with col_b:
+                st.markdown("<div class='link-text'>Forgot Password?</div>", unsafe_allow_html=True)
+                if st.button('Reset Password'):
+                    st.session_state.current_page = 'reset_password'
+                    st.experimental_rerun()
+
+# Sign up page
+def signup():
+    st.markdown('<p class="big-font">Sign Up for Performance Prediction</p>', unsafe_allow_html=True)
+
+    with st.form(key='signup_form', clear_on_submit=True):
+        email = st.text_input(label='Email', placeholder='Enter Your Email')
+        username = st.text_input(label='Username', placeholder='Enter Your Username')
+        password = st.text_input(label='Password', placeholder='Enter Your Password', type='password')
+        re_password = st.text_input(label='Confirm Password', placeholder='Confirm Your Password', type='password')
+
+        if st.form_submit_button('Sign Up'):
+            if not email or not username or not password or not re_password:
+                st.error("Enter all the Credentials")
+            elif len(password) <= 3:
+                st.error("Password too short")
+            elif password != re_password:
+                st.error("Passwords do not match! Please Re-enter")
+            else:
+                result = create_user(username, password, email, datetime.datetime.now(pytz.timezone('Asia/Kolkata')))
+                if result == 'success':
+                    st.success("Account created successfully!")
+                    st.session_state.sign_up_successful = True
+                    st.session_state.current_page = 'login'
+                    st.experimental_rerun()
+
+    if not st.session_state.get('sign_up_successful', False):
+        st.markdown("<br>", unsafe_allow_html=True)
+        st.markdown("<div class='link-text'>Already have an account?</div>", unsafe_allow_html=True)
+        if st.button('Login'):
+            st.session_state.current_page = 'login'
             st.experimental_rerun()
-        else:
-            st.error("Invalid username or password")
 
-def home_page():
-    st.title("Employee Performance Dashboard")
-    st.write(f"Welcome, {st.session_state.username}!")
-    
-    employees = get_employee_data()
-    
-    if not employees:
-        st.warning("No employee data available.")
-        return
-    
-    # Display employee information
-    for employee in employees:
-        st.subheader(f"Employee: {employee['name']}")
-        col1, col2, col3 = st.columns(3)
-        col1.metric("Employee ID", employee['employee_id'])
-        col2.metric("Age", employee['age'])
-        col3.metric("Rank", employee['employee_rank'])
-        
-        col4, col5 = st.columns(2)
-        col4.metric("Last Performance Score", f"{employee['last_performance_score']:.2f}")
-        col5.metric("Current Performance Score", f"{employee['current_performance_score']:.2f}")
-        
-        st.write("Future Advancement:", employee['future_advancement'])
-        st.markdown("---")
-    
-    # Performance visualization
-    st.subheader("Performance Overview")
-    df = pd.DataFrame(employees)
-    fig = px.scatter(df, x="age", y="current_performance_score", size="employee_rank", 
-                 hover_data=["name", "employee_id"], color="current_performance_score",
-                 labels={"current_performance_score": "Current Performance Score"})
-    st.plotly_chart(fig)
-    
-    if st.button("Logout"):
-        st.session_state.logged_in = False
+# Reset password page
+def reset_password_page():
+    st.markdown('<p class="big-font">Reset Password</p>', unsafe_allow_html=True)
+
+    with st.form(key='reset_password_form', clear_on_submit=True):
+        username = st.text_input(label='Username', value='', placeholder='Enter your username')
+        new_password = st.text_input(label='New Password', type='password', placeholder='Enter new password')
+        re_password = st.text_input(label='Confirm New Password', type='password', placeholder='Confirm new password')
+
+        if st.form_submit_button('Reset Password'):
+            if not username:
+                st.error("Enter your username.")
+            elif not username_exists(username):
+                st.error("Username not found. Enter a valid username.")
+            elif not new_password or not re_password:
+                st.error("Enter all the credentials.")
+            elif len(new_password) <= 3:
+                st.error("Password too short. Must be at least 4 characters.")
+            elif new_password != re_password:
+                st.error("Passwords do not match. Please re-enter.")
+            else:
+                reset_password(username, new_password)
+                st.success("Password has been reset successfully! Please login with your new password.")
+                st.session_state.current_page = 'login'
+                st.experimental_rerun()
+
+    st.markdown("<br>", unsafe_allow_html=True)
+    st.markdown("<div class='link-text'>Remember your password?</div>", unsafe_allow_html=True)
+    if st.button('Login'):
+        st.session_state.current_page = 'login'
         st.experimental_rerun()
 
-if __name__ == "__main__":
-    main()
+# Main app logic
+if st.session_state.current_page == 'login':
+    login()
+elif st.session_state.current_page == 'sign_up':
+    signup()
+elif st.session_state.current_page == 'reset_password':
+    reset_password_page()
+elif st.session_state.current_page == 'home':
+    st.write("Welcome to the Performance Prediction Dashboard!")
+    # Implement your home page / dashboard here
+else:
+    st.error("Page not found or not implemented yet.")