archana2324 commited on
Commit
3e8a358
·
verified ·
1 Parent(s): 7ca7c57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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: