From 45e99d8d4d4d57a08105744f1380ce75c2744e56 Mon Sep 17 00:00:00 2001
From: smathot <s.mathot@cogsci.nl>
Date: Fri, 10 Jan 2025 10:37:04 +0100
Subject: [PATCH] Implement flask-migrate

---
 pyproject.toml    | 1 +
 sigmund/server.py | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 89aa63b..9ecb587 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,6 +20,7 @@ dependencies = [
     "flask-sqlalchemy",
     "flask-wtf",
     "flask-cors",
+    "flask-migrate",
     "jinja2",
     "jq",
     "langchain",
diff --git a/sigmund/server.py b/sigmund/server.py
index 5dc2f14..6ef59ee 100644
--- a/sigmund/server.py
+++ b/sigmund/server.py
@@ -2,6 +2,7 @@
 os.environ['USE_FLASK_SQLALCHEMY'] = '1'
 from flask import Flask, Config, request
 from flask_login import LoginManager
+from flask_migrate import Migrate
 from . import config
 from .routes import api_blueprint, app_blueprint, User, subscribe_blueprint, \
     google_login_blueprint, public_blueprint
@@ -30,7 +31,8 @@ def create_app(config_class=SigmundConfig):
         db.create_all()
     # Initialize login manager
     login_manager = LoginManager()
-    
+    # Set up database migration
+    migrate = Migrate(app, db)
     @login_manager.user_loader
     def load_user(user_id):
         return User(user_id)