Update api_server.py

api_server.py

@@ -1,6 +1,6 @@
 # api_server.py
 
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Depends
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
@@ -148,7 +148,7 @@ SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 # 建立資料表
 Base.metadata.create_all(bind=engine)
 
-def get_db():
+def get_db:
     db = SessionLocal()
     try:
         yield db
@@ -201,7 +201,7 @@ class RecordCreate(BaseModel):
 # 6. 病患 API
 # =================================================================
 @app.post("/patients")
-def create_patient(data: PatientCreate, db: Session = Depends(get_db())):
+def create_patient(data: PatientCreate, db: Session = Depends(get_db)):
     patient = Patient(**data.dict())
     db.add(patient)
     db.commit()
@@ -210,12 +210,12 @@ def create_patient(data: PatientCreate, db: Session = Depends(get_db())):
 
 
 @app.get("/patients")
-def list_patients(db: Session = Depends(get_db())):
+def list_patients(db: Session = Depends(get_db)):
     return db.query(Patient).all()
 
 
 @app.get("/patients/{patient_id}")
-def get_patient(patient_id: int, db: Session = Depends(get_db())):
+def get_patient(patient_id: int, db: Session = Depends(get_db)):
     patient = db.query(Patient).filter(Patient.id == patient_id).first()
     if not patient:
         raise HTTPException(status_code=404, detail="找不到病患")
@@ -225,7 +225,7 @@ def get_patient(patient_id: int, db: Session = Depends(get_db())):
 # 7. 護士 API
 # =================================================================
 @app.post("/nurses")
-def create_nurse(data: NurseCreate, db: Session = Depends(get_db())):
+def create_nurse(data: NurseCreate, db: Session = Depends(get_db)):
     nurse = Nurse(**data.dict())
     db.add(nurse)
     db.commit()
@@ -234,14 +234,14 @@ def create_nurse(data: NurseCreate, db: Session = Depends(get_db())):
 
 
 @app.get("/nurses")
-def list_nurses(db: Session = Depends(get_db())):
+def list_nurses(db: Session = Depends(get_db)):
     return db.query(Nurse).all()
 
 # =================================================================
 # 8. 護理紀錄 API
 # =================================================================
 @app.post("/records")
-def create_record(data: RecordCreate, db: Session = Depends(get_db())):
+def create_record(data: RecordCreate, db: Session = Depends(get_db)):
     record = Record(**data.dict())
     db.add(record)
     db.commit()
@@ -250,5 +250,5 @@ def create_record(data: RecordCreate, db: Session = Depends(get_db())):
 
 
 @app.get("/records/{patient_id}")
-def list_records(patient_id: int, db: Session = Depends(get_db())):
+def list_records(patient_id: int, db: Session = Depends(get_db)):
     return db.query(Record).filter(Record.patient_id == patient_id).all()