Update app.py

app.py

@@ -244,7 +244,15 @@ def search_knowledge_base(query):
     except Exception as e:
         logger.error(f"Error searching knowledge base: {e}")
         return ["Error occurred during knowledge base search"]
-
+def clean_text(text):
+    # Remove HTML tags and inline styles/classes
+    text = re.sub(r'<span[^>]*>', '', text)  # Remove <span> tags with attributes
+    text = re.sub(r'</span>', '', text)      # Remove closing </span> tags
+    text = re.sub(r'<[^>]+>', '', text)      # Remove any remaining HTML tags
+    text = re.sub(r'\s+', ' ', text)         # Replace multiple spaces or newlines with a single space
+    text = re.sub(r'([a-z])(\d)', r'\1 \2', text)  # Add a space between letters and numbers if necessary
+    return text.strip()
+    
 def google_search(query):
     try:
         search_client = serpapi.Client(api_key=serper_api_key)
@@ -528,8 +536,7 @@ if prompt :
             })
             full_response = output["output"]
 
-            cleaned_text = re.sub(r'\s+', ' ', full_response)  # Replace multiple spaces or newlines with a single space
-            cleaned_text = re.sub(r'([a-z])(\d)', r'\1 \2', full_response)  # Add a space between letters and numbers if necessary
+            cleaned_text = clean_text(full_response) # Add a space between letters and numbers if necessary
 
             # Display the response
             if "**Trust Tip**" not in full_response and "**Suggestion**" not in full_response: