Update database.py

database.py

@@ -1,13 +1,13 @@
-from sqlalchemy import create_engine, MetaData, inspect
+from sqlalchemy import create_engine, MetaData, Table, Column, String, Integer, Float, inspect
 
-# Initialize the SQLite database
+# Initialize database engine (Persistent SQLite)
 engine = create_engine("sqlite:///database.db")
 metadata_obj = MetaData()
 
 # Function to check existing tables
 def get_existing_tables():
     """
-    Returns a list of tables currently in the database.
+    Returns a list of existing tables in the database.
 
     Returns:
         list: List of table names.
@@ -15,15 +15,39 @@ def get_existing_tables():
     inspector = inspect(engine)
     return inspector.get_table_names()
 
-# Function to safely start the database
+# Create a default table if no SQL file is uploaded
+def create_placeholder_table():
+    """
+    Creates a default 'users' table with example data if no SQL file has been uploaded.
+    """
+    if "users" not in get_existing_tables():
+        users_table = Table(
+            "users", metadata_obj,
+            Column("id", Integer, primary_key=True),
+            Column("name", String(50)),
+            Column("age", Integer),
+            Column("balance", Float)
+        )
+        metadata_obj.create_all(engine)
+        
+        with engine.connect() as con:
+            con.execute(users_table.insert(), [
+                {"id": 1, "name": "Alice", "age": 30, "balance": 100.50},
+                {"id": 2, "name": "Bob", "age": 24, "balance": 250.75},
+                {"id": 3, "name": "Charlie", "age": 35, "balance": 80.00}
+            ])
+        print("✅ Created placeholder 'users' table with sample data.")
+
+# Function to initialize database (Prevents crashes)
 def initialize_database():
     """
-    Starts the database without failing if no SQL file is uploaded yet.
+    Ensures the database starts up with a placeholder table if no SQL file is uploaded.
     """
     tables = get_existing_tables()
     
     if not tables:
-        print("No tables found. The database is waiting for an SQL file upload.")
+        print("No tables found. Creating a placeholder table...")
+        create_placeholder_table()
     else:
         print(f"Database initialized with tables: {tables}")