Update process_data.py

process_data.py

@@ -250,15 +250,6 @@ def parse_survey_stack_parameters(data):
     processed_data = {}
 
     processed_data["model_version"] = data[0]['data']['modelversion']['value'][0]
-
-    print("DATA")
-    print(data)
-
-    print("DATA 0 data")
-    print(data[0]['data'])
-
-    print("DATA 0 data preprompt")
-    print(data[0]['data']['preprompt'])
     
     #if data['group_2']['preprompt']['value'][0] == 'continue_preprompts':
     #    processed_data["pre_prompt"] = True
@@ -284,17 +275,20 @@ def parse_survey_stack_parameters(data):
     #    processed_data["combined_pre_prompt"] = None
 
     try:
-        if data['group_2']['preprompt']['value'][0] == 'continue_preprompts':
+        parameters_json = json.loads(data)
+        pre_promp_parameters = parameters_json['data']['group_2']
+
+        if pre_promp_parameters['preprompt']['value'][0] == 'continue_preprompts':
             processed_data["pre_prompt"] = True
     
             # Accessing context and other prompts, with defaults in case they are None
-            processed_data["context_pre_prompt"] = data['group_2'].get('contextpreprompt', {}).get('value', None)
-            processed_data["summary_pre_prompt"] = data['group_2'].get('summarypreprompt', {}).get('value', None)
-            processed_data["conversation_pre_prompt"] = data['group_2'].get('conversationpreprompt', {}).get('value', None)
-            processed_data["example_pre_prompt"] = data['group_2'].get('examplepreprompt', {}).get('value', None)
+            processed_data["context_pre_prompt"] = pre_promp_parameters.get('contextpreprompt', {}).get('value', None)
+            processed_data["summary_pre_prompt"] = pre_promp_parameters.get('summarypreprompt', {}).get('value', None)
+            processed_data["conversation_pre_prompt"] = pre_promp_parameters.get('conversationpreprompt', {}).get('value', None)
+            processed_data["example_pre_prompt"] = pre_promp_parameters.get('examplepreprompt', {}).get('value', None)
             
             # Check if chaining is set to "yes" or "no"
-            chaining_value = data['group_2'].get('prepromptchaining', {}).get('value', [None])[0]
+            chaining_value = pre_promp_parameters.get('prepromptchaining', {}).get('value', [None])[0]
             
             if chaining_value == "no":
                 # Combine prompts if chaining is "no"