Update app.py

app.py

@@ -99,13 +99,11 @@ def query_optimization_agent(state: SQLExecutionState) -> SQLExecutionState:
         temperature=0
     )
     optimized_query = response["choices"][0]["message"]["content"].strip()
-    if optimized_query.startswith("
-sql"):
-        optimized_query = optimized_query.replace("
-sql", "").replace("
-", "").strip()
+    if optimized_query.startswith("```sql"):
+        optimized_query = optimized_query.replace("```sql", "").replace("```", "").strip()
     return {"optimized_sql": optimized_query}
 
+
 graph.add_node("Query Optimization", query_optimization_agent)
 
 # ------------- 4. SQL Execution Agent -------------