Update app.py

app.py

@@ -44,15 +44,15 @@ def main():
     # Initialize database
     create_user_table()
 
-    # Authentication state
+    # Manage session state
     if "authenticated" not in st.session_state:
         st.session_state.authenticated = False
     if "username" not in st.session_state:
         st.session_state.username = None
     if "page" not in st.session_state:
-        st.session_state.page = st.query_params.get("page", ["login"])[0]
+        st.session_state.page = "login"
 
-    # Navigation logic
+    # Page routing logic
     if st.session_state.page == "login":
         login_page()
     elif st.session_state.page == "workspace":
@@ -67,20 +67,21 @@ def login_page():
         username = st.text_input("Username", key="login_username")
         password = st.text_input("Password", type="password", key="login_password")
 
+        # Handle single-click login
         if st.button("Log In"):
             if authenticate_user(username, password):
                 st.session_state.authenticated = True
                 st.session_state.username = username
                 st.session_state.page = "workspace"
-                st.query_params["page"] = "workspace"
             else:
                 st.error("Invalid username or password. Please try again.")
-
+    
     elif auth_mode == "Register":
         st.subheader("Register")
         username = st.text_input("Create Username", key="register_username")
         password = st.text_input("Create Password", type="password", key="register_password")
 
+        # Handle single-click registration
         if st.button("Register"):
             if username and password:
                 add_user(username, password)
@@ -92,10 +93,10 @@ def workspace_page():
     # Sidebar with logout button
     st.sidebar.title(f"Hello, {st.session_state.username}!")
     if st.sidebar.button("Log Out"):
+        # Handle single-click logout
         st.session_state.authenticated = False
         st.session_state.username = None
         st.session_state.page = "login"
-        st.query_params["page"] = "login"
 
     # Main content area
     st.subheader(f"Welcome to your workspace, {st.session_state.username}!")
@@ -103,3 +104,4 @@ def workspace_page():
 
 if __name__ == "__main__":
     main()
+