Update process_data.py

process_data.py

@@ -260,21 +260,58 @@ def parse_survey_stack_parameters(data):
     print("DATA 0 data preprompt")
     print(data[0]['data']['preprompt'])
     
-    if data[0]['data']['preprompt']['value'][0] == 'continue_preprompts':
+    #if data['group_2']['preprompt']['value'][0] == 'continue_preprompts':
+    #    processed_data["pre_prompt"] = True
+    #    processed_data["context_pre_prompt"] = data[0]['data']['contextpreprompt']['value']
+    #    processed_data["summary_pre_prompt"] = data[0]['data']['summarypreprompt']['value']
+    #    processed_data["conversation_pre_prompt"] = data[0]['data']['conversationpreprompt']['value']
+    #    processed_data["example_pre_prompt"] = data[0]['data']['examplepreprompt']['value']
+    #    processed_data["chaining"] = data[0]['data']['prepromptchaining']['value'][0]
+    #    if processed_data["chaining"] == "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["combined_prompt"] = combined_prompt
+    #        processed_data["chaining"] = False
+    #    else: 
+    #        processed_data["chaining"] = True
+    #        processed_data["combined_pre_prompt"] = None
+    #else:
+    #    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
+
+    if data['group_2']['preprompt']['value'][0] == 'continue_preprompts':
         processed_data["pre_prompt"] = True
-        processed_data["context_pre_prompt"] = data[0]['data']['contextpreprompt']['value']
-        processed_data["summary_pre_prompt"] = data[0]['data']['summarypreprompt']['value']
-        processed_data["conversation_pre_prompt"] = data[0]['data']['conversationpreprompt']['value']
-        processed_data["example_pre_prompt"] = data[0]['data']['examplepreprompt']['value']
-        processed_data["chaining"] = data[0]['data']['prepromptchaining']['value'][0]
-        if processed_data["chaining"] == "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"]]))
+
+        # 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]
+        processed_data["chaining"] = chaining_value == "yes"
+        
+        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["combined_prompt"] = combined_prompt
-            processed_data["chaining"] = False
-        else: 
-            processed_data["chaining"] = True
+        else:
+            # Set combined_pre_prompt to None if chaining is enabled
             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
@@ -282,6 +319,9 @@ def parse_survey_stack_parameters(data):
         processed_data["example_pre_prompt"] = None
         processed_data["chaining"] = None
         processed_data["combined_pre_prompt"] = None
+        
+    except KeyError as e:
+        print(f"An error occurred: {e}")
     
     return processed_data