Update app.py

app.py

@@ -20,6 +20,12 @@ if 'openai_api_key' not in st.session_state:
 st.session_state.openai_api_key = st.sidebar.text_input("Enter your OpenAI API Key:", type="password",
                                                          value=st.session_state.openai_api_key)
 
+# Add a text input for the username
+if 'username' not in st.session_state:
+    st.session_state.username = ""
+
+st.session_state.username = st.sidebar.text_input("Enter your username:")
+
 # Initialize session state for messages
 if "messages" not in st.session_state:
     st.session_state.messages = [
@@ -63,16 +69,24 @@ def load_data(uploaded_files):
 
 # Function to save the conversation
 def save_conversation():
-    with open("conversations.json", "a") as f:
-        json.dump(st.session_state.messages, f)
-        f.write("\n")
+    if st.session_state.username:
+        with open("conversations.json", "a") as f:
+            conversation_entry = {
+                "username": st.session_state.username,
+                "messages": st.session_state.messages
+            }
+            json.dump(conversation_entry, f)
+            f.write("\n")
+    else:
+        st.warning("Please enter a username before saving the conversation.")
 
 # Function to load previous conversations
 def load_conversations():
     if os.path.exists("conversations.json"):
         with open("conversations.json", "r") as f:
             conversations = [json.loads(line) for line in f]
-        return conversations
+            # Filter conversations based on the current username
+            return [conv for conv in conversations if conv["username"] == st.session_state.username]
     return []
 
 # Function to delete selected conversations
@@ -139,14 +153,14 @@ if st.sidebar.button("Toggle Previous Conversations"):
 
 # Show previous conversations if the toggle is enabled
 if st.session_state.show_conversations:
-    st.sidebar.subheader("Previous Conversations")
+    st.sidebar.subheader("Your Previous Conversations")
     conversations = load_conversations()
 
     if conversations:
         selected_indices = []
         for i, conv in enumerate(conversations):
             st.sidebar.write(f"Conversation {i + 1}:")
-            for message in conv:
+            for message in conv["messages"]:
                 st.sidebar.write(f"{message['role']}: {message['content']}")
             # Checkbox for selecting conversation to delete
             if st.sidebar.checkbox(f"Select Conversation {i + 1} for Deletion", key=f"delete_checkbox_{i}"):
@@ -161,4 +175,4 @@ if st.session_state.show_conversations:
     else:
         st.sidebar.write("No previous conversations found.")
 else:
-    st.sidebar.write("Previous conversations are hidden. Click 'Toggle Previous Conversations' to show.")
+    st.sidebar.write("Previous conversations are hidden. Click 'Toggle Previous Conversations' to show.")