Update process_data.py

process_data.py

@@ -138,20 +138,21 @@ def generate_json_pieces(input_data, parameters):
     print(specification)
     print(model_version)
     
-    if parameters["pre_prompt"] == True:
+    if parameters["pre_prompt"]:
         print("Pre prompt is true")
-        field_data_input = input_data["input_text_pieces"]["field_data_input"]
-        planting_data_input = input_data["input_text_pieces"]["planting_data_input"]
-        logs_data_input = input_data["input_text_pieces"]["logs_data_input"]
-        soil_data_input = input_data["input_text_pieces"]["soil_data_input"] 
-        yield_data_input = input_data["input_text_pieces"]["yield_data_input"] 
-    else:
-        print("Pre prompt is false")
         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"]
         logs_data_input = input_data["input_text_pieces"]["pre_processed_pieces"]["logs_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"] 
+    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"]
+        logs_data_input = input_data["input_text_pieces"]["logs_data_input"]
+        soil_data_input = input_data["input_text_pieces"]["soil_data_input"] 
+        yield_data_input = input_data["input_text_pieces"]["yield_data_input"] 
+
 
     print("Setting prompts")
     field_prompt = "Extract the field information."