Update app.py

app.py

@@ -30,12 +30,10 @@ user_query = st.chat_input("Type your message here...")
 if user_query is not None and user_query !="":
     response = get_response(user_query)
     chat_history.append(HumanMessage(content=user_query))
-            
-     with st.chat_message("Human"):
-        st.write(user_query)
+    chat_history.append(AIMessageMessage(content=response))
     
-    with st.chat_message("AI"):
-        st.write(response)
+            
+