Update process_data.py

process_data.py

@@ -183,6 +183,7 @@ def generate_json_pieces(input_data, parameters):
         # Call OpenAI API to generate structured output based on prompt
         print("Getting all responses in pieces, starting with field response")
 
+        # All of this is for the first schema for farm activities 
         print("Field prompt")
         print(field_prompt)
 
@@ -286,10 +287,14 @@ def generate_json_pieces(input_data, parameters):
         print("ADDED DICTS")
         print(farm_activities)
         print("FINAL JSON: ")
-        final_pretty_json = json.dumps(farm_activities, indent=4)
-        print(final_pretty_json)
+        final_pretty_farm_activity_json = json.dumps(farm_activities, indent=4)
+        print(final_pretty_farm_activity_json)
 
-        return final_pretty_json, final_pretty_json, final_pretty_json
+
+        # This is for the second schema now, interactions
+        
+
+        return final_pretty_farm_activity_json, final_pretty_json, final_pretty_json
     except Exception as e:
         return {"error": "Failed to generate valid JSON. " + str(e)}
     
@@ -542,6 +547,22 @@ def parse_survey_stack_data(data):
         processed_data["input_text_pieces"]["soil_data_input"] = farm_management_inputs.get('soil_data_input', {}).get('value', None)
         processed_data["input_text_pieces"]["yield_data_input"] = farm_management_inputs.get('yield_data_input', {}).get('value', None)
         processed_data["input_text"] = "EMPTY"
+
+        print("NEXT SCHEMA INPUTS")
+        interactions_inputs = data[0]['data']['group_5']
+        print("INTERACTIONS INPUTS" + str(interactions_inputs))
+        processed_data["input_text_pieces_second_schema"] = {}
+        processed_data["input_text_pieces_second_schema"]["interaction_data_input"] = farm_management_inputs.get('interaction_data_input', {}).get('value', None)
+        processed_data["input_text_pieces_second_schema"]["person_data_input"] = farm_management_inputs.get('person_data_input', {}).get('value', None)
+
+        print("NEXT SCHEMA INPUTS 2")
+        trials_inputs = data[0]['data']['group_6']
+        print("TRIALS INPUTS" + str(trials_inputs))
+        processed_data["input_text_pieces_third_schema"] = {}
+        processed_data["input_text_pieces_third_schema"]["trial_data_input"] = farm_management_inputs.get('trial_data_input', {}).get('value', None)
+        processed_data["input_text_pieces_third_schema"]["treatment_data_input"] = farm_management_inputs.get('treatment_data_input', {}).get('value', None)
+
+        
     elif processed_data["stepwise_json_creation"][0] == "singlejsoncreation":
         print("IN THE SINGLE")
         processed_data["input_text"] = data[0]['data']['onelonginputtext']['value']
@@ -554,6 +575,14 @@ def parse_survey_stack_data(data):
         processed_data["input_text_pieces"]["soil_data_input"] = "EMPTY"
         processed_data["input_text_pieces"]["yield_data_input"] = "EMPTY"
 
+        processed_data["input_text_pieces_second_schema"] = {}
+        processed_data["input_text_pieces_second_schema"]["interaction_data_input"] = "EMPTY"
+        processed_data["input_text_pieces_second_schema"]["person_data_input"] = "EMPTY"
+
+        processed_data["input_text_pieces_third_schema"] = {}
+        processed_data["input_text_pieces_third_schema"]["trial_data_input"] = "EMPTY"
+        processed_data["input_text_pieces_third_schema"]["treatment_data_input"] = "EMPTY"
+
     print("RETURNING DATA")
     print(processed_data)