from fastapi import FastAPI
from app import crud, schemas
from app.database import SessionLocal, engine
from app import models

# Create tables if not exist
models.Base.metadata.create_all(bind=engine)

app = FastAPI(title="Data Lab API")

@app.get("/")
def root():
    return {"message": "Welcome to Data Lab API"}

@app.get("/customers")
def get_customers():
    db = SessionLocal()
    customers = crud.get_customers(db)
    db.close()
    return customers

@app.get("/accounts")
def get_accounts():
    db = SessionLocal()
    accounts = crud.get_accounts(db)
    db.close()
    return accounts

@app.get("/transactions")
def get_accounts():
    db = SessionLocal()
    transactions = crud.get_transactions(db)
    db.close()
    return transactions