Allex21/Lora-trainer-all

1import os2import sys3# DON'T CHANGE THIS !!!4sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))5 6from flask import Flask, send_from_directory7from src.models.user import db8from src.routes.user import user_bp9from src.routes.lora_training import lora_bp10 11app = Flask(__name__, static_folder=os.path.join(os.path.dirname(__file__), 'static'))12app.config['SECRET_KEY'] = 'asdf#FGSgvasgf$5$WGT'13app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024  # 500MB max file size14 15app.register_blueprint(user_bp, url_prefix='/api')16app.register_blueprint(lora_bp, url_prefix='/api')17 18# uncomment if you need to use database19app.config['SQLALCHEMY_DATABASE_URI'] = f"sqlite:///{os.path.join(os.path.dirname(__file__), 'database', 'app.db')}"20app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False21db.init_app(app)22with app.app_context():23    db.create_all()24 25@app.route('/', defaults={'path': ''})26@app.route('/<path:path>')27def serve(path):28    static_folder_path = app.static_folder29    if static_folder_path is None:30            return "Static folder not configured", 40431 32    if path != "" and os.path.exists(os.path.join(static_folder_path, path)):33        return send_from_directory(static_folder_path, path)34    else:35        index_path = os.path.join(static_folder_path, 'index.html')36        if os.path.exists(index_path):37            return send_from_directory(static_folder_path, 'index.html')38        else:39            return "index.html not found", 40440 41 42if __name__ == '__main__':43    app.run(host='0.0.0.0', port=5001, debug=True)44