Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import re
 import os
 
-# ✅ Load Together API key securely
+# ✅ Securely load Together API Key
 def get_together_api_key():
     key = os.environ.get("TOGETHER_API_KEY")
     if key:
@@ -15,15 +15,17 @@ def get_together_api_key():
 
 TOGETHER_API_KEY = get_together_api_key()
 
-# 🧠 Generate SQL from prompt using Together API
+# 🧠 Generate SQL from prompt using Together's /inference endpoint
 def generate_sql_from_prompt(prompt, df):
     schema = ", ".join([f"{col} ({str(dtype)})" for col, dtype in df.dtypes.items()])
-    full_prompt = f"""You are a SQL expert. Here is a table called 'df' with the following schema:
+    full_prompt = f"""
+You are a SQL expert. Here is a table called 'df' with the following schema:
 {schema}
 
 User question: "{prompt}"
 
-Write a valid SQL query using the 'df' table. Return only the SQL code."""
+Write a valid SQL query using the 'df' table. Return only the SQL code.
+"""
 
     url = "https://api.together.xyz/inference"
     headers = {
@@ -34,7 +36,7 @@ Write a valid SQL query using the 'df' table. Return only the SQL code."""
         "model": "meta-llama/Llama-3-8B-Instruct",
         "prompt": full_prompt,
         "max_tokens": 300,
-        "temperature": 0.5,
+        "temperature": 0.7,
     }
 
     response = requests.post(url, headers=headers, json=payload)
@@ -51,7 +53,7 @@ def clean_sql_for_duckdb(sql, df_columns):
             sql = re.sub(pattern, f'"{col}"', sql)
     return sql
 
-# 💬 Main Gradio function
+# 💬 Main chatbot function
 def chatbot_interface(file, question):
     try:
         df = pd.read_excel(file)