Update test.py

test.py

@@ -1,7 +1,17 @@
 import pandas as pd
 import re
 
-
+def clean_llm_response(text):
+    # Remove lines with '--- | --- | ---' pattern
+    clean_text = re.sub(r'---(\s*\|\s*---)+', '', text)
+    
+    # Remove lines containing '**'
+    clean_text = re.sub(r'\*\*.*\*\*', '', clean_text)
+    
+    # Remove any resulting empty lines
+    clean_text = re.sub(r'\n\s*\n', '\n', clean_text)
+    
+    return clean_text
 
 # Function to parse tables from the response
 def parse_tables(response_text):