Update app.py

app.py

@@ -89,23 +89,25 @@ if submitted:
             client = OpenAI(api_key=api_key)
             
             # System prompt with schema
-            system_context = """You are a SQL expert. Given these SQL tables:
-                CREATE TABLE Produkte (
-                    ProduktID INTEGER PRIMARY KEY,
-                    Produktname TEXT NOT NULL,
-                    Preis REAL NOT NULL
-                );
-                CREATE TABLE Bestellungen (
-                    BestellungID INTEGER PRIMARY KEY,
-                    ProduktID INTEGER,
-                    Menge INTEGER,
-                    Bestelldatum TEXT,
-                    Person TEXT,
-                    FOREIGN KEY (ProduktID) REFERENCES Produkte(ProduktID)
-                );
-                Generate ONLY the SQL query for the user's request. Output raw SQL without any formatting or explanations."""
-            
-            # Generate SQL query
+           system_context = """You are a SQL expert. Given these SQL tables:
+            CREATE TABLE Produkte (
+                ProduktID INTEGER PRIMARY KEY,
+                Produktname TEXT NOT NULL,
+                Preis REAL NOT NULL
+            );
+            CREATE TABLE Bestellungen (
+                BestellungID INTEGER PRIMARY KEY,
+                ProduktID INTEGER,
+                Menge INTEGER,
+                Bestelldatum TEXT,
+                Person TEXT,
+                FOREIGN KEY (ProduktID) REFERENCES Produkte(ProduktID)
+            );
+            Generate ONLY the raw SQL query for the user's request. 
+            NEVER use markdown code blocks or any formatting. 
+            ONLY output the pure SQL statement."""
+        
+           # Generate SQL query
             response = client.chat.completions.create(
                 model="gpt-4o",
                 messages=[
@@ -115,8 +117,10 @@ if submitted:
                 temperature=0.3
             )
             
+            # Clean up the response
             sql_query = response.choices[0].message.content.strip()
-
+            sql_query = sql_query.replace("```sql", "").replace("```", "").strip()  # Add this line
+            
             # Execute and display results
             with sqlite3.connect('database.db') as conn:
                 cursor = conn.cursor()