Update streamlit_app.py

streamlit_app.py

@@ -5,7 +5,7 @@ from pandasql import sqldf
 
 # 🛠 Page setup
 st.set_page_config(page_title="🧠 Excel SQL Assistant", layout="centered")
-st.title("SQL CHATBOT - POC")
+st.title("📊 Excel to SQL with Together AI")
 
 # 🔐 Load API Key securely
 TOGETHER_API_KEY = st.secrets.get("TOGETHER_API_KEY", None)
@@ -52,6 +52,7 @@ IMPORTANT:
 - Use SQL compatible with SQLite.
 - Table name is 'df'.
 - If column names contain spaces or special characters, wrap them in double quotes.
+- DO NOT return unmatched quotes or symbols like *, ', ", ; — unless used correctly in SQL syntax.
 """
 
             # 🔗 Together API request
@@ -82,6 +83,14 @@ IMPORTANT:
                 # 🧼 Clean the response
                 sql_query = response.json()['choices'][0]['message']['content'].strip()
                 sql_query = sql_query.replace("```sql", "").replace("```", "").strip()
+
+                # Auto-correct common malformed syntax
+                sql_query = sql_query.replace("*\"", "*").replace("\" FROM", " FROM").replace(";*", ";")
+                if sql_query.startswith('"SELECT'):
+                    sql_query = sql_query.replace('"SELECT', 'SELECT')
+                if sql_query.startswith('SELECT "'):
+                    sql_query = sql_query.replace('SELECT "', 'SELECT ')
+
                 st.code(sql_query, language='sql')
 
                 # 🧪 Execute SQL