from flask import Flask
# from ultralytics import YOLO

def create_app():
    app = Flask(__name__)
    from .api import ocr_bp
    app.register_blueprint(ocr_bp)

    # with app.app_context():
    #     # Load model once
    #     app.models = {
    #         'adhaarModel': YOLO('models/aadhaarYolov8.pt'),
    #         'panModel': YOLO('models/PanModel_v6.pt')  # Load additional models as needed
    #     }

    return app