Update app.py

app.py

@@ -50,7 +50,7 @@ def main():
     if "username" not in st.session_state:
         st.session_state.username = None
     if "page" not in st.session_state:
-        st.session_state.page = "login"
+        st.session_state.page = st.query_params.get("page", ["login"])[0]
 
     # Navigation logic
     if st.session_state.page == "login":
@@ -72,8 +72,7 @@ def login_page():
                 st.session_state.authenticated = True
                 st.session_state.username = username
                 st.session_state.page = "workspace"
-                # Trigger re-render
-                st.experimental_set_query_params(page="workspace")
+                st.query_params["page"] = "workspace"
             else:
                 st.error("Invalid username or password. Please try again.")
 
@@ -96,16 +95,11 @@ def workspace_page():
         st.session_state.authenticated = False
         st.session_state.username = None
         st.session_state.page = "login"
-        # Trigger re-render
-        st.experimental_set_query_params(page="login")
+        st.query_params["page"] = "login"
 
     # Main content area
     st.subheader(f"Welcome to your workspace, {st.session_state.username}!")
     st.write("This is your personal workspace. All your saved work will appear here.")
 
 if __name__ == "__main__":
-    # Check query parameters to manage page state
-    query_params = st.experimental_get_query_params()
-    if "page" in query_params:
-        st.session_state.page = query_params["page"][0]
-    main()
+    main()