Update app.py

app.py

@@ -42,7 +42,7 @@ class SALTAnalytics:
             
             # Get schema information
             schema_result = self.con.execute("DESCRIBE salt_data").fetchall()
-            self.schema_info = "\n".join([f"{col[0]}: {col[1]}" for col in schema_result])
+            self.schema_info = "\n".join([f"{col}: {col}" for col in schema_result])[1]
             
             self.data_loaded = True
             return f"✅ Successfully loaded {len(df)} records into DuckDB"
@@ -130,12 +130,12 @@ class SALTAnalytics:
                 temperature=0.1
             )
             
-            sql_query = response.choices[0].message.content.strip()
+            sql_query = response.choices.message.content.strip()
             
-            # Clean SQL query
-            if sql_query.startswith("```
+            # Clean SQL query - FIXED: Use single quotes to avoid syntax error
+            if sql_query.startswith('```sql'):
                 sql_query = sql_query[6:-3]
-            elif sql_query.startswith("```"):
+            elif sql_query.startswith('```
                 sql_query = sql_query[3:-3]
             
             # Execute query
@@ -155,9 +155,9 @@ class SALTAnalytics:
                 temperature=0.3
             )
             
-            explanation = explanation_response.choices[0].message.content
+            explanation = explanation_response.choices.message.content
             
-            return f"**SQL Query:**\n``````\n\n**Results:**\n{result_df.to_string(index=False)}\n\n**Explanation:**\n{explanation}"
+            return f"**SQL Query:**\n```sql\n{sql_query}\n```
             
         except Exception as e:
             return f"Error: {str(e)}"