Update app.py

app.py

@@ -6,13 +6,11 @@ import re
 import io
 import os
 
-# ✅ Read API Key
+# ✅ READ API KEY from Hugging Face Secret
 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.")
+    raise RuntimeError("❌ TOGETHER_API_KEY not found. Set it in Hugging Face > Settings > Secrets.")
 
-# 🔍 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"""
@@ -23,7 +21,6 @@ User question: "{prompt}"
 
 Write a valid SQL query using the 'df' table. Return only the SQL code.
 """
-
     url = "https://api.together.xyz/v1/chat/completions"
     headers = {
         "Authorization": f"Bearer {TOGETHER_API_KEY}",
@@ -41,7 +38,6 @@ 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()
 
-# 🧽 SQL cleanup
 def clean_sql_for_duckdb(sql, df_columns):
     sql = sql.replace("`", '"')
     for col in df_columns:
@@ -50,7 +46,6 @@ def clean_sql_for_duckdb(sql, df_columns):
             sql = re.sub(pattern, f'"{col}"', sql)
     return sql
 
-# 🔄 Main Gradio function
 def chatbot_interface(file, question):
     try:
         df = pd.read_excel(file)
@@ -61,7 +56,6 @@ def chatbot_interface(file, question):
     except Exception as e:
         return f"❌ Error: {str(e)}", pd.DataFrame()
 
-# 🧱 Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("## 📊 Excel SQL Chatbot with Together API")
     file_input = gr.File(label="📂 Upload Excel File (.xlsx)")
@@ -71,6 +65,5 @@ with gr.Blocks() as demo:
     result_table = gr.Dataframe()
     submit.click(fn=chatbot_interface, inputs=[file_input, question], outputs=[sql_output, result_table])
 
-# ▶️ Run the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()