4

alejandro-ao · Jan 28, 2024 · 6e7144e · 6e7144e
1 parent 94c2876
commit 6e7144e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/app.py b/src/app.py
@@ -26,5 +26,12 @@ def get_response(user_input):
   st.session_state.chat_history.append(HumanMessage(content=user_query))
   st.session_state.chat_history.append(AIMessage(content=response))
 
-with st.sidebar:
-  st.write(st.session_state.chat_history)
+# conversation 
+for message in st.session_state.chat_history:
+  if isinstance(message, AIMessage):
+    with st.chat_message("AI"):
+      st.write(message.content)
+  elif isinstance(message, HumanMessage):
+    with st.chat_message("Human"):
+      st.write(message.content)
+