Update script_for_automation.py

script_for_automation.py

@@ -141,22 +141,28 @@ def get_recipes():
         my_recipes = []
         for row in results: 
             print(f"Row ID: {row.get('id')}, Data: {row}")
-            recipe_schema_processing_model = row.get("Schema Processing Model")
+            recipe_id = row.get("Recipe ID")
+            testing_strategy_text = row.get("Testing Strategy for Set")
+            schema_processing_model = row.get("Schema Processing Model")
             pre_processing_strategy = row.get("Pre-Processing Strategy")
-            pre_processing_text = row.get("Preprocessing Model")
+            pre_processing_text = row.get("Pre-Prompt Text")
+            pre_processing_model = row.get("Preprocessing Model")
             prompting_strategy = row.get("Prompting Strategy")
             plantings_and_fields_prompt = row.get("Plantings and Fields Prompting Text")
             interactions_prompt = row.get("Interactions Prompting Text")
             treatments_prompt = row.get("Treatments Prompting Text")
 
             recipe_dict = {
-                "schema_processing_model": recipe_schema_processing_model,
-                "pre_processing_strategy": pre_processing_strategy,
-                "pre_processing_text": pre_processing_text,
-                "prompting_strategy": prompting_strategy,
-                "plantings_and_fields_prompt": plantings_and_fields_prompt,
-                "interactions_prompt": interactions_prompt,
-                "treatments_prompt": treatments_prompt
+                "recipe_id": recipe_id,
+                "testing_strategy_text": testing_strategy_text,
+                "schema_processing_model", schema_processing_model,
+                "pre_processing_strategy", pre_processing_strategy,
+                "pre_processing_text", pre_processing_text,
+                "pre_processing_model", pre_processing_model,
+                "prompting_strategy", prompting_strategy,
+                "plantings_and_fields_prompt", plantings_and_fields_prompt,
+                "interactions_prompt", interactions_prompt,
+                "treatments_prompt", treatments_prompt
             }
             
             my_recipes.append(recipe_dict)
@@ -175,6 +181,8 @@ def fill_out_survey(recipe_dict, input_data):
         pre_processing = False
         pre_process = "no"
         pre_process_model_version = "None"
+    else:
+        pre_process = recipe_dict
 
     # Set the prompting strategy to be a variable from the list 
     # Do this here 
@@ -190,15 +198,17 @@ def fill_out_survey(recipe_dict, input_data):
                     "secondschemaprompt": recipe_dict["interactions_prompt"],
                     "thirdschemaprompt": recipe_dict["treatments_prompt"],
                 },
-                "parameters": {
-                    "model_version": recipe_dict["schema_processing_model"],
-                    "preprocessdata": ["yes"],
-                    "promptstyle": recipe_dict["prompting_strategy"], # replace this one
-                    "preprocessmodelversion": "2.0",
-                    "multiplepreprompts": "no",
-                    "prepromptstyle": recipe_dict["pre_processing_strategy"],
-                    "preprocessingprompt1": recipe_dict["pre_processing_text"],
-                }
+            },
+            "parameters": {
+                "modelversion": recipe_dict["schema_processing_model"],
+                "preprocessdata": ["yes"],
+                "promptstyle": recipe_dict["prompting_strategy"],
+                "preprocessmodelversion": recipe_dict["prompting_strategy"],
+                "multiplepreprompts": "no",
+                "prepromptstyle": recipe_dict["pre_processing_strategy"],
+                "preprocessingprompt1": recipe_dict["pre_processing_text"],
+                "preprocessingprompt2": "",
+                "preprocessingprompt3": ""
             }
         }
         
@@ -213,11 +223,18 @@ def fill_out_survey(recipe_dict, input_data):
                     "secondschemaprompt": recipe_dict["interactions_prompt"],
                     "thirdschemaprompt": recipe_dict["treatments_prompt"],
                 },
-                "parameters": {
-                    "model_version": recipe_dict["schema_processing_model"],
-                    "pre_process": "no",
-                    "prompt_style": recipe_dict["prompting_strategy"], # replace this one
-                }
+            }
+            "parameters": {
+                "modelversion": recipe_dict["schema_processing_model"],
+                "preprocessdata": ["no"],
+                "promptstyle": recipe_dict["prompting_strategy"],
+                "preprocessmodelversion": None,
+                "multiplepreprompts": "no",
+                "prepromptstyle": None,
+                "preprocessingprompt1": None,
+                "preprocessingprompt2": None,
+                "preprocessingprompt3": None
+                
             }
         }