Update app.py

app.py

@@ -16,8 +16,16 @@ def load_answer(question):
     st.session_state.sessionMessages.append(HumanMessage
                                             (content=question))
     assistant_answer = chat_model.invoke(st.session_state.sessionMessages)
-    st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content if hasattr(assistant_answer, "content") else str(assistant_answer)))
-    return assistant_answer.content
+
+    if isinstance(assistant_answer, AIMessage):
+        response_text = assistant_answer.content
+    elif isinstance(assistant_answer, dict) and "content" in assistant_answer:
+        response_text = assistant_answer["content"]
+    else:
+        response_text = str(assistant_answer)
+    
+    st.session_state.sessionMessages.append(AIMessage(content=response_text))
+    return response_text
 
 def get_text():
     input_text = st.text_input("You: ")