handle numerical output

app.py

@@ -25,13 +25,16 @@ def sql_engine(query: str) -> str:
         if not rows:
             return "No results found."
 
+        # If query returns a single value (e.g., AVG, SUM, COUNT), return as a string
+        if len(rows) == 1 and len(rows[0]) == 1:
+            return str(rows[0][0])  # Convert numerical result to string
+
         # Convert query results into a clean, readable format
         return "\n".join([", ".join(map(str, row)) for row in rows])
 
     except Exception as e:
         return f"Error: {str(e)}"
 
-@tool
 def query_sql(user_query: str) -> str:
     """
     Converts natural language input to an SQL query using CodeAgent
@@ -61,8 +64,8 @@ def query_sql(user_query: str) -> str:
     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."
+    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)
@@ -70,7 +73,11 @@ def query_sql(user_query: str) -> str:
     # Log the SQL query result
     print(f"SQL Query Result: {result}")
 
-    return result  # Return only the final query result, NOT the generated SQL
+    try:
+        float_result = float(result)
+        return f"{float_result:.2f}" 
+    except ValueError:
+        return result
 
 def handle_query(user_input: str) -> str:
     """