Update process_data.py

process_data.py

@@ -31,6 +31,8 @@ def generate_json(input_data, parameters):
         farm_prompt = input_data["input_context"] + farm_prompt
         interactions_prompt = input_data["input_context"] + interactions_prompt
         trial_prompt = input_data["input_context"] + trial_prompt
+
+        
         
     try:
         #Call OpenAI API to generate structured output based on prompt
@@ -209,20 +211,20 @@ def pre_processing(input_data, parameters):
         pre_processing_list = [parameters["context_pre_prompt"], parameters["summary_pre_prompt"], parameters["conversation_pre_prompt"], parameters["example_pre_prompt"]]
         
         for pre_prompt in pre_processing_list:
-            
-            response = client.chat.completions.create(
-                model=parameters["model_version"],
-                messages=[
-                    {"role": "system", "content": pre_prompt},
-                    {"role": "user", "content": input_text}
-                ]
-            )
-        
-            response_text = response['choices'][0]['message']['content']
-            
             try:
-                partially_processed_input = json.loads(response_text)
-                input_text = partially_processed_input
+                
+                if pre_prompt:
+                    response = client.chat.completions.create(
+                        model=parameters["model_version"],
+                        messages=[
+                            {"role": "system", "content": pre_prompt},
+                            {"role": "user", "content": input_text}
+                        ]
+                    )
+                
+                    response_text = response['choices'][0]['message']['content']
+                    partially_processed_input = json.loads(response_text)
+                    input_text = partially_processed_input
                 
             except Exception as e:
                 print("Failed to parse response as JSON. Error was:")