Update app.py

app.py

@@ -11,8 +11,8 @@ os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 client = OpenAI()
 
 # These are the necessary components that make up the Trials 
-#class Variables(BaseModel):
-#    controlled: List[str] = Field(..., title="Controlled Variables", description="A list of controlled variables, which will be constant (controlled) across all trials")
+class Variables(BaseModel):
+    controlled: List[str] = Field(..., title="Controlled Variables", description="A list of controlled variables, which will be constant (controlled) across all trials")
 #    independent: List[str] = Field(..., title="Independent Variables", description="A list of independent variables (ie treatments), which will be intentionally varied across one or more trials")
 #    outcome: List[str] = Field(..., title="Outcome Variables", description="A list of outcome variables (ie dependent or response variables)")
 
@@ -22,8 +22,8 @@ class Treatment(BaseModel):
     crops: List[str] = Field(..., title="Crops", description="A list of crops being tested in this treatment")
     fields: List[str] = Field(..., title="Fields", description="A list of fields in which this treatment has occured or will occur")
     learnings: List[str] = Field(..., title="Learnings", description="A list of lessons learned from this experiment")
+    variables: Variables = Field(..., title="Variables", description="Variables (ie factors) in this experiment. Some variables are constant (controlled) and some will vary in order to learn something (independent)")
     confoundingFactors: List[str] = Field(..., title="Confounding Factors", description="A list of factors which may impact the outcomes of the experiment that were not planned for")
-    #variables: Variables = Field(..., title="Variables", description="Variables (ie factors) in this experiment. Some variables are constant (controlled) and some will vary in order to learn something (independent)")
 
 class Trial(BaseModel):
     name: str = Field(..., title="Name", description="The name of this trial")