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:
|