Update app.py

app.py

@@ -2274,6 +2274,11 @@ def handle_prompt(prompt):
                     full_response = output["output"]
 
                     cleaned_text = minimal_clean_text(full_response)
+                    html_content = f"""
+                    <div style="white-space: pre-wrap; word-break: break-word;">
+                    {cleaned_text}
+                    </div>
+                    """
 
                     # Use Markdown with a styled div to ensure proper word wrapping
                     
@@ -2284,11 +2289,11 @@ def handle_prompt(prompt):
                         trust_tip, suggestion = get_trust_tip_and_suggestion()
                         
                         # Combine the response text with Trust Tip and Suggestion
-                        combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
+                        combined_text = f"{html_content}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
                         
                     else:
                         # If already present, just use the full response
-                        combined_text = cleaned_text
+                        combined_text = html_content
                     with response_placeholder:
                         with st.chat_message("assistant"):
                             st.markdown(combined_text, unsafe_allow_html=True)