Update test.py

test.py

@@ -1,28 +1,18 @@
 import pandas as pd
 import re
 
-import pandas as pd
 
-{
-
-Biomarker Name Source: entity_name, 
-Biomarker Test Type: test_method_name, 
-Biomarker Variant Type:variant_type,  
-Exon Number Start: exon_location_start,
-Exon Number End: exon_location_end, 
-Chromosome Number: exon_location_chromosome_number,
-Genomic Position: exon_location_genomic_position, 
-Protein Alteration (p Notation Source): protein_change, 
-Trancript ID (Protein): protein_change_transcript_id, 
-DNA Alteration (c notation Source): nucleotide_change, 
-DNA Alteration (c notation Source): nucleotide_change_transcript_id,
-Biomarker Test Result Value Numeric 1: numeric_result_numeric_1, 
-Biomarker Test Result Value Numeric 2: numeric_result_numeric_2,
- Biomarker Test Threshold Value Numeric 1: numeric_result_threshold_1, 
-Biomarker Test Threshold Value Numeric 2: numeric_result_threshold_2,
-Biomarker Test Result Source: categorical_result
-
-}
+json_string = re.search(r"```(.*?)```", llm_output, re.DOTALL).group(1).strip()
+
+# Load the JSON string into a dictionary
+data = json.loads(json_string)
+
+# Convert the 'attributes' list to a DataFrame
+df = pd.DataFrame(data['attributes'])
+
+# Display the DataFrame
+print(df)
+
 
 def extract_table(text):
     # Find the start and end of the table