Update app.py

app.py

@@ -17,6 +17,8 @@ def query_model(api_url, payload):
     return response.json()
 
 def add_message_to_conversation(user_message, bot_message, model_name):
+    if "conversation" not in st.session_state:
+        st.session_state.conversation = []
     st.session_state.conversation.append((user_message, bot_message, model_name))
 
 def initialize_memory():
@@ -83,8 +85,9 @@ st.markdown(
 )
 
 # Display the conversation
-st.write('<div class="chat-container">', unsafe_allow_html=True)
-for user_message, bot_message, model_name in st.session_state.conversation:
-    st.write(f'<div class="chat-bubble user">You: {user_message}</div>', unsafe_allow_html=True)
-    st.write(f'<div class="chat-bubble bot">{model_name}: {bot_message}</div>', unsafe_allow_html=True)
-st.write('</div>', unsafe_allow_html=True)
+if "conversation" in st.session_state:
+    st.write('<div class="chat-container">', unsafe_allow_html=True)
+    for user_message, bot_message, model_name in st.session_state.conversation:
+        st.write(f'<div class="chat-bubble user">You: {user_message}</div>', unsafe_allow_html=True)
+        st.write(f'<div class="chat-bubble bot">{model_name}: {bot_message}</div>', unsafe_allow_html=True)
+    st.write('</div>', unsafe_allow_html=True)