Update process_data.py

process_data.py

@@ -283,45 +283,46 @@ def parse_survey_stack_parameters(data):
     #    processed_data["chaining"] = None
     #    processed_data["combined_pre_prompt"] = None
 
-    if data['group_2']['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)
-        
-        # Check if chaining is set to "yes" or "no"
-        chaining_value = data['group_2'].get('prepromptchaining', {}).get('value', [None])[0]
-        
-        if chaining_value == "no":
-            # Combine prompts if chaining is "no"
-            combined_prompt = " ".join(
-                filter(None, [
-                    processed_data["context_pre_prompt"], 
-                    processed_data["summary_pre_prompt"], 
-                    processed_data["conversation_pre_prompt"], 
-                    processed_data["example_pre_prompt"]
-                ])
-            )
-            processed_data["chaining"] = False
-            processed_data["combined_prompt"] = combined_prompt
+    try:
+        if data['group_2']['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)
+            
+            # Check if chaining is set to "yes" or "no"
+            chaining_value = data['group_2'].get('prepromptchaining', {}).get('value', [None])[0]
+            
+            if chaining_value == "no":
+                # Combine prompts if chaining is "no"
+                combined_prompt = " ".join(
+                    filter(None, [
+                        processed_data["context_pre_prompt"], 
+                        processed_data["summary_pre_prompt"], 
+                        processed_data["conversation_pre_prompt"], 
+                        processed_data["example_pre_prompt"]
+                    ])
+                )
+                processed_data["chaining"] = False
+                processed_data["combined_prompt"] = combined_prompt
+            else:
+                # Set combined_pre_prompt to None if chaining is enabled
+                processed_data["chaining"] = True
+                processed_data["combined_pre_prompt"] = None
         else:
-            # Set combined_pre_prompt to None if chaining is enabled
-            processed_data["chaining"] = True
+            # Set fields to None if preprompt is not "continue_preprompts"
+            processed_data["pre_prompt"] = False
+            processed_data["context_pre_prompt"] = None
+            processed_data["summary_pre_prompt"] = None
+            processed_data["conversation_pre_prompt"] = None
+            processed_data["example_pre_prompt"] = None
+            processed_data["chaining"] = None
             processed_data["combined_pre_prompt"] = None
-    else:
-        # Set fields to None if preprompt is not "continue_preprompts"
-        processed_data["pre_prompt"] = False
-        processed_data["context_pre_prompt"] = None
-        processed_data["summary_pre_prompt"] = None
-        processed_data["conversation_pre_prompt"] = None
-        processed_data["example_pre_prompt"] = None
-        processed_data["chaining"] = None
-        processed_data["combined_pre_prompt"] = None
-        
-    except KeyError as e:
+            
+    except Exception as e:
         print(f"An error occurred: {e}")
     
     return processed_data