Update app.py

app.py

@@ -491,19 +491,22 @@ if prompt :
                 "chat_history": st.session_state.chat_history
             })
             full_response = output["output"]
-            st.write(full_response, unsafe_allow_html=True)
+
+            
+            full_response = re.sub(r'\s+', ' ', full_response).strip()
+
 
 
             # Display the response
 
-            #st.write(full_response)
+            st.write(full_response)
             follow_up_output = agent_executor.invoke({
                 "input": f"Extract 1-2 follow-up questions from the following text: {full_response}",
                 "chat_history": st.session_state.chat_history
             })
             follow_up_questions = follow_up_output["output"]
             if follow_up_questions:
-                st.write(f"Follow-up questions:\n{follow_up_questions}")
+                st.markdown(f"Follow-up questions:\n{follow_up_questions} \n")
 
 
         except Exception as e: