instantiate db

app.py

@@ -1,52 +1,16 @@
 import os
 from dotenv import load_dotenv
 import gradio as gr
-from sqlalchemy import (
-    create_engine,
-    MetaData,
-    Table,
-    Column,
-    String,
-    Integer,
-    Float,
-    insert,
-    text,
-)
+from sqlalchemy import text
 from smolagents import tool, CodeAgent, HfApiModel
 
+# Import the database setup from database.py
+from database import engine, receipts
+
 # Load Hugging Face token from environment variables
 load_dotenv(override=True)
 hf_token = os.getenv("HF_TOKEN")
 
-# Initialize in-memory SQLite database
-engine = create_engine("sqlite:///:memory:")
-metadata_obj = MetaData()
-
-# Create 'receipts' table
-receipts = Table(
-    "receipts",
-    metadata_obj,
-    Column("receipt_id", Integer, primary_key=True),
-    Column("customer_name", String(16), primary_key=True),
-    Column("price", Float),
-    Column("tip", Float),
-)
-metadata_obj.create_all(engine)
-
-# Function to insert data
-def insert_rows_into_table(rows, table):
-    with engine.begin() as connection:
-        connection.execute(insert(table), rows)
-
-# Insert sample data
-rows = [
-    {"receipt_id": 1, "customer_name": "Alan Payne", "price": 12.06, "tip": 1.20},
-    {"receipt_id": 2, "customer_name": "Alex Mason", "price": 23.86, "tip": 0.24},
-    {"receipt_id": 3, "customer_name": "Woodrow Wilson", "price": 53.43, "tip": 5.43},
-    {"receipt_id": 4, "customer_name": "Margaret James", "price": 21.11, "tip": 1.00},
-]
-insert_rows_into_table(rows, receipts)
-
 # SQL Execution function
 @tool
 def sql_engine(query: str) -> str: