Update app.py

app.py

@@ -77,24 +77,21 @@ def parse_query(nl_query):
     """
     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
-
-    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:"
-            )
-        }
-    ]
+
+    # new
+    from openai import AsyncOpenAI
+
+    client = AsyncOpenAI()
+    completion = await client.chat.completions.create(model="gpt-3.5-turbo", 
+                                                      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.ChatCompletion.create(