Update process_data.py

process_data.py

@@ -15,17 +15,18 @@ os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 client = OpenAI()
 
 
-def generate_json(input_data, model_version): 
+def generate_json(input_data, parameters): 
     """
     Function to prompt OpenAI API to generate structured JSON output.
     """
 
     input_text = input_data["input_text"]
+    model_version = parameters["model_version"]
 
     farm_prompt = "Extract the farm information."
     interactions_prompt = "Extract the interactions information."
     trial_prompt = "Extract the trial information."
-
+    
     if input_data["input_context"]:
         farm_prompt = input_data["input_context"] + farm_prompt
         interactions_prompt = input_data["input_context"] + interactions_prompt
@@ -240,10 +241,10 @@ def pre_processing(input_data, parameters):
 def process_specifications(input_data, parameters):
     # here is where parsing and other things will happen before 
     if parameters["pre_prompt"] == True:
-        input_data = pre_processing(input_data, parameters)
-        return generate_json(input_data, parameters["model_version"])
+        processed_input = pre_processing(input_data, parameters)
+        return generate_json(processed_input, parameters)
     else:
-        return generate_json(input_data["input_text"], parameters["model_version"])
+        return generate_json(input_data, parameters)
     
     
 def parse_survey_stack_parameters(data):
@@ -317,7 +318,7 @@ def parse_survey_stack_parameters(data):
             processed_data["summary_pre_prompt"] = None
             processed_data["conversation_pre_prompt"] = None
             processed_data["example_pre_prompt"] = None
-            processed_data["chaining"] = None
+            processed_data["chaining"] = False
             processed_data["combined_pre_prompt"] = None
             
     except Exception as e: