Update database.py

database.py

@@ -1,30 +1,19 @@
-from sqlalchemy import create_engine, MetaData, inspect
-
-# Initialize database engine (Persistent SQLite)
-engine = create_engine("sqlite:///database.db")
-metadata_obj = MetaData()
-
-# Function to check existing tables
-def get_existing_tables():
+@tool
+def sql_engine(query: str) -> str:
     """
-    Returns a list of existing tables in the database.
+    Executes an SQL SELECT query and returns the results.
 
-    Returns:
-        list: List of table names.
-    """
-    inspector = inspect(engine)
-    return inspector.get_table_names()
+    Parameters:
+        query (str): The SQL query string to execute. Only SELECT queries are allowed.
 
-# Function to initialize database (without forcing a table)
-def initialize_database():
-    """
-    Ensures the database starts up without failing if no SQL file is uploaded yet.
+    Returns:
+        str: A formatted string containing the query results, or an error message if the query fails.
     """
-    tables = get_existing_tables()
-    if not tables:
-        print("No tables found. Waiting for SQL file upload...")
-    else:
-        print(f"Database initialized with tables: {tables}")
-
-initialize_database()
-
+    try:
+        with engine.connect() as con:
+            rows = con.execute(text(query)).fetchall()
+        if not rows:
+            return "No results found."
+        return "\n".join([", ".join(map(str, row)) for row in rows])
+    except Exception as e:
+        return f"Error: {str(e)}"