Update app.py

app.py

@@ -27,13 +27,13 @@ User question: "{prompt}"
 Write a valid SQL query using the 'df' table. Return only the SQL code.
 """
 
-    url = "https://api.together.xyz/inference"
+    url = "https://api.together.xyz/v1/completions"
     headers = {
         "Authorization": f"Bearer {TOGETHER_API_KEY}",
         "Content-Type": "application/json"
     }
     payload = {
-        "model": "meta-llama/Llama-3-8B-Instruct",
+        "model": "mistralai/Mixtral-8x7B-Instruct-v0.1",  # You can use "meta-llama/Llama-3-8B-Instruct" too
         "prompt": full_prompt,
         "max_tokens": 300,
         "temperature": 0.7,
@@ -42,7 +42,7 @@ Write a valid SQL query using the 'df' table. Return only the SQL code.
     response = requests.post(url, headers=headers, json=payload)
     response.raise_for_status()
     result = response.json()
-    return result['output'].strip("```sql").strip("```").strip()
+    return result['choices'][0]['text'].strip("```sql").strip("```").strip()
 
 # 🧽 Clean SQL for DuckDB compatibility
 def clean_sql_for_duckdb(sql, df_columns):