Update app.py

app.py

@@ -13,13 +13,17 @@ initialize_database()
 @tool
 def sql_engine(query: str) -> str:
     """
-    Executes an SQL SELECT query on the database.
+    Executes an SQL SELECT query and returns the results.
 
-    Args:
-        query (str): The SQL query to be executed. Only SELECT queries are allowed.
+    Parameters:
+        query (str): The SQL query string that should be executed.
 
     Returns:
-        str: A formatted string with query results, or an error message if the query fails.
+        str: The result of the SQL query formatted as a string.
+    
+    Example:
+        >>> sql_engine("SELECT * FROM users")
+        '1, Alice, 30, 100.50\\n2, Bob, 24, 250.75\\n3, Charlie, 35, 80.00'
     """
     try:
         with engine.connect() as con:
@@ -30,11 +34,6 @@ def sql_engine(query: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Initialize CodeAgent for SQL query generation
-agent = CodeAgent(
-    tools=[sql_engine],
-    model=HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct"),
-)
 
 # Function to execute an uploaded SQL script
 def execute_sql_script(file_path):