Update app.py

app.py

@@ -50,30 +50,39 @@ def authenticate_user(username, password):
 
 # UI Layout
 st.set_page_config(page_title="Construction To-Do List", layout="wide")
-st.sidebar.title("User Authentication")
-choice = st.sidebar.radio("Select Option", ["Sign In", "Sign Up"])
-
-if choice == "Sign Up":
-    new_username = st.sidebar.text_input("Enter a username")
-    new_password = st.sidebar.text_input("Enter a password", type="password")
-    if st.sidebar.button("Register") and new_username and new_password:
-        register_user(new_username, new_password)
-        st.success("Registration successful! Please sign in.")
-        st.stop()
-
-elif choice == "Sign In":
-    username = st.sidebar.text_input("Enter your username")
-    password = st.sidebar.text_input("Enter your password", type="password")
-    if st.sidebar.button("Login"):
-        if authenticate_user(username, password):
-            st.success("Login successful!")
-        else:
-            st.error("Invalid username or password")
-            st.stop()
+init_db()
 
-menu = st.sidebar.radio("Go to", ["Dashboard", "View Tasks by Status"])
+if "authenticated" not in st.session_state:
+    st.session_state["authenticated"] = False
+    st.session_state["username"] = ""
+
+if not st.session_state["authenticated"]:
+    st.title("🔐 User Authentication")
+    choice = st.radio("Select Option", ["Sign In", "Sign Up"])
+
+    if choice == "Sign Up":
+        new_username = st.text_input("Enter a username")
+        new_password = st.text_input("Enter a password", type="password")
+        if st.button("Register") and new_username and new_password:
+            register_user(new_username, new_password)
+            st.success("Registration successful! Please sign in.")
+
+    elif choice == "Sign In":
+        username = st.text_input("Enter your username")
+        password = st.text_input("Enter your password", type="password")
+        if st.button("Login"):
+            if authenticate_user(username, password):
+                st.session_state["authenticated"] = True
+                st.session_state["username"] = username
+                st.rerun()
+            else:
+                st.error("Invalid username or password")
+    st.stop()
 
-init_db()
+# Main Dashboard After Login
+st.sidebar.title(f"Welcome, {st.session_state['username']}")
+menu = st.sidebar.radio("Go to", ["Dashboard", "View Tasks by Status"])
+username = st.session_state['username']
 
 if menu == "Dashboard":
     st.title("🛠️ Construction Program Manager - Dashboard")