Update streamlit_app.py

streamlit_app.py

@@ -17,6 +17,18 @@ if not TOGETHER_API_KEY:
 # 📂 File upload
 uploaded_file = st.file_uploader("Upload your Excel file", type=["xlsx"])
 
+if uploaded_file is not None:
+    try:
+        df = pd.read_excel(uploaded_file)
+        st.success("✅ File loaded successfully")
+        st.subheader("📄 Preview of Uploaded Data")
+        st.dataframe(df.head(10))
+    except Exception as e:
+        st.error(f"❌ Error reading Excel file: {e}")
+        st.stop()
+else:
+    st.info("📂 Please upload an Excel file to begin.")
+
 # 💬 User input
 user_query = st.text_input("💬 Ask a question about your dataset")
 
@@ -24,10 +36,6 @@ user_query = st.text_input("💬 Ask a question about your dataset")
 if st.button("Generate & Run SQL"):
     if uploaded_file is not None and user_query.strip():
         try:
-            df = pd.read_excel(uploaded_file)
-            st.success("✅ File loaded successfully")
-            st.dataframe(df.head(3))
-
             # 🧠 Prompt construction
             preview = df.head(5).to_string(index=False)
             prompt = f"""
@@ -43,6 +51,7 @@ IMPORTANT:
 - Only return a valid SQL query (no Python, no comments, no markdown, no explanations).
 - Use SQL compatible with SQLite.
 - Table name is 'df'.
+- If column names contain spaces or special characters, wrap them in double quotes.
 """
 
             # 🔗 Together API request