Updated streamlit app to include user login, creation, and deletion

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-
 from postly.clients.postly_client import PostlyClient
 
 # Initialize the PostlyClient in Streamlit's session state
@@ -8,40 +7,68 @@ if "client" not in st.session_state:
 
 client = st.session_state.client
 
+# Initialize user session state
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+if "current_user" not in st.session_state:
+    st.session_state.current_user = None
 
-def add_user():
-    st.title("Add User")
+def register():
+    st.title("Register")
     user_name = st.text_input("Enter user name")
-    if st.button("Add User"):
-        client.add_user(user_name)
-        st.success(f"User '{user_name}' added successfully.")
-
+    password = st.text_input("Enter password", type="password")
+    if st.button("Register"):
+        if user_name and password:
+            try:
+                client.add_user(user_name, password)
+                st.session_state.logged_in = True
+                st.session_state.current_user = user_name
+                st.success(f"User '{user_name}' registered and logged in successfully.")
+                st.rerun()
+            except ValueError as e:
+                st.error(f"Error: {e}")
+        else:
+            st.error("Please enter both user name and password.")
+
+def login():
+    st.title("Login")
+    user_name = st.text_input("Enter user name")
+    password = st.text_input("Enter password", type="password")
+    if st.button("Login"):
+        if client.authenticate_user(user_name, password):
+            st.session_state.logged_in = True
+            st.session_state.current_user = user_name
+            st.success(f"User '{user_name}' logged in successfully.")
+            st.rerun()
+        else:
+            st.error("Invalid user name or password.")
+
+def logout():
+    st.session_state.logged_in = False
+    st.session_state.current_user = None
+    st.success("Logged out successfully.")
+    st.rerun()
+
+def delete_own_user():
+    st.title("Delete Account")
+    if st.button("Delete Account"):
+        try:
+            client.delete_user(st.session_state.current_user)
+            st.success(f"User '{st.session_state.current_user}' deleted successfully.")
+            logout()
+        except KeyError as e:
+            st.error(f"Error: {e}")
 
 def add_post():
     st.title("Add Post")
-    users = client.get_users()
-    user_name = st.selectbox("Select user name", users)
     post_text = st.text_area("Enter post text")
     if st.button("Add Post"):
         try:
-            client.add_post(user_name, post_text)
+            client.add_post(st.session_state.current_user, post_text)
             st.success("Post added successfully.")
         except Exception as e:
             st.error(f"Error: {e}")
 
-
-def delete_user():
-    st.title("Delete User")
-    users = client.get_users()
-    user_name = st.selectbox("Select user name", users)
-    if st.button("Delete User"):
-        try:
-            client.delete_user(user_name)
-            st.success(f"User '{user_name}' deleted successfully.")
-        except KeyError as e:
-            st.error(f"Error: {e}")
-
-
 def get_posts_for_user():
     st.title("Get Posts for User")
     users = client.get_users()
@@ -55,7 +82,6 @@ def get_posts_for_user():
         except KeyError as e:
             st.error(f"Error: {e}")
 
-
 def get_posts_for_topic():
     st.title("Get Posts for Topic")
     topics = client.get_topics()
@@ -66,7 +92,6 @@ def get_posts_for_topic():
         for post in posts:
             st.write(post)
 
-
 def get_trending_topics():
     st.title("Get Trending Topics")
     current_timestamp = client.get_current_timestamp()
@@ -82,8 +107,6 @@ def get_trending_topics():
                 st.write(topic)
         except ValueError as e:
             st.error(f"Error: {e}")
-        st.rerun()
-
 
 def get_all_posts():
     st.title("All Posts")
@@ -98,38 +121,43 @@ def get_all_posts():
         st.markdown(f"{post.content}")
         st.markdown("---")
 
-
 def main():
     st.sidebar.title("Postly\nSimple social media platform")
-    page = st.sidebar.selectbox(
-        "Choose an action",
-        [
-            "Add User",
-            "Add Post",
-            "Delete User",
-            "Get Posts for User",
-            "Get Posts for Topic",
-            "Get Trending Topics",
-            "View All Posts",
-        ],
-        index=6,
-    )
-
-    if page == "Add User":
-        add_user()
-    elif page == "Add Post":
-        add_post()
-    elif page == "Delete User":
-        delete_user()
-    elif page == "Get Posts for User":
-        get_posts_for_user()
-    elif page == "Get Posts for Topic":
-        get_posts_for_topic()
-    elif page == "Get Trending Topics":
-        get_trending_topics()
-    elif page == "View All Posts":
-        get_all_posts()
-
+    if st.session_state.logged_in:
+        st.sidebar.write(f"Logged in as: {st.session_state.current_user}")
+        if st.sidebar.button("Logout"):
+            logout()
+        page = st.sidebar.selectbox(
+            "Choose an action",
+            [
+                "Add Post",
+                "Delete Account",
+                "Get Posts for User",
+                "Get Posts for Topic",
+                "Get Trending Topics",
+                "View All Posts",
+            ],
+            index=5,
+        )
+
+        if page == "Add Post":
+            add_post()
+        elif page == "Delete Account":
+            delete_own_user()
+        elif page == "Get Posts for User":
+            get_posts_for_user()
+        elif page == "Get Posts for Topic":
+            get_posts_for_topic()
+        elif page == "Get Trending Topics":
+            get_trending_topics()
+        elif page == "View All Posts":
+            get_all_posts()
+    else:
+        page = st.sidebar.selectbox("Choose an action", ["Login", "Register"], index=0)
+        if page == "Login":
+            login()
+        elif page == "Register":
+            register()
 
 if __name__ == "__main__":
     main()