From 9962b54521642ab8839fc11a527c4f4eeb7d69a1 Mon Sep 17 00:00:00 2001
From: Cameron <leaf33golf@gmail.com>
Date: Wed, 10 Dec 2025 22:57:16 +0000
Subject: [PATCH] Update app/main.py

---
 app/main.py | 35 +++++++++++++----------------------
 1 file changed, 13 insertions(+), 22 deletions(-)

diff --git a/app/main.py b/app/main.py
index d55b8af..3cb4325 100644
--- a/app/main.py
+++ b/app/main.py
@@ -1,7 +1,7 @@
-from fastapi import FastAPI
-from app import crud, schemas
-from app.database import SessionLocal, engine
-from app import models
+from fastapi import FastAPI, Depends
+from sqlalchemy.orm import Session
+from app import crud, models, schemas
+from app.database import engine, get_db
 
 # Create tables if not exist
 models.Base.metadata.create_all(bind=engine)
@@ -12,23 +12,14 @@ app = FastAPI(title="Data Lab API")
 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("/customers", response_model=list[schemas.Customer])
+def get_customers(db: Session = Depends(get_db)):
+    return crud.get_customers(db)
 
-@app.get("/accounts")
-def get_accounts():
-    db = SessionLocal()
-    accounts = crud.get_accounts(db)
-    db.close()
-    return accounts
+@app.get("/accounts", response_model=list[schemas.Account])
+def get_accounts(db: Session = Depends(get_db)):
+    return crud.get_accounts(db)
 
-@app.get("/transactions")
-def get_accounts():
-    db = SessionLocal()
-    transactions = crud.get_transactions(db)
-    db.close()
-    return transactions
+@app.get("/transactions", response_model=list[schemas.Transaction])
+def get_transactions(db: Session = Depends(get_db)):
+    return crud.get_transactions(db)