update to include stepwise json creation

app.py

@@ -10,22 +10,22 @@ os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 client = OpenAI()
 
 # These are the necessary components that make up the Interactions
-class Role(str, Enum):
-    PARTNER = 'partner'
-    STAFF = 'staff'
-    AGRONOMIST = 'agronomist'
-    OTHER = 'other'
+#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: Role = Field(..., title="Role", description="Role of this person")
+#class Person(BaseModel):
+#    name: str = Field(..., title="Name", description="Name of this person") 
+#    role: Role = Field(..., title="Role", description="Role of this person")
     
-class Interactions(BaseModel):
-    people: List[Person] = Field(..., title="People", description="People involved or mentioned during interaction")
-    date: str = Field(..., title="Date of current interaction", description="Date of the interaction")
-    nextMeeting: str = Field(..., title="Date of next meeting", description="Proposed date of the next future interaction")
-    nextSteps: 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")
+#class Interactions(BaseModel):
+#    people: List[Person] = Field(..., title="People", description="People involved or mentioned during interaction")
+#    date: str = Field(..., title="Date of current interaction", description="Date of the interaction")
+#    nextMeeting: str = Field(..., title="Date of next meeting", description="Proposed date of the next future interaction")
+#    nextSteps: 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):
@@ -95,6 +95,17 @@ class FarmActivities(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.")
+
+# These are extra for the modular approach 
+class FarmActivitiesLite(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.")
+    
+class PlantingLite(BaseModel):
+    name: str = Field(..., title="Name", description="The name of the planting")
+    status: 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")
     
 # These are the components for Trials
 # TBD
@@ -131,13 +142,84 @@ def generate_json(specification, model_version):
     except Exception as e:
         return {"error": "Failed to generate valid JSON. " + str(e)}
 
+# This is for the step-wise JSON creation
+def generate_json_pieces(specification, model_version, additional_json_creation_options, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input): 
+    
+    if additional_json_creation_options == "Explicit specific pieces":
+    
+        try:
+            # Call OpenAI API to generate structured output based on prompt
+            field_response = client.beta.chat.completions.parse(
+                model=model_version,  # Use GPT model that supports structured output
+                messages=[
+                    {"role": "system", "content": "Extract the field information."},
+                    {"role": "user", "content": field_data_input}
+                ],
+                response_format=FarmActivitiesLite,
+            )
+            
+            plant_response = client.beta.chat.completions.parse(
+                model=model_version,  # Use GPT model that supports structured output
+                messages=[
+                    {"role": "system", "content": "Extract the planting information."},
+                    {"role": "user", "content": planting_data_input}
+                ],
+                response_format=PlantingLite,
+            )
+
+            log_response = client.beta.chat.completions.parse(
+                model=model_version,  # Use GPT model that supports structured output
+                messages=[
+                    {"role": "system", "content": "Extract the planting information."},
+                    {"role": "user", "content": logs_data_input}
+                ],
+                response_format=Log,
+            )
+
+            soil_response = client.beta.chat.completions.parse(
+                model=model_version,  # Use GPT model that supports structured output
+                messages=[
+                    {"role": "system", "content": "Extract the planting information."},
+                    {"role": "user", "content": soil_data_input}
+                ],
+                response_format=Soil,
+            )
+
+            yield_response = client.beta.chat.completions.parse(
+                model=model_version,  # Use GPT model that supports structured output
+                messages=[
+                    {"role": "system", "content": "Extract the planting information."},
+                    {"role": "user", "content": yield_data_input}
+                ],
+                response_format=Yield,
+            )
+            
+            combined_json = field_response.choices[0].message.parsed.copy()
+            combined_json["plantings"] = plant_response.choices[0].message.parsed
+            combined_json["plantings"]["logs"] = log_response.choices[0].message.parsed
+            combined_json["plantings"]["soil"] = soil_response.choices[0].message.parsed
+            combined_json["plantings"]["yield"] = yield_response.choices[0].message.parsed
+            
+            print(combined_json) # debugging
+
+            pretty_json = combined_json.json()
+
+            if 'error' in response:
+                raise ValueError(f"API error: {response['error']['message']}")
+            
+            return pretty_json
+    
+        except ValidationError as e:
+            return {"error": str(e)}
+        except Exception as e:
+            return {"error": "Failed to generate valid JSON. " + str(e)}
     
-def process_specifications(data, model_version, json_creation):
+def process_specifications(data, model_version, json_creation, additional_json_creation_options, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input):
     # This method just drives the process
     if json_creation == "Single JSON Creation":
         resulting_schema = generate_json(data, model_version)
     elif json_creation == "Step-wise JSON Creation":
-        pass 
+        resulting_schema = generate_json_pieces(data, model_version, additional_json_creation_options, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input) 
     return resulting_schema 
 
 with gr.Blocks() as demo: 
@@ -147,6 +229,13 @@ with gr.Blocks() as demo:
 
     additional_json_creation_options = gr.Radio(["Parse from one big input text", "Explicit specific pieces"], label="Additional Step-wise JSON Options", visible=False)
 
+    # Explicit Specific Pieces
+    field_data_input = gr.Textbox(label="Enter your data for field", placeholder="Field Name and Description", visible=False)
+    planting_data_input = gr.Textbox(label="Enter your data for plantings", placeholder="Name, Status (active/archived), Crop, Crop variety", visible=False)
+    logs_data_input = gr.Textbox(label="Enter your log data", placeholder="Convention, Date, Description", visible=False)
+    soil_data_input = gr.Textbox(label="Enter your soil data", placeholder="Description, Structure, Biology", visible=False)
+    yield_data_input = gr.Textbox(label="Enter your yield data", placeholder="Quantity, Quality", visible=False)
+    
     output_box = gr.Textbox(label="JSON Data Output")
     
     def update_visibility(radio):
@@ -155,12 +244,21 @@ with gr.Blocks() as demo:
             return gr.Radio(visible=bool(1))
         else:
             return gr.Radio(visible=bool(0))
+
+    def update_visibility2(radio):
+        value = radio  
+        if value == "Explicit specific pieces":
+            return gr.Textbox(visible=bool(1))
+        else:
+            return gr.Radio(visible=bool(0))
             
     json_creation_input.change(update_visibility, json_creation_input, additional_json_creation_options)
+    additional_json_creation_options.change(update_visibility_option2, additional_json_creation_options, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input)
+    
     submit_button = gr.Button("Generate JSON")
     submit_button.click(
         fn=process_specifications,
-        inputs=[data_input, model_version_input, json_creation_input, additional_json_creation_options],
+        inputs=[data_input, model_version_input, json_creation_input, additional_json_creation_options, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input],
         outputs=[output_box]
     )