Update app.py

app.py

@@ -87,30 +87,30 @@ def login_page():
     # Add custom styling and an image at the top
     customize_login_page()
     st.markdown(
-    """
-    <style>
-    .center-title {
-        display: flex;
-        justify-content: center;
-        align-items: center;
-        height: 50px; /* Optional: Adjust height */
-        font-size: 36px; /* Optional: Adjust font size */
-        font-weight: bold; /* Optional: Adjust font weight */
-        color: #084C3C; /* Optional: Set text color */
-    }
-    </style>
-    <div class="center-title">Login Page</div>
-    """,
-    unsafe_allow_html=True
-)
+        """
+        <style>
+        .center-title {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            height: 50px; /* Optional: Adjust height */
+            font-size: 36px; /* Optional: Adjust font size */
+            font-weight: bold; /* Optional: Adjust font weight */
+            color: #084C3C; /* Optional: Set text color */
+        }
+        </style>
+        <div class="center-title">Login Page</div>
+        """,
+        unsafe_allow_html=True
+    )
 
-    
     username = st.text_input("Username")
     password = st.text_input("Password", type="password")
-    
+
     # Check login state
     if "login_success" not in st.session_state:
         st.session_state["login_success"] = False
+
     st.markdown(
         """
         <style>
@@ -139,17 +139,86 @@ def login_page():
         unsafe_allow_html=True,
     )
 
-# Functional and visually styled login button
+
+
     if st.button("Login"):
         if username in users and users[username]["password"] == password:
             st.session_state["user_type"] = users[username]["role"]
             st.session_state["logged_in"] = True
             st.session_state["login_success"] = True
+
             st.success(f"Logged in successfully as {users[username]['role']}")
-            time.sleep(1)
+
+            # Navigate to the appropriate page based on the user's role
+            if users[username]["role"] == "Customer":
+                st.session_state["page"] = "Submit Ticket"
+            elif users[username]["role"] == "Manager":
+                st.session_state["page"] = "Management Dashboard"
+            elif users[username]["role"] == "Employee":
+                st.session_state["page"] = "Respond to a Ticket"
         else:
             st.error("Invalid username or password.")
 
+    if st.button("Forget Password"):
+        st.session_state["page"] = "Forget Password"
+
+            
+def navigate_to_page(page_name):
+    st.session_state["current_page"] = page_name  # Update the page in session state
+
+def get_current_page():
+    return st.session_state.get("current_page", "Login")  # Default to "Login"
+
+def forget_password_page():
+    # Page title and instructions
+    st.markdown('<h1 style="color: #084C3C; text-align: center;">Forget Password</h1>', unsafe_allow_html=True)
+    st.markdown('<p style="color: #084C3C; text-align: center;">Enter your username or email to reset your password.</p>', unsafe_allow_html=True)
+
+    # Input field for username or email
+    username_or_email = st.text_input("Username or Email")
+
+    # Custom button styling
+    st.markdown(
+        """
+        <style>
+        div.stButton > button {
+            background-color: #4CAF50 !important;
+            color: white !important;
+            border: none;
+            padding: 10px 20px !important;
+            text-align: center !important;
+            text-decoration: none;
+            font-size: 16px !important;
+            font-weight: bold !important;
+            cursor: pointer;
+            border-radius: 4px !important;
+            transition: background-color 0.3s ease;
+            display: inline-block;
+            margin-top: 10px; /* Space above the button */
+            box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.1); /* Optional shadow */
+        }
+        div.stButton > button:hover {
+            background-color: #45a049 !important;
+            color: white !important;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )
+
+    # Reset Password button
+    if st.button("Reset Password"):
+        if username_or_email.strip():
+            st.success(f"Password reset link has been sent to {username_or_email}.")
+        else:
+            st.error("Please enter a valid username or email.")
+
+    # Back to Login button
+    if st.button("Back to Login"):
+        st.session_state["page"] = "Login"  # Navigate back to Login
+
+
+
 def add_return_to_login():
     if "logged_in" in st.session_state and st.session_state["logged_in"]:
         # Add custom styling for the Back button
@@ -184,53 +253,84 @@ def add_return_to_login():
         # Functional and visually styled "Back" button
         if st.sidebar.button("Back"):
             # Reset the 'logged_in' state and force a redirect to the login page
-            st.session_state["logged_in"] = False; st.session_state["login_success"] = False; st.stop();st.experimental_rerun()
-
+            st.session_state["logged_in"] = False; st.session_state["login_success"] = False; st.session_state["page"] = "Login"
+            hide_sidebar()
+            
+def hide_sidebar():
+    st.markdown(
+        """
+        <style>
+        section[data-testid="stSidebar"] {
+            display: none; /* Hide the sidebar */
+        }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )
 
 def main():
-    if "logged_in" not in st.session_state or not st.session_state["logged_in"]:
-        login_page()
-        return
-
+    if "page" not in st.session_state:
+        st.session_state["page"] = "Login"  # Default to the Login page
     add_return_to_login()
-    user_type = st.session_state.get("user_type", "Customer")
-    st.sidebar.title("Navigation")
-    
-
-    user_type = st.session_state.get("user_type", "Customer")
-    st.sidebar.title("Navigation")
-
-    if user_type == "Customer":
-        page = st.sidebar.radio("Go to", ["Submit Ticket", "Check Ticket Status", "FAQ & Info"])
-    elif user_type == "Manager":
-        page = st.sidebar.radio("Go to", ["Management Dashboard", "Ticket Dashboard", "Customer Dashboard"])
-    elif user_type == "Employee":
-        page = st.sidebar.radio("Go to", ["Respond to a Ticket", "Ticket Status"])
-    else:
-        st.sidebar.write("No pages available for this user type.")
-        return
-
+    # Show the navigation menu only if not on Login or Forget Password pages
+    if st.session_state["page"] not in ["Login", "Forget Password"]:
+        st.sidebar.title("Navigation")
+        user_type = st.session_state.get("user_type", "Customer")
+
+        if user_type == "Customer":
+            page = st.sidebar.radio("Go to", ["Submit Ticket", "Check Ticket Status", "FAQ & Info"])
+        elif user_type == "Manager":
+            page = st.sidebar.radio("Go to", ["Management Dashboard", "Ticket Dashboard", "Customer Dashboard"])
+        elif user_type == "Employee":
+            page = st.sidebar.radio("Go to", ["Respond to a Ticket", "Ticket Status"])
+        else:
+            st.sidebar.write("No pages available for this user type.")
+            return
+        st.session_state["page"] = page  # Update the page state based on navigation
 
-    if page == "Submit Ticket":
+    # Render the appropriate page
+    if st.session_state["page"] == "Login":
+        login_page()
+    elif st.session_state["page"] == "Forget Password":
+        forget_password_page()
+    elif st.session_state["page"] == "Submit Ticket":
         submit_ticket_page()
-    elif page == "Check Ticket Status":
+    elif st.session_state["page"] == "Check Ticket Status":
         check_ticket_status_page()
-    elif page == "FAQ & Info":
+    elif st.session_state["page"] == "FAQ & Info":
         faq_info_page()
-    elif page == "Management Dashboard":
+    elif st.session_state["page"] == "Management Dashboard":
         management_dashboard_page()
-    elif page == "Ticket Dashboard":
+    elif st.session_state["page"] == "Ticket Dashboard":
         ticket_dashboard_page()
-    elif page == "Customer Dashboard":
+    elif st.session_state["page"] == "Customer Dashboard":
         customer_dashboard_page()
-    elif page == "Respond to a Ticket":
+    elif st.session_state["page"] == "Respond to a Ticket":
         respond_to_ticket_page()
-    elif page == "Ticket Status":
+    elif st.session_state["page"] == "Ticket Status":
         employee_ticket_status_page()
 
+def center_logo():
+    # Create three columns
+    col1, col2, col3 = st.columns([1, 2, 1])  # Adjust column width ratios
+
+    with col1:
+        st.empty()  # Empty content in the first column
+
+    with col2:
+        st.image("/Users/klnimri/Desktop/Newtest/Test/bin/logo.png", width=300, use_container_width=True)
+
+    with col3:
+        st.empty()
+   
 def submit_ticket_page():
-    # Add logo and title
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
+    vspace()     
     st.markdown('<h1 style="color: #084C3C;">Submit a Ticket</h1>', unsafe_allow_html=True)
     st.markdown('<p style="color: #084C3C;">Please fill out the form below to submit a ticket.</p>', unsafe_allow_html=True)
 
@@ -283,8 +383,15 @@ def submit_ticket_page():
                 st.error("Please fill out the Customer ID.")
 
 def check_ticket_status_page():
-    # Add logo and title
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
     st.markdown('<h1 style="color: #084C3C;">Check Ticket Status</h1>', unsafe_allow_html=True)
     st.markdown('<p style="color: #084C3C;">Enter your Customer ID to view your tickets.</p>', unsafe_allow_html=True)
 
@@ -342,7 +449,11 @@ def check_ticket_status_page():
             st.error("Please enter your Customer ID.")
 
 def faq_info_page():
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
     st.markdown('<h1 style="color: #084C3C;">FAQ & Information</h1>', unsafe_allow_html=True)
 
     faq_data = [
@@ -358,7 +469,11 @@ def faq_info_page():
 
 def respond_to_ticket_page():
     # Add logo and title
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
     st.markdown('<h1 style="color: #084C3C;">Respond to a Ticket</h1>', unsafe_allow_html=True)
     
     # Display ticket data
@@ -422,7 +537,11 @@ def respond_to_ticket_page():
 
 def employee_ticket_status_page():
     # Add logo and title
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
     st.markdown('<h1 style="color: #084C3C;">Ticket Status</h1>', unsafe_allow_html=True)
 
     # Input field for Customer ID
@@ -477,7 +596,11 @@ def employee_ticket_status_page():
             st.error("Please enter your Customer ID.")
 
 def ticket_dashboard_page():
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
     st.markdown('<h1 style="color: #084C3C;">Ticket Dashboard</h1>', unsafe_allow_html=True)
     st.components.v1.iframe(
         "https://app.powerbi.com/view?r=eyJrIjoiMDEwNzA2YjUtNGY0MC00NTFjLTg1ZTctYTZlZjQzOTUwNWUxIiwidCI6ImI0NTNkOTFiLTZhYzEtNGI2MS1iOGI4LTVlNjVlNDIyMjMzZiIsImMiOjl9",
@@ -488,7 +611,11 @@ def ticket_dashboard_page():
 
 
 def customer_dashboard_page():
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
     st.markdown('<h1 style="color: #084C3C;">Customer Dashboard</h1>', unsafe_allow_html=True)
     customer_data = {
         "Customer ID": ["C001", "C002", "C003"],
@@ -505,7 +632,11 @@ def customer_dashboard_page():
     )
 
 def management_dashboard_page():
-    st.image("logo.png", width=200)
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
     #center_logo("logo.png")  # Center the logo
     st.markdown('<h1 style="color: #084C3C;">Management Dashboard</h1>', unsafe_allow_html=True)
     
@@ -573,14 +704,15 @@ def customize_login_page():
     st.markdown('<div class="center-content">', unsafe_allow_html=True)
     
     # Display the image in the center
-    st.image(
-        "logo.png", 
-        width=200, 
-        use_container_width=False
-    )
- 
-
+    center_logo()
+    vspace()
+    vspace()
+    vspace()
+    vspace()
    
+def vspace():
+    st.empty()
+    st.text("")
 
 def customize_submit_button():
     st.markdown(