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)
    os.makedirs(app.config["TRANSFORMERS_CACHE"], exist_ok=True)

    # Set HuggingFace cache env so sentence-transformers respects it
    os.environ["TRANSFORMERS_CACHE"] = app.config["TRANSFORMERS_CACHE"]
    os.environ["HF_HOME"] = app.config["TRANSFORMERS_CACHE"]

    # 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)