from flask import Flask
from watermark_generate.controller.secret_controller import secret
from watermark_generate.controller.dataset_controller import dataset

app = Flask(__name__)

# 注册路由
app.register_blueprint(secret)
app.register_blueprint(dataset)

# 运行
if __name__ == '__main__':
    app.run(debug=True, port=5000)