Update app.py
Browse files
app.py
CHANGED
@@ -51,12 +51,13 @@ def load_answer(question):
|
|
51 |
st.session_state.sessionMessages.append(HumanMessage(content=question))
|
52 |
assistant_answer = llm.invoke(st.session_state.sessionMessages)
|
53 |
|
|
|
54 |
if isinstance(assistant_answer, AIMessage):
|
55 |
st.session_state.sessionMessages.append(assistant_answer)
|
56 |
-
return assistant_answer.content
|
57 |
else:
|
58 |
st.session_state.sessionMessages.append(AIMessage(content=assistant_answer))
|
59 |
-
return assistant_answer
|
60 |
|
61 |
# Display the chat history
|
62 |
for message in st.session_state.sessionMessages:
|
|
|
51 |
st.session_state.sessionMessages.append(HumanMessage(content=question))
|
52 |
assistant_answer = llm.invoke(st.session_state.sessionMessages)
|
53 |
|
54 |
+
# Ensure the answer is clean and formatted correctly
|
55 |
if isinstance(assistant_answer, AIMessage):
|
56 |
st.session_state.sessionMessages.append(assistant_answer)
|
57 |
+
return assistant_answer.content.strip() # Strip any unnecessary whitespace or newlines
|
58 |
else:
|
59 |
st.session_state.sessionMessages.append(AIMessage(content=assistant_answer))
|
60 |
+
return assistant_answer.strip() # Also strip here for consistency
|
61 |
|
62 |
# Display the chat history
|
63 |
for message in st.session_state.sessionMessages:
|