Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 import time
+import requests
 
 # Sample user data for login
 users = {
@@ -9,137 +10,96 @@ users = {
     "customer789": {"password": "customerpass", "role": "Customer"},
 }
 
-def main():
-    # Custom CSS for styling
-    st.markdown(
-        """
-        <style>
-        /* Background color for the app */
-        .stApp {
-            background-color: #E4ECE3;
-        }
-
-        /* Sidebar styling */
-        section[data-testid="stSidebar"] {
-            background-color: #084C3C; /* Light green background for the sidebar */
-        }
-
-        /* Sidebar title color */
-        section[data-testid="stSidebar"] p {
-            color: white; /* Title text color for "Navigation" */
-        }
-
-        /* Input fields styling */
-        input, textarea, .stFileUploader {
-            background-color: white !important; /* White background */
-            border: 2px solid #084C3C !important; /* Dark green border */
-            color: #084C3C !important; /* Dark green text */
-            border-radius: 5px !important; /* Rounded corners */
-        }
-
-        /* Submit button styling */
-        div[data-testid="stFormSubmitButton"] button, div.stButton > button {
-            background-color: #084C3C !important; /* Dark green background */
-            color: white !important; /* White text */
-            border-radius: 5px !important;
-            padding: 10px 20px !important;
-        }
-
-        div[data-testid="stFormSubmitButton"] button:hover, div.stButton > button:hover {
-            background-color: #066B4E !important; /* Lighter green on hover */
-        }
-
-        /* Text color */
-        .stApp div:not([data-testid="stSidebar"]) label {
-            color: #084C3C;
-        }
-
-        /* Table styles */
-        table {
-            border-collapse: collapse;
-            width: 100%;
-        }
-
-        th, td {
-            text-align: left;
-            padding: 8px;
-            color: #084C3C;
-            border: 1px solid #084C3C;
-        }
-
-        th {
-            background-color: #E4ECE3;
-        }
-
-        tr:nth-child(even) {
-            background-color: #f9f9f9;
-        }
-        </style>
-        """,
-        unsafe_allow_html=True
-    )
-
-    if "logged_in" not in st.session_state:
-        st.session_state["logged_in"] = False
-        st.session_state["user_role"] = None
-
-    if not st.session_state["logged_in"]:
-        login_page()
-    else:
-        user_role = st.session_state["user_role"]
-        st.sidebar.title("Navigation")
-        if user_role == "Customer":
-            page = st.sidebar.radio(
-                "Go to", ["Submit Ticket", "Check Ticket Status", "FAQ & Info"]
-            )
-        elif user_role == "Manager":
-            page = st.sidebar.radio(
-                "Go to",
-                ["Management Dashboard", "Ticket Dashboard", "Customer Dashboard"],
-            )
-        elif user_role == "Employee":
-            page = st.sidebar.radio(
-                "Go to", ["Respond to a Ticket", "Ticket Status"]
-            )
-        else:
-            st.error("Unknown user role!")
-            return
-
-        if page == "Submit Ticket":
-            submit_ticket_page()
-        elif page == "Check Ticket Status":
-            check_ticket_status_page()
-        elif page == "FAQ & Info":
-            faq_info_page()
-        elif page == "Management Dashboard":
-            management_dashboard_page()
-        elif page == "Ticket Dashboard":
-            ticket_dashboard_page()
-        elif page == "Customer Dashboard":
-            customer_dashboard_page()
-        elif page == "Respond to a Ticket":
-            respond_to_ticket_page()
-        elif page == "Ticket Status":
-            employee_ticket_status_page()
+# Function to load a Lottie animation from a URL
+def load_lottie_url(url: str):
+    r = requests.get(url)
+    if r.status_code != 200:
+        return None
+    return r.json()
+
+# URL for the Lottie animation
+lottie_url = "https://assets6.lottiefiles.com/packages/lf20_5hhtik.json"
+lottie_json = load_lottie_url(lottie_url)
+
+# Function to display the Lottie animation
+def show_loading_animation():
+    if lottie_json:
+        st_lottie(lottie_json, height=300, key="loading")
 
 def login_page():
+    st.set_page_config(page_title="Login", layout="centered")
     st.title("Login Page")
-    username = st.text_input("Username")
-    password = st.text_input("Password", type="password")
-
-    if st.button("Login"):
-        if username in users and users[username]["password"] == password:
-            st.session_state["logged_in"] = True
-            st.session_state["user_role"] = users[username]["role"]
-            st.success(f"Welcome, {users[username]['role']}!")
-            time.sleep(1)
-            st.experimental_rerun()
-        else:
-            st.error("Invalid username or password.")
+
+    if "login_success" not in st.session_state:
+        st.session_state["login_success"] = False
+
+    if not st.session_state["login_success"]:
+        username = st.text_input("Username")
+        password = st.text_input("Password", type="password")
+
+        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)
+            else:
+                st.error("Invalid username or password.")
+
+def add_return_to_login():
+    if "logged_in" in st.session_state and st.session_state["logged_in"]:
+        if st.sidebar.button("Return to Login Page"):
+            st.session_state.clear()
+            st.rerun()
+
+def main():
+    if "logged_in" not in st.session_state or not st.session_state["logged_in"]:
+        login_page()
+        return
+
+    st.set_page_config(page_title="Ticket System", layout="centered", initial_sidebar_state="expanded")
+    add_return_to_login()
+
+    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
+
+    with st.spinner("Loading..."):
+        show_loading_animation()
+        time.sleep(0.5)
+
+    if page == "Submit Ticket":
+        submit_ticket_page()
+    elif page == "Check Ticket Status":
+        check_ticket_status_page()
+    elif page == "FAQ & Info":
+        faq_info_page()
+    elif page == "Management Dashboard":
+        management_dashboard_page()
+    elif page == "Ticket Dashboard":
+        ticket_dashboard_page()
+    elif page == "Customer Dashboard":
+        customer_dashboard_page()
+    elif page == "Respond to a Ticket":
+        respond_to_ticket_page()
+    elif page == "Ticket Status":
+        employee_ticket_status_page()
 
 def submit_ticket_page():
+    st.image("logo.png", width=200)
     st.title("Submit a Ticket")
     with st.form(key="submit_ticket_form"):
+        customer_id = st.text_input("Customer ID")
         st.text_input("First Name")
         st.text_input("Last Name")
         st.text_input("National ID / Residency Number")
@@ -149,18 +109,15 @@ def submit_ticket_page():
         st.text_area("Describe your issue/request")
         st.file_uploader("Upload Attachments (if any)", accept_multiple_files=True)
         if st.form_submit_button("Submit"):
-            st.success("Your ticket has been submitted.")
+            st.success(f"Your ticket has been submitted successfully. Customer ID: {customer_id}")
 
 def check_ticket_status_page():
+    st.image("logo.png", width=200)
     st.title("Check Ticket Status")
-    national_id = st.text_input("National ID / Residency Number")
+    customer_id = st.text_input("Customer ID")
     if st.button("Check Status"):
-        if national_id:
-            st.markdown(
-                f"<p style='color: #084C3C;'>Displaying tickets for National ID: {national_id}</p>",
-                unsafe_allow_html=True,
-            )
-            # Simulated data
+        if customer_id:
+            st.success(f"Displaying tickets for Customer ID: {customer_id}")
             ticket_data = {
                 "Ticket Number": ["141", "565", "999"],
                 "Date": ["2025-01-18", "2025-03-28", "2025-04-01"],
@@ -168,23 +125,56 @@ def check_ticket_status_page():
                 "Status": ["Resolved", "Pending", "In Progress"],
             }
             df = pd.DataFrame(ticket_data)
-            st.table(df)
+            st.dataframe(df)
         else:
-            st.warning("Please enter your National ID.")
+            st.warning("Please enter your Customer ID.")
 
 def faq_info_page():
+    st.image("logo.png", width=200)
     st.title("FAQ & Information")
-    faqs = [
+    faq_data = [
         ("How to open a new account?", "Visit your nearest branch or use our app."),
         ("Branch Operating Hours", "Sunday to Thursday, 9:00 AM to 4:00 PM."),
     ]
-    for question, answer in faqs:
+    for question, answer in faq_data:
         st.subheader(question)
         st.write(answer)
 
+def respond_to_ticket_page():
+    st.image("logo.png", width=200)
+    st.title("Respond to a Ticket")
+    ticket_data = {
+        "Ticket Number": ["141", "565", "999"],
+        "Customer ID": ["C001", "C002", "C003"],
+        "Date": ["2025-01-18", "2025-03-28", "2025-04-01"],
+        "Type": ["General", "Billing", "Technical"],
+        "Status": ["Pending", "Pending", "In Progress"],
+    }
+    df = pd.DataFrame(ticket_data)
+    st.dataframe(df)
+    ticket_to_respond = st.selectbox("Select a ticket to respond to:", df["Ticket Number"].tolist())
+    if st.button("Respond to Ticket"):
+        st.text_area("Response:", placeholder="Write your response here...")
+        st.button("Submit Response")
+
+def employee_ticket_status_page():
+    st.image("logo.png", width=200)
+    st.title("Ticket Status")
+    customer_id = st.text_input("Customer ID")
+    if st.button("Check Ticket Status"):
+        if customer_id:
+            st.success(f"Displaying tickets for Customer ID: {customer_id}")
+            ticket_data = {
+                "Ticket Number": ["141", "565", "999"],
+                "Date": ["2025-01-18", "2025-03-28", "2025-04-01"],
+                "Status": ["Resolved", "Pending", "In Progress"],
+            }
+            df = pd.DataFrame(ticket_data)
+            st.dataframe(df)
+
 def management_dashboard_page():
     st.title("Management Dashboard")
-    st.write("Management overview and statistics.")
+    st.write("Employee and department overviews.")
 
 def ticket_dashboard_page():
     st.title("Ticket Dashboard")
@@ -192,15 +182,7 @@ def ticket_dashboard_page():
 
 def customer_dashboard_page():
     st.title("Customer Dashboard")
-    st.write("Customer-related statistics.")
-
-def respond_to_ticket_page():
-    st.title("Respond to a Ticket")
-    st.write("Respond to tickets submitted by customers.")
-
-def employee_ticket_status_page():
-    st.title("Ticket Status")
-    st.write("View ticket statuses for employees.")
+    st.write("Insights and data on customer interactions.")
 
 if __name__ == "__main__":
     main()