52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
import os
|
|
from flask import Flask
|
|
from flask_login import LoginManager
|
|
from config import Config
|
|
from models import db, User
|
|
|
|
|
|
login_manager = LoginManager()
|
|
|
|
|
|
def create_app(config_class=Config):
|
|
app = Flask(__name__)
|
|
app.config.from_object(config_class)
|
|
|
|
# Ensure required directories exist
|
|
os.makedirs(app.config["UPLOAD_FOLDER"], exist_ok=True)
|
|
os.makedirs(app.config["VECTORDB_PATH"], exist_ok=True)
|
|
|
|
# Extensions
|
|
db.init_app(app)
|
|
login_manager.init_app(app)
|
|
login_manager.login_view = "auth.login"
|
|
login_manager.login_message_category = "info"
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
return User.query.get(int(user_id))
|
|
|
|
# Blueprints
|
|
from blueprints.auth import auth_bp
|
|
from blueprints.documents import documents_bp
|
|
from blueprints.context import context_bp
|
|
from blueprints.chat import chat_bp
|
|
from blueprints.main import main_bp
|
|
|
|
app.register_blueprint(auth_bp)
|
|
app.register_blueprint(documents_bp)
|
|
app.register_blueprint(context_bp)
|
|
app.register_blueprint(chat_bp)
|
|
app.register_blueprint(main_bp)
|
|
|
|
# Create DB tables
|
|
with app.app_context():
|
|
db.create_all()
|
|
|
|
return app
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = create_app()
|
|
app.run(host="0.0.0.0", port=5000)
|