Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,8 +5,11 @@ import requests
 import re
 import io
 
-# 🔐 Set your Together API key securely
-TOGETHER_API_KEY = st.secrets["TOGETHER_API_KEY"] if "TOGETHER_API_KEY" in st.secrets else st.text_input("Enter Together API Key", type="password")
+# 🔐 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:
+    TOGETHER_API_KEY = st.text_input("🔐 Enter Together API Key", type="password")
 
 # 🧠 Generate SQL using Together API
 def generate_sql_from_prompt(prompt, df):
@@ -52,7 +55,7 @@ st.markdown("Upload your **Excel file**, ask a question in natural language, and
 
 uploaded_file = st.file_uploader("📂 Upload Excel file", type=["xlsx"])
 
-if uploaded_file and TOGETHER_API_KEY:
+if TOGETHER_API_KEY and uploaded_file:
     df = pd.read_excel(uploaded_file)
     st.success(f"✅ Loaded: {uploaded_file.name} with shape {df.shape}")
     st.dataframe(df.head(), use_container_width=True)
@@ -75,5 +78,6 @@ if uploaded_file and TOGETHER_API_KEY:
 
         except Exception as e:
             st.error(f"❌ Error: {e}")
-else:
-    st.info("🔑 Please upload a file and provide the API key to continue.")
+
+elif not TOGETHER_API_KEY:
+    st.warning("🔑 Please enter your Together API key to continue.")