Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -100,10 +100,14 @@ else:
|
|
100 |
|
101 |
|
102 |
# conversation
|
|
|
103 |
for message in st.session_state.chat_history:
|
104 |
-
if
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
|
|
|
|
|
|
|
100 |
|
101 |
|
102 |
# conversation
|
103 |
+
first = True
|
104 |
for message in st.session_state.chat_history:
|
105 |
+
if first == True:
|
106 |
+
first = False
|
107 |
+
else:
|
108 |
+
if isinstance(message, AIMessage):
|
109 |
+
with st.chat_message("AI"):
|
110 |
+
st.write(message.content)
|
111 |
+
elif isinstance(message, HumanMessage):
|
112 |
+
with st.chat_message("Human"):
|
113 |
+
st.write(message.content)
|