Update app.py

app.py

@@ -5,25 +5,22 @@ import requests
 import re
 import os
 
-# Load API Key
+# Securely load Together API Key
 def get_api_key():
     key = os.getenv("TOGETHER_API_KEY")
-    print("🔐 API KEY:", "FOUND" if key else "NOT FOUND")
     if not key:
-        raise RuntimeError("🔴 MISSING TOGETHER_API_KEY!")
+        raise RuntimeError("🔴 TOGETHER_API_KEY not found!")
     return key
 
 API_KEY = get_api_key()
 
-# Generate SQL with detailed logging
+# Generate SQL from prompt
 def generate_sql(prompt, df):
     schema = ", ".join([f"{col} ({dtype})" for col, dtype in df.dtypes.items()])
     full_prompt = f"""You are a SQL expert. Table 'df' schema: {schema}
-
 User question: "{prompt}"
 Return only valid SQL."""
-    print("📢 FULL PROMPT:\n", full_prompt)
-
+    
     url = "https://api.together.xyz/v1/completions"
     headers = {
         "Authorization": f"Bearer {API_KEY}",
@@ -32,20 +29,15 @@ Return only valid SQL."""
     payload = {
         "model": "mistralai/Mixtral-8x7B-Instruct-v0.1",
         "prompt": full_prompt,
-        "max_tokens": 200,
-        "temperature": 0.2
+        "max_tokens": 300,
+        "temperature": 0.3
     }
-    print("📡 Calling Together API:", url, payload)
+
     resp = requests.post(url, headers=headers, json=payload)
-    print("🟢 HTTP Status:", resp.status_code)
-    print("🧾 Response Body:", resp.text)
     resp.raise_for_status()
-    j = resp.json()
-    if "choices" in j:
-        return j["choices"][0]["text"].strip()
-    return j.get("output", "")
+    return resp.json()["choices"][0]["text"].strip()
 
-# Clean SQL
+# Clean SQL for DuckDB
 def clean_sql(sql, cols):
     sql = sql.replace("`", '"')
     for c in cols:
@@ -53,27 +45,27 @@ def clean_sql(sql, cols):
             sql = re.sub(rf'\b{re.escape(c)}\b', f'"{c}"', sql)
     return sql
 
-# Main function
-def chat_interface(file, q):
+# Main interface logic
+def chat_interface(file, question):
     try:
         df = pd.read_excel(file)
-        sql = generate_sql(q, df)
+        sql = generate_sql(question, df)
         sql = clean_sql(sql, df.columns)
-        print("✅ Final SQL:", sql)
-        df_res = duckdb.query(sql).to_df()
-        return f"📜 SQL:\n```sql\n{sql}\n```", df_res
+        result_df = duckdb.query(sql).to_df()
+        return f"📜 SQL:\n```sql\n{sql}\n```", result_df
     except Exception as e:
-        print("🔥 EXCEPTION:", repr(e))
         return f"❌ Error: {e}", pd.DataFrame()
 
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## SQL Chatbot – Debug Mode")
-    file_in = gr.File(label="Upload Excel")
-    q_in = gr.Textbox(label="Question")
-    bt = gr.Button("Run")
-    md = gr.Markdown()
-    tbl = gr.Dataframe()
-    bt.click(chat_interface, [file_in, q_in], [md, tbl])
+    gr.Markdown("## 📊 Excel SQL Chatbot using Together API")
+    file_input = gr.File(label="📂 Upload Excel File (.xlsx)")
+    question = gr.Textbox(label="🧠 Ask your SQL question")
+    submit = gr.Button("🚀 Generate & Run")
+    sql_output = gr.Markdown()
+    result_table = gr.Dataframe()
+
+    submit.click(fn=chat_interface, inputs=[file_input, question], outputs=[sql_output, result_table])
 
 if __name__ == "__main__":
     demo.launch()