Update app.py

app.py

@@ -151,13 +151,16 @@ def clean_text(text):
     text = text.replace('\\n', '\n')
     
     # Remove any span and italic tags
+    text = re.sub(r'<[^>]+>', '', text)
+
     text = re.sub(r'<span[^>]*>', '', text)
     text = re.sub(r'</span>', '', text)
     text = re.sub(r'<i[^>]*>', '', text)
     text = re.sub(r'</i>', '', text)
     text = re.sub(r'<span[^>]*>.*?</span>', '', text, flags=re.DOTALL)
     text = re.sub(r'<span[^>]*>.*?</span>', '', text, flags=re.DOTALL)
-  
+
+
              
     
     # Preserve and correctly format markdown links (don't modify URLs)
@@ -195,7 +198,19 @@ def clean_text(text):
     cleaned_text = '\n\n'.join(para for para in cleaned_paragraphs if para)
     
     return cleaned_text
-
+    
+def add_trust_tip_and_suggestion(cleaned_text):
+    # Check if Trust Tip is already present
+    if "**Trust Tip**" not in cleaned_text:
+        trust_tip = random.choice(trust_tips)
+        cleaned_text += f"\n\n**Trust Tip**: {trust_tip}"
+    
+    # Check if Suggestion is already present
+    if "**Suggestion**" not in cleaned_text:
+        suggestion = random.choice(suggestions)
+        cleaned_text += f"\n\n**Suggestion**: {suggestion}"
+    
+    return cleaned_text
 
 def side():        
     with st.sidebar.form(key='feedback_form'):
@@ -601,18 +616,13 @@ if prompt :
                 cleaned_text = clean_text(full_response)
     
                 
-                cleaned_text = re.sub(r'</span>', '', cleaned_text) 
-                cleaned_text = re.sub(r'<span[^>]*>', '', cleaned_text)
+                #cleaned_text = re.sub(r'</span>', '', cleaned_text) 
+                #cleaned_text = re.sub(r'<span[^>]*>', '', cleaned_text)
                 # Display the response
-                if "**Trust Tip**" not in cleaned_text:
-                    trust_tip = random.choice(trust_tips)
-                    cleaned_text += f"\n\n**Trust Tip**: {trust_tip}"
-                
-                if "**Suggestion**" not in cleaned_text:
-                    suggestion = random.choice(suggestions)
-                    cleaned_text += f"\n\n**Suggestion**: {suggestion}"
+                final_text = add_trust_tip_and_suggestion(cleaned_text)
+
                     
-                st.markdown(cleaned_text, unsafe_allow_html=True)
+                st.markdown(final_text, unsafe_allow_html=True)
     
     
     
@@ -625,7 +635,7 @@ if prompt :
     
         # Add AI response to chat history
 
-        st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
+        st.session_state.chat_history.append({"role": "assistant", "content": final_text})
         copy_to_clipboard(full_response)