update to allow model version to be changed

app.py

@@ -100,7 +100,7 @@ class FarmActivities(BaseModel):
 # TBD
 
 # This is to make stuff happen
-def generate_json(specification): 
+def generate_json(specification, model_version): 
     """
     Function to prompt OpenAI API to generate structured JSON output.
     """
@@ -108,7 +108,7 @@ def generate_json(specification):
     try:
         # Call OpenAI API to generate structured output based on prompt
         response = client.beta.chat.completions.parse(
-            model="gpt-4o-2024-08-06",  # Use GPT model that supports structured output
+            model=model_version,  # Use GPT model that supports structured output
             messages=[
                 {"role": "system", "content": "Extract the farm information."},
                 {"role": "user", "content": specification}
@@ -132,15 +132,16 @@ def generate_json(specification):
         return {"error": "Failed to generate valid JSON. " + str(e)}
 
     
-def process_specifications(data):
+def process_specifications(data, model_version):
     # This method just drives the process
-    resulting_schema = generate_json(data)
+    resulting_schema = generate_json(data, model_version)
     return resulting_schema 
     
     
 demo = gr.Interface(
     fn=process_specifications, 
     inputs=[gr.Textbox(label="Enter your data")],
+            gr.Radio(["gpt-4o-2024-08-06", "gpt-4o-mini-2024-07-18"], label="Model Versions", info="Mini is a smaller version for faster responses")
             outputs=[gr.Textbox(label="JSON Data Output")],
             title="JSON Schema Data Translation Unstructured to Structured Data",
             description="Input your data, receive the JSON Schema!",