Update process_data.py

process_data.py

@@ -672,18 +672,20 @@ def parse_survey_stack(data):
         
 
     # parameters
+    parameter_data = data[0]['data']['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'][0] # this is yes / no
-    processed_data["parameters"]["prompt_style"] = data[0]['data']['promptstyle']['value'] # this is a list
+    processed_data["parameters"]["model_version"] = parameter_data.get('modelversion', {}).get('value')
+    processed_data["parameters"]["pre_process"] = parameter_data.get('preprocessdata', {}).get('value')
+    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"] = data[0]['data']['preprocessmodelversion']['value'][0]
-        processed_data["parameters"]["multiple_pre_prompts"] = data[0]['data']['multiplepreprompts']['value'] # this is a value in string words
-        processed_data["parameters"]["preprompt_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)
+        processed_data["parameters"]["pre_process_model_version"] = parameter_data.get('preprocessmodelversion', {}).get('value') 
+        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')
         
     print("RETURNING DATA")
     print(processed_data)