Update app.py

app.py

@@ -95,18 +95,20 @@ class SALTAnalytics:
             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 = ['``````']
+        """Safely clean SQL response from OpenAI - COMPLETELY FIXED VERSION"""
+        # Define markers as variables to avoid syntax errors
+        triple_backticks = "```
+        sql_marker = "```sql"
         
-        for marker in sql_markers:
-            if sql_query.startswith(marker):
-                sql_query = sql_query[len(marker):]
-                break
+        # Remove markdown code blocks safely
+        if sql_query.startswith(sql_marker):
+            sql_query = sql_query[len(sql_marker):]
+        elif sql_query.startswith(triple_backticks):
+            sql_query = sql_query[len(triple_backticks):]
         
         # Remove trailing backticks
-        if sql_query.endswith('```
-            sql_query = sql_query[:-3]
+        if sql_query.endswith(triple_backticks):
+            sql_query = sql_query[:-len(triple_backticks)]
             
         return sql_query.strip()
     
@@ -146,7 +148,7 @@ class SALTAnalytics:
                 temperature=0.1
             )
             
-            sql_query = response.choices.message.content.strip()
+            sql_query = response.choices[0].message.content.strip()
             
             # Clean SQL query using safe method
             sql_query = self.clean_sql_response(sql_query)
@@ -168,9 +170,13 @@ class SALTAnalytics:
                 temperature=0.3
             )
             
-            explanation = explanation_response.choices.message.content
+            explanation = explanation_response.choices[0].message.content
+            
+            # Use variables for backticks in output
+            code_start = "```
+            code_end = "\n```"
             
-            return f"**SQL Query:**\n```sql\n{sql_query}\n```
+            return f"**SQL Query:**\n{code_start}{sql_query}{code_end}\n\n**Results:**\n{result_df.to_string(index=False)}\n\n**Explanation:**\n{explanation}"
             
         except Exception as e:
             return f"Error: {str(e)}"
@@ -302,7 +308,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 5.43+**: Secure interactive web interface
+        - **Gradio 5**: Secure interactive web interface
         
         **Business Value**:
         - Automate sales order completion (70-80% accuracy)