Update process_data.py

process_data.py

@@ -134,19 +134,26 @@ def generate_json_pieces(input_data, parameters):
     specification = input_data["input_text"]
     model_version = parameters["model_version"]
 
+    print("Specification and Model Version")
+    print(specification)
+    print(model_version)
+    
     if parameters["pre_prompt"] == True:
+        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"] 
 
+    print("Setting prompts")
     field_prompt = "Extract the field information."
     plant_prompt = "Extract the planting information."
     log_prompt = "Extract the log information."
@@ -166,6 +173,7 @@ def generate_json_pieces(input_data, parameters):
 
     try:
         # Call OpenAI API to generate structured output based on prompt
+        print("Getting all responses in pieces, starting with field response")
         field_response = client.beta.chat.completions.parse(
             model=model_version,  # Use GPT model that supports structured output
             messages=[