Update test.py

test.py

@@ -3,6 +3,26 @@ import re
 
 import pandas as pd
 
+def extract_table(text):
+    # Find the start and end of the table
+    start = text.find("| Biomarker Name |")
+    end = text.rfind("|", start)
+
+    # Extract the table portion
+    table_text = text[start:end].strip()
+
+    # Convert the table to a list of rows
+    rows = [row.strip().split("|")[1:-1] for row in table_text.split("\n") if "|" in row]
+
+    # Create a DataFrame from the rows
+    df = pd.DataFrame(rows[1:], columns=rows[0])
+    return df
+
+# Extract the final result table
+df_final_result = extract_table(llm_response)
+
+
+
 # Load your data (replace 'your_file.csv' with the actual file path)
 merge_data = pd.read_csv('your_file.csv')