Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,14 +4,15 @@ import duckdb
 import requests
 import re
 import io
+import os
 
-# 🔐 Handle Together API Key (from secrets or user input)
-if "TOGETHER_API_KEY" in st.secrets:
-    TOGETHER_API_KEY = st.secrets["TOGETHER_API_KEY"]
-else:
+# === 🔐 Safe API key input (streamlit secrets optional) ===
+TOGETHER_API_KEY = os.environ.get("TOGETHER_API_KEY", "")  # Preferred for Hugging Face or env deployment
+
+if not TOGETHER_API_KEY:
     TOGETHER_API_KEY = st.text_input("🔐 Enter Together API Key", type="password")
 
-# 🧠 Generate SQL using Together API
+# === SQL Generator Function ===
 def generate_sql_from_prompt(prompt, df):
     schema = ", ".join([f"{col} ({str(dtype)})" for col, dtype in df.dtypes.items()])
     full_prompt = f"""
@@ -39,7 +40,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()
 
-# 🧽 Clean SQL for DuckDB
+# === SQL Cleaner ===
 def clean_sql_for_duckdb(sql, df_columns):
     sql = sql.replace("`", '"')
     for col in df_columns:
@@ -48,10 +49,10 @@ def clean_sql_for_duckdb(sql, df_columns):
             sql = re.sub(pattern, f'"{col}"', sql)
     return sql
 
-# === Streamlit UI ===
+# === UI ===
 st.set_page_config(page_title="🧠 Excel SQL Chatbot", layout="centered")
 st.title("📊 Excel SQL Chatbot with LLM")
-st.markdown("Upload your **Excel file**, ask a question in natural language, and get results from SQL queries generated by an LLM.")
+st.markdown("Upload your **Excel file**, ask a question in natural language, and get results via generated SQL.")
 
 uploaded_file = st.file_uploader("📂 Upload Excel file", type=["xlsx"])