Update app.py

app.py

@@ -71,14 +71,21 @@ def save_conversation():
             "messages": st.session_state.messages,
             "file_names": st.session_state.uploaded_file_names
         }
-        json.dump(conversation_data, f)
-        f.write("\n")
+        if isinstance(conversation_data, dict):
+            json.dump(conversation_data, f)
+            f.write("\n")
+        else:
+            print("Warning: Attempted to save a non-dictionary conversation_data:", conversation_data)
 
 # 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 []
 
@@ -155,9 +162,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['messages']:
+            for message in conv.get('messages', []):
                 st.sidebar.write(f"{message['role']}: {message['content']}")
-            st.sidebar.write(f"Files: {', '.join(conv['file_names'])}")
+            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)