Update app.py

app.py

@@ -61,15 +61,21 @@ with st.sidebar:
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
+# Ensure messages are a list of dictionaries
+if not isinstance(st.session_state.messages, list):
+    st.session_state.messages = []
+if not all(isinstance(msg, dict) for msg in st.session_state.messages):
+    st.session_state.messages = []
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-
 # React to user input
-st.session_state["history"] = [{"role": "system", "content": "You are a helpful assistant."}]
+if "history" not in st.session_state:
+    st.session_state["history"] = [{"role": "system", "content": "You are a helpful assistant."}]
+
 if prompt := st.chat_input("😉 Ask any question or feel free to use the examples provided in the left sidebar."):
 
     # Display user message in chat message container
@@ -77,11 +83,10 @@ if prompt := st.chat_input("😉 Ask any question or feel free to use the exampl
 
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
-    st.session_state["history"].append({"role": "user", "content": prompt})
 
     # API Call
     bot = ChatBot()
-    bot.history = st.session_state["history"]
+    bot.history = st.session_state.messages  # Update history from messages
     response = bot.generate_response(prompt)
 
     # Display assistant response in chat message container
@@ -90,4 +95,3 @@ if prompt := st.chat_input("😉 Ask any question or feel free to use the exampl
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
-    st.session_state["history"].append({"role": "assistant", "content": response})