Update app.py

app.py

@@ -47,6 +47,8 @@ def main():
     # Authentication 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
 
     # Central login/register interface
     if not st.session_state.authenticated:
@@ -80,9 +82,34 @@ def main():
 
     # 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.")
 
 if __name__ == "__main__":
     main()
 
+
+