Update app.py

app.py

@@ -65,6 +65,7 @@ def main():
                     st.success(f"Welcome back, {username}!")
                     st.session_state.authenticated = True
                     st.session_state.username = username
+                    st.experimental_rerun()
                 else:
                     st.error("Invalid username or password. Please try again.")
 
@@ -79,37 +80,21 @@ def main():
                     st.success("Account created successfully! You can now log in.")
                 else:
                     st.error("Please fill in all fields.")
-
-    # Authenticated user workspace
-    if st.session_state.authenticated:
-        # Display logout button in the upper-right corner
-        with st.container():
-            st.markdown(
-                """
-                <style>
-                .logout-button {
-                    position: fixed;
-                    top: 15px;
-                    right: 15px;
-                }
-                </style>
-                """,
-                unsafe_allow_html=True,
-            )
-            if st.button("Log Out", key="logout"):
-                # Show a confirmation dialog
-                if st.button("Confirm Logout"):
-                    st.session_state.authenticated = False
-                    st.session_state.username = None
-                    st.success("You have been logged out.")
-                    st.experimental_rerun()
-
-        # Workspace content
-        st.subheader(f"Hello, {st.session_state.username}!")
-        st.write("This is your workspace. All your saved work will appear here.")
+    else:
+        # Authenticated user workspace with sidebar
+        st.sidebar.title(f"Hello, {st.session_state.username}!")
+        if st.sidebar.button("Log Out"):
+            st.session_state.authenticated = False
+            st.session_state.username = None
+            st.experimental_rerun()
+
+        # 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()
 
 
 
+