diff --git a/backend/main.py b/backend/main.py index b6f46c9f7032bec3eaa2c60cd63d804e5f50ac0c..7dd9e3b3fab7b7deda34f639bd022a0f4490ff9b 100644 --- a/backend/main.py +++ b/backend/main.py @@ -134,9 +134,17 @@ def on_startup(): if environment == "prod": from sqlalchemy.engine import reflection + # inspector = reflection.Inspector.from_engine(engine) + # tables_exist = inspector.get_table_names() + from app.models import ExperimentParameters # adjust the import as needed + inspector = reflection.Inspector.from_engine(engine) tables_exist = inspector.get_table_names() + if ExperimentParameters.__tablename__ not in tables_exist: + print("Creating missing table: ExperimentParameters") + ExperimentParameters.__table__.create(bind=engine) + # Ensure the production database is initialized if not tables_exist: print("Production database is empty. Initializing...")