Update app.py

app.py

@@ -20,12 +20,15 @@ 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)
 
-# Initialize session state for messages
+# Initialize session state for messages and uploaded files
 if "messages" not in st.session_state:
     st.session_state.messages = [
         {"role": "assistant", "content": "Ask me a question about the documents you uploaded!"}
     ]
 
+if 'uploaded_file_names' not in st.session_state:
+    st.session_state.uploaded_file_names = []
+
 # Function to read PDF files
 def read_pdf(file):
     with pdfplumber.open(file) as pdf:
@@ -64,7 +67,11 @@ 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)
+        conversation_data = {
+            "messages": st.session_state.messages,
+            "file_names": st.session_state.uploaded_file_names
+        }
+        json.dump(conversation_data, f)
         f.write("\n")
 
 # Function to load previous conversations
@@ -88,6 +95,8 @@ def delete_selected_conversations(selected_indices):
 uploaded_files = st.file_uploader("Upload PDF or DOCX files", type=["pdf", "docx"], accept_multiple_files=True)
 
 if uploaded_files and st.session_state.openai_api_key:
+    # Store the names of the uploaded files
+    st.session_state.uploaded_file_names = [uploaded_file.name for uploaded_file in uploaded_files]
     index = load_data(uploaded_files)
     chat_engine = index.as_chat_engine(chat_mode="condense_question", verbose=True)
 
@@ -146,8 +155,9 @@ if st.session_state.show_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']}")
+            st.sidebar.write(f"Files: {', '.join(conv['file_names'])}")
             # Checkbox for selecting conversation to delete
             if st.sidebar.checkbox(f"Select Conversation {i + 1} for Deletion", key=f"delete_checkbox_{i}"):
                 selected_indices.append(i)
@@ -155,7 +165,7 @@ if st.session_state.show_conversations:
         if st.sidebar.button("Delete Selected Conversations"):
             if selected_indices:
                 delete_selected_conversations(selected_indices)
-                st.success("Selected conversations deleted. Please Refresh to See the Effect!")
+                st.success("Selected conversations deleted. Please refresh to see the effect!")
                 st.session_state.messages = []  # Optional: reset messages for a fresh start
 
     else: