Update app.py

app.py

@@ -144,6 +144,7 @@ def send_feedback_via_email(name, email, feedback):
     except Exception as e:
         st.error(f"Error sending email: {e}")
 
+
 def clean_text(text):
     # Use BeautifulSoup to parse and clean HTML
     soup = BeautifulSoup(text, 'html.parser')
@@ -152,10 +153,17 @@ def clean_text(text):
     for a in soup.find_all('a'):
         a.replace_with(f"[{a.get_text()}]({a['href']})")
     
-    # Remove unwanted tags but preserve their text
-    for tag in ['span', 'i', 'b', 'u', 'em', 'strong']:
+    # Remove <i> and <em> tags but keep their content
+    for tag in ['i', 'em']:
         for element in soup.find_all(tag):
             element.unwrap()  # Remove the tag but keep the content
+
+    # Convert <b> and <strong> tags to Markdown bold
+    for tag in ['b', 'strong']:
+        for element in soup.find_all(tag):
+            element.insert_before('**')
+            element.insert_after('**')
+            element.unwrap()
     
     # Get the cleaned text
     cleaned_text = soup.get_text()
@@ -602,7 +610,8 @@ if prompt :
                 st.write(full_response)
     
         # Add AI response to chat history
-        st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
+        cleaned_response = clean_text(full_response)
+        st.session_state.chat_history.append({"role": "assistant", "content": cleaned_response})
         copy_to_clipboard(full_response)