AdamyaG commited on
Commit
d051f64
1 Parent(s): c8e3ace

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -100,10 +100,14 @@ else:
100
 
101
 
102
  # conversation
 
103
  for message in st.session_state.chat_history:
104
- if isinstance(message, AIMessage):
105
- with st.chat_message("AI"):
106
- st.write(message.content)
107
- elif isinstance(message, HumanMessage):
108
- with st.chat_message("Human"):
109
- st.write(message.content)
 
 
 
 
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)