Update process_data.py

process_data.py

@@ -100,9 +100,9 @@ def generate_json(processed_data):
     input_text = processed_data["input_text"]
     model_version = processed_data["parameters"]["model_version"]
 
-    farm_prompt = processed_data["prompts"]["first_schema_prompt"]
-    interactions_prompt = processed_data["prompts"]["second_schema_prompt"]
-    trial_prompt = processed_data["prompts"]["third_schema_prompt"]
+    farm_prompt = processed_data["prompts"]["firstschemaprompt"]
+    interactions_prompt = processed_data["prompts"]["secondschemaprompt"]
+    trial_prompt = processed_data["prompts"]["thirdschemaprompt"]
 
         
     try:
@@ -476,13 +476,13 @@ def pre_processing(processed_data):
     print("Starting preprocessing")
 
 
-    pre_processing_list = [processed_data.get('parameters', {}).get('pre_prompt_first', None), processed_data.get('parameters', {}).get('pre_prompt_second', None), processed_data.get('parameters', {}).get('pre_prompt_third', None)]
+    pre_processing_list = [processed_data.get('parameters', {}).get('preprocessingprompt1', None), processed_data.get('parameters', {}).get('preprocessingprompt2', None), processed_data.get('parameters', {}).get('preprocessingprompt3', None)]
 
     print("Preprocessing list")
     print(pre_processing_list)
 
     print("Model Version")
-    model_version = processed_data["parameters"]["pre_process_model_version"]
+    model_version = processed_data["parameters"]["preprocessmodelversion"]
     print(model_version)
     
     if processed_data["inputstyle"] == "individual-pieces-input-text":
@@ -669,9 +669,9 @@ def parse_survey_stack(data):
         print(processed_data["input_text"])
 
         print("NOW ONTO THE PROMPTS from Input Item")
-        processed_data["prompts"]["first_schema_prompt"] = input_item.get('firstschemaprompt', {}).get('value')
-        processed_data["prompts"]["second_schema_prompt"] = input_item.get('secondschemaprompt', {}).get('value')
-        processed_data["prompts"]["third_schema_prompt"] = input_item.get('thirdschemaprompt', {}).get('value')
+        processed_data["prompts"]["firstschemaprompt"] = input_item.get('firstschemaprompt', {}).get('value')
+        processed_data["prompts"]["secondschemaprompt"] = input_item.get('secondschemaprompt', {}).get('value')
+        processed_data["prompts"]["thirdschemaprompt"] = input_item.get('thirdschemaprompt', {}).get('value')
         
 
     # parameters
@@ -679,17 +679,17 @@ def parse_survey_stack(data):
     print("PARAMETERS" + str(parameter_data))
     
     processed_data["parameters"] = {}
-    processed_data["parameters"]["model_version"] = parameter_data.get('modelversion', {}).get('value')[0]
-    processed_data["parameters"]["pre_process"] = parameter_data.get('preprocessdata', {}).get('value')[0]
-    processed_data["parameters"]["prompt_style"] = parameter_data.get('promptstyle', {}).get('value') 
-
-    if processed_data["parameters"]["pre_process"] == "yes":
-        processed_data["parameters"]["pre_process_model_version"] = parameter_data.get('preprocessmodelversion', {}).get('value')[0] 
-        processed_data["parameters"]["multiple_pre_prompts"] =  parameter_data.get('multiplepreprompts', {}).get('value')
-        processed_data["parameters"]["preprompt_style"] = parameter_data.get('prepromptstyle', {}).get('value')
-        processed_data["parameters"]["pre_prompt_first"] = parameter_data.get('preprocessingprompt1', {}).get('value') 
-        processed_data["parameters"]["pre_prompt_second"] = parameter_data.get('preprocessingprompt2', {}).get('value')
-        processed_data["parameters"]["pre_prompt_third"] = parameter_data.get('preprocessingprompt3', {}).get('value')
+    processed_data["parameters"]["modelversion"] = parameter_data.get('modelversion', {}).get('value')[0]
+    processed_data["parameters"]["preprocessdata"] = parameter_data.get('preprocessdata', {}).get('value')[0]
+    processed_data["parameters"]["promptstyle"] = parameter_data.get('promptstyle', {}).get('value') 
+
+    if processed_data["parameters"]["preprocessdata"] == "yes":
+        processed_data["parameters"]["preprocessmodelversion"] = parameter_data.get('preprocessmodelversion', {}).get('value')[0] 
+        processed_data["parameters"]["multiplepreprompts"] =  parameter_data.get('multiplepreprompts', {}).get('value')
+        processed_data["parameters"]["prepromptstyle"] = parameter_data.get('prepromptstyle', {}).get('value')
+        processed_data["parameters"]["preprocessingprompt1"] = parameter_data.get('preprocessingprompt1', {}).get('value') 
+        processed_data["parameters"]["preprocessingprompt2"] = parameter_data.get('preprocessingprompt2', {}).get('value')
+        processed_data["parameters"]["preprocessingprompt3"] = parameter_data.get('preprocessingprompt3', {}).get('value')
         
     print("RETURNING DATA")
     print(processed_data)