Update test.py

test.py

@@ -1,6 +1,40 @@
 import pandas as pd
 import re
 
+import json
+import pandas as pd
+import re
+
+# The text output from the LLM (you would replace this with the actual output)
+llm_output = """
+... [Your provided text goes here] ...
+"""
+
+# Extract JSON strings using regex
+json_strings = re.findall(r'```json\n(.*?)```', llm_output, re.DOTALL)
+
+# Parse each JSON string and collect the data
+data = []
+for json_str in json_strings:
+    try:
+        parsed = json.loads(json_str)
+        entity_name = parsed['entity_name']
+        attributes = parsed['attributes'][0]
+        attributes['entity_name'] = entity_name
+        data.append(attributes)
+    except json.JSONDecodeError:
+        print(f"Error parsing JSON: {json_str}")
+
+# Create a pandas DataFrame
+df = pd.DataFrame(data)
+
+# Reorder columns to have 'entity_name' first
+cols = ['entity_name'] + [col for col in df.columns if col != 'entity_name']
+df = df[cols]
+
+# Display the DataFrame
+print(df)
+
 
 json_string = re.search(r"```(.*?)```", llm_output, re.DOTALL).group(1).strip()