Update process_data.py

process_data.py

@@ -173,11 +173,7 @@ def generate_json_pieces(processed_data):
 
     Args: 
         Args: 
-        input_data: (dict) The input data, preprocessed, from the user. Aka what will fill in the JSON
-            input_data["input_text"] = (str) the preprocessed input text
-            input_data["input_context"] = (str) depending on levers, empty or what is put in front of the prompt
-            input_data["input_text_pieces"] = (dict) containing the individual split up prompt pieces: field_data_input, planting_data_input, log_data_input, soil_data_input, yield_data_input
-            
+
         parameters: (dict) All of the individual parameters and "flippers" 
             parameters["model_version"] = (str) what model should be used 
             parameters["chaining"] = (bool) whether or not the preprocessed input context should be chained (given to multiple models)
@@ -200,30 +196,30 @@ def generate_json_pieces(processed_data):
     
     if processed_data["parameters"]["pre_process"] == "yes":
         print("Pre prompt is true")
-        field_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["field_data_input"]
-        planting_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["planting_data_input"]
-        log_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["log_data_input"]
-        soil_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["soil_data_input"] 
-        yield_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["yield_data_input"] 
+        field_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["field_data_input"]
+        planting_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["planting_data_input"]
+        log_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["log_data_input"]
+        soil_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["soil_data_input"] 
+        yield_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["yield_data_input"] 
 
-        interaction_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["interaction_data_input"]
-        person_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["person_data_input"]
+        interaction_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["interaction_data_input"]
+        person_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["person_data_input"]
 
-        trial_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["trial_data_input"]
-        treatment_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["treatment_data_input"]
+        trial_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["trial_data_input"]
+        treatment_data_input = processed_data["input_text_pieces"]["pre_processed_pieces"]["treatment_data_input"]
     else:
         print("Pre prompt is false")
-        field_data_input = input_data["input_text_pieces"]["field_data_input"]
-        planting_data_input = input_data["input_text_pieces"]["planting_data_input"]
-        log_data_input = input_data["input_text_pieces"]["log_data_input"]
-        soil_data_input = input_data["input_text_pieces"]["soil_data_input"] 
-        yield_data_input = input_data["input_text_pieces"]["yield_data_input"] 
+        field_data_input = processed_data["input_text_pieces"]["field_data_input"]
+        planting_data_input = processed_data["input_text_pieces"]["planting_data_input"]
+        log_data_input = processed_data["input_text_pieces"]["log_data_input"]
+        soil_data_input = processed_data["input_text_pieces"]["soil_data_input"] 
+        yield_data_input = processed_data["input_text_pieces"]["yield_data_input"] 
 
-        interaction_data_input = input_data["input_text_pieces"]["interaction_data_input"]
-        person_data_input = input_data["input_text_pieces"]["person_data_input"]
+        interaction_data_input = processed_data["input_text_pieces"]["interaction_data_input"]
+        person_data_input = processed_data["input_text_pieces"]["person_data_input"]
 
-        trial_data_input = input_data["input_text_pieces"]["trial_data_input"]
-        treatment_data_input = input_data["input_text_pieces"]["treatment_data_input"]
+        trial_data_input = processed_data["input_text_pieces"]["trial_data_input"]
+        treatment_data_input = processed_data["input_text_pieces"]["treatment_data_input"]
 
 
     # Fix these prompts for all