Update script_for_automation.py

script_for_automation.py

@@ -341,6 +341,15 @@ def format_json(json_data, truncate_length=500):
         # If it's not valid JSON, return the string as it is
         return json_data[:truncate_length] + "..." if len(json_data) > truncate_length else json_data
 
+# Custom method to handle all objects
+def custom_serializer(obj):
+    if isinstance(obj, Enum):
+        return obj.name  # Or obj.value, depending on what you need
+    if isinstance(obj, Soil):
+        return obj.to_dict()
+    if isinstance(obj, Yield):
+        return obj.to_dict()
+    return obj.__dict__  # Default case: use the __dict__ method for other custom objects
 
 def sanitize_json_for_yaml(data):
     if isinstance(data, dict):
@@ -416,7 +425,7 @@ def generate_markdown_output(df):
         markdown += f"\n### Recipe ID: {row['Recipe_ID']}\n"
         for key in ["planting", "interactions", "trials"]:
             gold = json.dumps(row['Gold_Standard_JSON'].get(key, {}), indent=2)
-            machine = json.dumps(row['Machine_Generated_JSON'].get(key, {}), indent=2)
+            machine = json.dumps(row['Machine_Generated_JSON'].get(key, {}), default=custom_serializer, indent=2)
             markdown += f"#### {key.capitalize()}\n"
             markdown += f"**Gold Standard JSON**:\n```json\n{gold}\n```\n"
             markdown += f"**Machine Generated JSON**:\n```json\n{machine}\n```\n"