Update app.py

app.py

@@ -1,5 +1,3 @@
-# app.py
-
 import gradio as gr
 import pandas as pd
 import duckdb
@@ -8,10 +6,13 @@ import re
 import io
 import os
 
-# Load Together API Key from environment variable or fallback
-TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY", "tgp_v1_QxHjcs582Y4kSGPd5a2VyrKDD6S81ctp-M-rT_ioDNE")  # Replace with HF Secret
+# ✅ Read API Key
+TOGETHER_API_KEY = os.environ.get("TOGETHER_API_KEY")
+
+if not TOGETHER_API_KEY:
+    raise RuntimeError("❌ TOGETHER_API_KEY not found. Please set it in Hugging Face Secrets tab.")
 
-# Function to generate SQL from a prompt
+# 🔍 SQL generation
 def generate_sql_from_prompt(prompt, df):
     schema = ", ".join([f"{col} ({str(dtype)})" for col, dtype in df.dtypes.items()])
     full_prompt = f"""
@@ -40,7 +41,7 @@ Write a valid SQL query using the 'df' table. Return only the SQL code.
     result = response.json()
     return result['choices'][0]['message']['content'].strip("```sql").strip("```").strip()
 
-# Function to clean SQL for DuckDB
+# 🧽 SQL cleanup
 def clean_sql_for_duckdb(sql, df_columns):
     sql = sql.replace("`", '"')
     for col in df_columns:
@@ -49,7 +50,7 @@ def clean_sql_for_duckdb(sql, df_columns):
             sql = re.sub(pattern, f'"{col}"', sql)
     return sql
 
-# Combined Gradio function
+# 🔄 Main Gradio function
 def chatbot_interface(file, question):
     try:
         df = pd.read_excel(file)
@@ -60,18 +61,16 @@ def chatbot_interface(file, question):
     except Exception as e:
         return f"❌ Error: {str(e)}", pd.DataFrame()
 
-# Gradio UI
+# 🧱 Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## 📊 Excel SQL Chatbot with Together API")
     file_input = gr.File(label="📂 Upload Excel File (.xlsx)")
-    question = gr.Textbox(label="🧠 Ask a question about your data", placeholder="e.g., Show me the average sales by region")
+    question = gr.Textbox(label="🧠 Ask a question about your data")
     submit = gr.Button("🚀 Generate & Query")
-
     sql_output = gr.Markdown()
     result_table = gr.Dataframe()
-
     submit.click(fn=chatbot_interface, inputs=[file_input, question], outputs=[sql_output, result_table])
 
-# Launch app
+# ▶️ Run the app
 if __name__ == "__main__":
     demo.launch()