pramodmisra commited on
Commit
90b379d
·
verified ·
1 Parent(s): 1a704c9

Upload app/database.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app/database.py +2 -2
app/database.py CHANGED
@@ -1,8 +1,9 @@
 
1
  from sqlalchemy import create_engine
2
  from sqlalchemy.orm import sessionmaker
3
  from app.models import Base
4
 
5
- DATABASE_URL = "sqlite:///./data/app.db"
6
 
7
  engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
8
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@@ -13,7 +14,6 @@ def init_db():
13
 
14
 
15
  def get_db():
16
- """FastAPI dependency — yields a DB session, auto-closes after request."""
17
  db = SessionLocal()
18
  try:
19
  yield db
 
1
+ import os
2
  from sqlalchemy import create_engine
3
  from sqlalchemy.orm import sessionmaker
4
  from app.models import Base
5
 
6
+ DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./data/app.db")
7
 
8
  engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
9
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
14
 
15
 
16
  def get_db():
 
17
  db = SessionLocal()
18
  try:
19
  yield db