update funcs with handling for output

app.py

@@ -10,7 +10,7 @@ from database import engine, receipts
 @tool
 def sql_engine(query: str) -> str:
     """
-    Executes an SQL query on the 'receipts' table and returns results.
+    Executes an SQL query on the 'receipts' table and returns formatted results.
     
     Table Schema:
         - receipt_id: INTEGER
@@ -21,17 +21,20 @@ def sql_engine(query: str) -> str:
         query: The SQL query to execute.
     
     Returns:
-        Query result as a string.
+        Query result as a formatted string.
     """
-    output = ""
     try:
         with engine.connect() as con:
-            rows = con.execute(text(query))
-            for row in rows:
-                output += "\n" + str(row)
+            rows = con.execute(text(query)).fetchall()
+
+        if not rows:
+            return "No results found."
+
+        # Convert results to a readable string
+        return "\n".join([", ".join(map(str, row)) for row in rows])
+
     except Exception as e:
-        output = f"Error: {str(e)}"
-    return output.strip()
+        return f"Error: {str(e)}"
 
 # Initialize CodeAgent to generate SQL queries from natural language
 agent = CodeAgent(
@@ -43,18 +46,26 @@ def query_sql(user_query: str) -> str:
     """
     Converts natural language input to an SQL query using CodeAgent
     and returns the execution results.
-    
+
     Args:
         user_query: The user's request in natural language.
-    
+
     Returns:
-        The query result from the database.
+        The query result from the database as a formatted string.
     """
     # Generate SQL from natural language
     generated_sql = agent.run(f"Convert this request into SQL: {user_query}")
 
+    # Log the generated SQL for debugging
+    print(f"Generated SQL: {generated_sql}")
+
     # Execute the SQL query and return the result
-    return sql_engine(generated_sql)
+    result = sql_engine(generated_sql)
+
+    # Log the result for debugging
+    print(f"SQL Query Result: {result}")
+
+    return result 
 
 # Define Gradio interface
 demo = gr.Interface(