Update process_data.py

process_data.py

@@ -7,7 +7,7 @@ from enum import Enum
 from gradio_toggle import Toggle
 import json
 
-from schema_classes import FarmActivities, Interactions, Trial, FarmActivitiesLite, PlantingLite, Log, Soil, Yield, InteractionsLite, TrialLite, Person
+from schema_classes import FarmActivities, Interactions, Trial, FarmActivitiesLite, PlantingLite, Log, Soil, Yield, InteractionsLite, TrialLite, Person, Treatment
 
 
 # This API key must be in a "secret" in your environment. This is generated from OpenAI or the company's website that creates the model you wish to engage with. 
@@ -448,7 +448,7 @@ def generate_json_pieces(processed_data):
 
         trials = {
             **trial_pretty_json,
-            "treatments": treatment_generated_json
+            "treatments": treatment_pretty_json
         }
 
         print("ADDED DICTS 3")