Update backend/app/db.py

backend/app/db.py

@@ -1,16 +1,36 @@
+import os
 from sqlalchemy import create_engine
 from sqlalchemy.orm import declarative_base, sessionmaker
 
-from backend.app.config import DB_PATH
-
+DB_PATH = os.getenv("DB_PATH", "/data/clusterforge.db")
 DATABASE_URL = f"sqlite:///{DB_PATH}"
-engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
-SessionLocal = sessionmaker(bind=engine, autoflush=False, autocommit=False)
+
+engine = create_engine(
+    DATABASE_URL,
+    connect_args={"check_same_thread": False},
+)
+
+SessionLocal = sessionmaker(
+    autocommit=False,
+    autoflush=False,
+    bind=engine,
+)
+
 Base = declarative_base()
 
+
+def init_db() -> None:
+    import backend.app.models.dataset
+    import backend.app.models.experiment
+    import backend.app.models.cluster_result
+    import backend.app.models.cluster_summary
+
+    Base.metadata.create_all(bind=engine)
+
+
 def get_db():
     db = SessionLocal()
     try:
         yield db
     finally:
-        db.close()
+        db.close()