update for modular jsons

app.py

@@ -146,73 +146,85 @@ def generate_json(specification, model_version):
 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":
+        field_data_specification = field_data_input
+        planting_data_specification = planting_data_input
+        logs_data_specification = logs_data_input
+        soil_data_specification = soil_data_input
+        yield_data_specification = yield_data_input
+        
+    elif additional_json_creation_options == "Parse from one big input text":
+        field_data_specification = specification
+        planting_data_specification = specification
+        logs_data_specification = specification
+        soil_data_specification = specification
+        yield_data_specification = specification
     
-        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,
-            )
+    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_specification}
+            ],
+            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_specification}
+            ],
+            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_specification}
+            ],
+            response_format=Log,
+        )
 
-            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_specification}
+            ],
+            response_format=Soil,
+        )
 
-            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_specification}
+            ],
+            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
 
-            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()
 
-            pretty_json = combined_json.json()
+        if 'error' in response:
+            raise ValueError(f"API error: {response['error']['message']}")
+        
+        return pretty_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)}
+    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, 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
@@ -248,12 +260,12 @@ with gr.Blocks() as demo:
     def update_visibility2(radio):
         value = radio  
         if value == "Explicit specific pieces":
-            return [gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1))]
+            return [gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(1))]
         else:
-            return [gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0))]
+            return [gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0))]
             
     json_creation_input.change(update_visibility, json_creation_input, additional_json_creation_options)
-    additional_json_creation_options.change(fn=update_visibility2, inputs=[additional_json_creation_options], outputs=[field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input])
+    additional_json_creation_options.change(fn=update_visibility2, inputs=[additional_json_creation_options], outputs=[data_input, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input])
     
     submit_button = gr.Button("Generate JSON")
     submit_button.click(