Update database.py

database.py

@@ -1,19 +1,29 @@
-@tool
-def sql_engine(query: str) -> str:
-    """
-    Executes an SQL SELECT query and returns the results.
+from sqlalchemy import create_engine, MetaData, inspect
+
+# Initialize database engine (Persistent SQLite)
+engine = create_engine("sqlite:///database.db")
+metadata_obj = MetaData()
 
-    Parameters:
-        query (str): The SQL query string to execute. Only SELECT queries are allowed.
+# Function to check existing tables
+def get_existing_tables():
+    """
+    Returns a list of existing tables in the database.
 
     Returns:
-        str: A formatted string containing the query results, or an error message if the query fails.
+        list: List of table names.
     """
-    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)}"
+    inspector = inspect(engine)
+    return inspector.get_table_names()
+
+# 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.
+    """
+    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()