Update process_data.py

process_data.py

@@ -784,3 +784,93 @@ def parse_survey_stack_data(data):
     print(processed_data)
         
     return processed_data
+
+def parse_survey_stack(data):
+    """
+    Parse the incoming data from the survey stack survey
+
+    Args: 
+        data: (json) JSON retrieved from surveystack API after retrieving survey info/details
+
+    Returns: 
+        processed_data
+            processed_data["input_text"] = (str) the raw input text
+    """
+    print("PROCESSING SURVEY STACK DATA")
+    processed_data = {}
+
+    print("JUST PRINTING OUT THE DATA FOR YA")
+
+    print(data)
+
+    
+    processed_data["inputstyle"] = data[0]['data']['inputstyle']['value']
+    print("STEPWISE?: " + str(processed_data["inputstyle"]))
+
+    parameter_inputs = data[0]['data']['group_7']
+    processed_data["prompts"] = {}
+    
+    if processed_data["inputstyle"][0] == "individual-pieces-input-text":
+        print("IN THE STEP")
+        farm_management_inputs = data[0]['data']['group_4']
+        print("FARM MANAGEMENT INPUTS" + str(farm_management_inputs))
+        
+        processed_data["input_text_pieces"] = {}
+        processed_data["input_text_pieces"]["field_data_input"] = farm_management_inputs.get('field_data_input', {}).get('value', None)
+        processed_data["input_text_pieces"]["planting_data_input"] = farm_management_inputs.get('planting_data_input', {}).get('value', None)
+        processed_data["input_text_pieces"]["log_data_input"] = farm_management_inputs.get('log_data_input', {}).get('value', None)
+        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"]["interaction_data_input"] = interactions_inputs.get('interaction_data_input', {}).get('value', None)
+        processed_data["input_text_pieces"]["person_data_input"] = interactions_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"]["trial_data_input"] = trials_inputs.get('trial_data_input', {}).get('value', None)
+        processed_data["input_text_pieces"]["treatment_data_input"] = trials_inputs.get('treatment_data_input', {}).get('value', None)
+
+        print("NOW ONTO THE PROMPTS")
+        processed_data["prompts"]["first_schema_prompt_one"] = parameter_inputs.get('firstschemaprompt1', {}).get('value', None)
+        processed_data["prompts"]["first_schema_prompt_two"] = parameter_inputs.get('firstschemaprompt2', {}).get('value', None)
+        processed_data["prompts"]["first_schema_prompt_three"] = parameter_inputs.get('firstschemaprompt3', {}).get('value', None)
+        processed_data["prompts"]["first_schema_prompt_four"] = parameter_inputs.get('firstschemaprompt4', {}).get('value', None)
+        processed_data["prompts"]["first_schema_prompt_five"] = parameter_inputs.get('firstschemaprompt5', {}).get('value', None)
+        processed_data["prompts"]["second_schema_prompt_one"] = parameter_inputs.get('secondschemaprompt1', {}).get('value', None)
+        processed_data["prompts"]["second_schema_prompt_two"] = parameter_inputs.get('secondschemaprompt2', {}).get('value', None)
+        processed_data["prompts"]["third_schema_prompt_one"] = parameter_inputs.get('thirdschemaprompt1', {}).get('value', None)
+        processed_data["prompts"]["third_schema_prompt_two"] = parameter_inputs.get('thirdschemaprompt2', {}).get('value', None)
+        
+    elif processed_data["stepwise_json_creation"][0] == "big-block-input-text":
+        print("IN THE SINGLE")
+        processed_data["input_text"] = data[0]['data']['onelonginputtext']['value']
+        print(processed_data["input_text"])
+
+        print("NOW ONTO THE PROMPTS")
+        processed_data["prompts"]["first_schema_prompt"] = parameter_inputs.get('firstschemaprompt', {}).get('value', None)
+        processed_data["prompts"]["second_schema_prompt"] = parameter_inputs.get('secondschemaprompt', {}).get('value', None)
+        processed_data["prompts"]["third_schema_prompt"] = parameter_inputs.get('thirdschemaprompt', {}).get('value', None)
+        
+
+    # parameters
+    processed_data["parameters"] = {}
+    processed_data["parameters"]["model_version"] = data[0]['data']['modelversion']['value'][0] # this is one string
+    processed_data["parameters"]["pre_process"] = data[0]['data']['preprocessdata']['value'] # this is boolean
+    processed_data["parameters"]["prompt_style"] = data[0]['data']['promptstyle']['value'] # this is a list
+
+    if processed_data["parameters"]["pre_process"] == "yes":
+        processed_data["parameters"]["multiple_pre_prompts"] = data[0]['data']['multiplepreprompts']['value'] # this is a value in string words
+        processed_data["parameters"]["pre_prompt_style"] = data[0]['data']['prepromptstyle']['value'] # this is a list
+        processed_data["parameters"]["pre_prompt_first"] = data[0]['data']['preprocessingprompt1']['value']
+        processed_data["parameters"]["pre_prompt_second"] = data[0]['data']['preprocessingprompt2'].get('value', None)
+        processed_data["parameters"]["pre_prompt_third"] = data[0]['data']['preprocessingprompt3'].get('value', None)
+        
+    print("RETURNING DATA")
+    print(processed_data)
+        
+    return processed_data