Update app.py

app.py

@@ -69,10 +69,11 @@ def login_page():
 
         if st.button("Log In"):
             if authenticate_user(username, password):
-                st.success(f"Welcome back, {username}!")
                 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")
             else:
                 st.error("Invalid username or password. Please try again.")
 
@@ -95,15 +96,16 @@ 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")
 
     # 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__":
-    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()