Update app.py

app.py

@@ -2260,6 +2260,9 @@ def handle_prompt(prompt):
                         "chat_history": st.session_state.chat_history
                     })
                     full_response = output["output"]
+                    import html
+                    escaped_text = full_response.replace("$", "\$")
+
 
                    
 
@@ -2271,10 +2274,8 @@ def handle_prompt(prompt):
                     trust_tip, suggestion = get_trust_tip_and_suggestion()
                         
                         # Combine the response text with Trust Tip and Suggestion
-                    combined_text = f"{full_response}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
-                    import html
-                    escaped_text = combined_text.replace("$", "\$")
-                    escaped_text_html = f"<pre>{escaped_text}</pre>"
+                    combined_text = f"{escaped_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
+                    
 
 
 
@@ -2288,8 +2289,8 @@ def handle_prompt(prompt):
 
 
                             #st.write(combined_text,unsafe_allow_html=True)
-                    st.session_state.chat_history.append({"role": "assistant", "content": escaped_text_html})
-                    copy_to_clipboard(escaped_text_html)
+                    st.session_state.chat_history.append({"role": "assistant", "content": combined_text})
+                    copy_to_clipboard(combined_text)
 
 
                 except Exception as e: