Update app.py

app.py

@@ -20,15 +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)
 
-# Initialize session state for messages and uploaded files
+# Initialize session state for messages
 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:
@@ -67,23 +64,14 @@ def load_data(uploaded_files):
 # Function to save the conversation
 def save_conversation():
     with open("conversations.json", "a") as f:
-        conversation_data = {
-            "messages": st.session_state.messages,
-            "file_names": st.session_state.uploaded_file_names
-        }
-        if isinstance(conversation_data, dict):
-            json.dump(conversation_data, f)
-            f.write("\n")
+        json.dump(st.session_state.messages, f)
+        f.write("\n")
 
 # 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]
-        # Ensure each conversation is a dictionary
-        for conv in conversations:
-            if not isinstance(conv, dict):
-                print("Warning: Loaded conversation is not a dictionary:", conv)
         return conversations
     return []
 
@@ -100,8 +88,6 @@ 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)
 
@@ -159,21 +145,17 @@ if st.session_state.show_conversations:
     if conversations:
         selected_indices = []
         for i, conv in enumerate(conversations):
-            if isinstance(conv, dict):  # Ensure conv is a dictionary
-                st.sidebar.write(f"Conversation {i + 1}:")
-                for message in conv.get('messages', []):  # Use get safely
-                    st.sidebar.write(f"{message['role']}: {message['content']}")
-                st.sidebar.write(f"Files: {', '.join(conv.get('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)
-            else:
-                print("Warning: Encountered a non-dictionary conversation:", conv)
+            st.sidebar.write(f"Conversation {i + 1}:")
+            for message in conv:
+                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}"):
+                selected_indices.append(i)
 
         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: