try to fix output

app.py

@@ -43,16 +43,36 @@ def query_sql(user_query: str) -> str:
     Returns:
         The query result from the database as a formatted string.
     """
-    # Provide the AI with the correct schema
+    # Provide the AI with the correct schema and strict instructions
     schema_info = (
         "The database has a table named 'receipts' with the following schema:\n"
         "- receipt_id (INTEGER, primary key)\n"
         "- customer_name (VARCHAR(16))\n"
         "- price (FLOAT)\n"
         "- tip (FLOAT)\n"
-        "Generate an SQL query using ONLY these column names.\n"
+        "Generate a valid SQL SELECT query using ONLY these column names.\n"
+        "DO NOT explain your reasoning, and DO NOT return anything other than the SQL query itself."
     )
 
+    # Generate SQL query using the provided schema
+    generated_sql = agent.run(f"{schema_info} Convert this request into SQL: {user_query}")
+
+    # Log the generated SQL for debugging
+    print(f"Generated SQL: {generated_sql}")
+
+    # Ensure we only execute valid SELECT queries
+    if not generated_sql.strip().lower().startswith(("select", "show", "pragma")):
+        return "Error: Only SELECT queries are allowed."
+
+    # Execute the SQL query and return the result
+    result = sql_engine(generated_sql)
+
+    # Log the SQL query result
+    print(f"SQL Query Result: {result}")
+
+    return result  # Return the final result, NOT the generated SQL
+
+
     # Generate SQL query using the provided schema
     generated_sql = agent.run(f"{schema_info} Convert this request into SQL: {user_query}")