handle numerical output

app.py

@@ -60,25 +60,25 @@ def query_sql(user_query: str) -> str:
     # Generate SQL query using the provided schema
     generated_sql = agent.run(f"{schema_info} Convert this request into SQL: {user_query}")
 
+    # Ensure generated_sql is always a string
+    if not isinstance(generated_sql, str):
+        return f"Unexpected result: {generated_sql}"  # Handle unexpected numerical result
+
     # 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}")
-
-    # Ensure proper formatting based on content
+    
     try:
         float_result = float(result)
-        return f"{float_result:.2f}"  # Format numbers to 2 decimal places
+        return f"{float_result:.2f}"
     except ValueError:
-        return result  # Return text results directly without modification
+        return result 
 
 def handle_query(user_input: str) -> str:
     """