Update app.py

app.py

@@ -3,16 +3,105 @@ from pydantic import BaseModel, Field, validator, ValidationError
 import gradio as gr
 from openai import OpenAI
 from typing import List, Dict, Any, Optional, Literal, Union
+from enum import Enum
 
 # Chatbot model 
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 client = OpenAI()
 
+# These are the necessary components that make up the Interactions
+class Date(BaseModel):
+    date: datetime = Field(..., description="Date of the interaction")
 
+class Role(str, Enum):
+    PARTNER = 'partner'
+    STAFF = 'staff'
+    AGRONOMIST = 'agronomist'
+    OTHER = 'other'
+    
+class Person(BaseModel):
+    name: str = Field(..., title="Name", description="Name of this person") 
+    role: str[Role] = Field(..., title="Role", description="Role of this person")
+    
 class Interactions(BaseModel):
-    pass
+    people: List[Person] = Field(..., title="People", description="People involved or mentioned during interaction")
+    date: Date = Field(..., title="Date", description="Date of the interaction")
+    next_meeting: Date = Field(..., title="Date of next meeting", description="Proposed date of the next future interaction")
+    next_steps: List[str] = Field(..., title="Next Steps", description="List of individual next steps derived from the interaction")
+    summary: str = Field(..., title="Summary", description="Summary of the interaction")
+
+# These are the components for Farm Activities, Fields, and Plantings
+class Status(str, Enum):
+    ACTIVE = 'active'
+    ARCHIVED = 'archived'
+
+# Depending on how well this works, come back and hard-code this based on some parameter(s)
+class Convention(str, Enum):
+    ACTIVITY: 'log--activity'
+    OBSERVATION: 'log--observation'
+    FLAMING: 'log--activity--flaming'
+    GRAZING: 'log--activity--grazing'
+    MOWING: 'log--activity--mowing'
+    SOLARIZATION: 'log--activity--solarization'
+    TERMINATION: 'log--activity--termination'
+    TILLAGE: 'log--activity--tillage'
+    HARVEST: 'log--activity--harvest'
+    HERBICIDE: 'log--input--herbicide_or_pesticide'
+    IRRIGATION: 'log--input--irrigation'
+    LIME: 'log--input--lime'
+    ORGANIC: 'log--input--organic_matter'
+    SEEDTREAT: 'log--input--seed_treatment'
+    SEEDLINGTREAT: 'log--input--seedling_treatment'
+    MODUS: 'log--lab_test--modus_lab_test'
+    SEEDING: 'log--seeding--seeding'
+    TRANSPLANT: 'log--transplanting--transplant'
+
+class Structure(str, Enum):
+    CLAY: 'clay'
+    SANDYCLAY: 'sandy clay'
+    SILTYCLAY: 'silty clay'
+    SANDYCLAYLOAM: 'sandy clay loam'
+    SILYCLAMLOAM: 'silty clay loam'
+    CLAYLOAM: 'clay loam'
+    SANDYLOAM: 'sandy loam'
+    SILTLOAM: 'silt loam'
+    LOAM: 'loam'
+    LOAMYSAND: 'loamy sand'
+    SAND: 'sand'
+    SILT: 'silt'
+    
+class Soil(BaseModel): 
+    description: str = Field(..., title="Description", description="A general description of the soil")
+    structure: str[Structure] = Field(..., title="Structure", description="The structure of the soil using options from the major soil texture classes (sand, clay, silt)")
+    
+class Log(BaseModel): 
+    convention: str[Convention] = Field(..., title="Logs", description="This log's convention (i.e. this log's category or type)")
+    date: Date = Field(..., title="Date", description="Date the log (i.e. action of the activity or input) was performed")
+    description: str = Field(..., title="Description", description="A description of the details of the log (i.e. details about farm activity performed")
+
+class Planting(BaseModel):
+    name: str = Field(..., title="Name", description="The name of the planting")
+    status: str[Status] = Field(..., title="Status", description="The status of the planting. \"active\" is a planting which is currently still in the field.  \"archived\" is a planting which is no longer in the field (has been terminated or harvested)")
+    crop: List[str] = Field(..., title="Crop", description="A list of the crops in this planting")
+    variety: List[str] = Field(..., title="Variety", description="A list of the crop varieties in this planting")
+    logs: List[Log] = Field(..., title="Logs", description="A list of all the logs that are associated with the farm activities")
+    structure: List[Structure] = Field(..., title="Structure", description="Information about the soil associated with or observed during the time of this planting")
+    biology: str = Field(..., title="Biology", description="Biological activity levels of the soil, including fluffiness, worms and bugs, and other evidence of soil biological activity")
 
+class Yield(BaseModel):
+    quantity: str = Field(..., title="Quantity", description="A description of the total yield (harvested amount) from this planting, including units when available")
+    quality: str = Field(..., title="Quality", description="The product quality of the harvest.  For example, small or large fruits, sweet or tart flavor, easily molding or containing mold, high number of product seconds, etc.")
     
+class Farm_Activities(BaseModel):
+    name: str = Field(..., title="Name", description="The name of the agricultural field.")
+    description: str = Field(..., title="Description", description="The description of the agricultural field.")
+    plantings: List[Planting] = Field(..., title="Plantings", description="All of the plantings which have occurred on this field.")
+    harvest_yield: Yield = Field(..., title="Yield", description="Quantitative and qualitative observations regarding the yield of harvest") 
+    
+# These are the components for Trials
+# TBD
+
+# This is to make stuff happen
 def generate_json(specification): 
     """
     Function to prompt OpenAI API to generate structured JSON output.
@@ -23,10 +112,10 @@ def generate_json(specification):
         response = client.beta.chat.completions.parse(
             model="gpt-4o-2024-08-06",  # Use GPT model that supports structured output
             messages=[
-                {"role": "system", "content": "Extract the farm management information."},
+                {"role": "system", "content": "Extract the farm information."},
                 {"role": "user", "content": specification}
             ],
-            response_format=Interactions,
+            response_format=Farm_Activities,
         )
         
         generated_json = response.choices[0].message.parsed