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}: {col}" for col in schema_result])[1]
+            self.schema_info = "\n".join([f"{col[0]}: {col[1]}" for col in schema_result])
             
             self.data_loaded = True
             return f"✅ Successfully loaded {len(df)} records into DuckDB"
@@ -94,6 +94,22 @@ class SALTAnalytics:
         except Exception as e:
             return f"Error generating insights: {str(e)}"
     
+    def clean_sql_response(self, sql_query: str) -> str:
+        """Safely clean SQL response from OpenAI - FIXED VERSION"""
+        # Remove markdown code blocks safely
+        sql_markers = ['``````']
+        
+        for marker in sql_markers:
+            if sql_query.startswith(marker):
+                sql_query = sql_query[len(marker):]
+                break
+        
+        # Remove trailing backticks
+        if sql_query.endswith('```
+            sql_query = sql_query[:-3]
+            
+        return sql_query.strip()
+    
     def natural_language_query(self, question: str, api_key: str):
         """Convert natural language to SQL and execute"""
         if not self.data_loaded:
@@ -132,11 +148,8 @@ class SALTAnalytics:
             
             sql_query = response.choices.message.content.strip()
             
-            # 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('```
-                sql_query = sql_query[3:-3]
+            # Clean SQL query using safe method
+            sql_query = self.clean_sql_response(sql_query)
             
             # Execute query
             result_df = self.con.execute(sql_query).fetchdf()
@@ -289,7 +302,7 @@ with gr.Blocks(title="SAP SALT Analytics Demo", theme=gr.themes.Soft()) as demo:
         - **DuckDB**: High-performance analytics database
         - **OpenAI GPT-4**: Natural language to SQL conversion
         - **Hugging Face**: Dataset hosting and deployment
-        - **Gradio**: Interactive web interface
+        - **Gradio 5.43+**: Secure interactive web interface
         
         **Business Value**:
         - Automate sales order completion (70-80% accuracy)