Update app.py

app.py

@@ -75,34 +75,35 @@ def load_dataset_schema():
 
 def parse_query(nl_query):
     """
-    Converts a natural language query into a SQL query using OpenAI's GPT-3 Completion API.
+    Converts a natural language query into a SQL query using OpenAI's GPT-4-turbo model.
     """
     openai.api_key = os.getenv('OPENAI_API_KEY')  # Ensure your API key is set as an environment variable
 
-    prompt = f"""
-    You are an assistant that converts natural language queries into SQL queries for a DuckDB database named 'contract_data'. Use the provided schema to form accurate SQL queries.
-
-    Schema:
-    {json.dumps(schema, indent=2)}
-
-    Natural Language Query:
-    "{nl_query}"
-
-    SQL Query:
-    """
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "You are an assistant that converts natural language queries into SQL queries "
+                "for a DuckDB database named 'contract_data'. Use the provided schema to form accurate SQL queries."
+            )
+        },
+        {
+            "role": "user",
+            "content": (
+                f"Schema:\n{json.dumps(schema, indent=2)}\n\n"
+                f"Natural Language Query:\n\"{nl_query}\"\n\nSQL Query:"
+            )
+        }
+    ]
 
     try:
-        response = openai.Completion.create(
-            model="gpt-3.5-turbo-instruct",  # Use the appropriate model for completions
-            prompt=prompt,
+        response = openai.ChatCompletion.create(
+            model="gpt-4-turbo",
+            messages=messages,
             temperature=0,  # Set to 0 for deterministic output
             max_tokens=150,
-            top_p=1,
-            frequency_penalty=0,
-            presence_penalty=0,
-            stop=["\n\n"]  # Stop generation after two newlines
         )
-        sql_query = response.choices[0].text.strip()
+        sql_query = response.choices[0].message['content'].strip()
         return sql_query
     except Exception as e:
         return f"Error generating SQL query: {e}"