Allex21/Lora-trainer-all

1#!/usr/bin/env python32"""3Arquivo principal para deploy no Hugging Face Spaces4LoRA Trainer - Ferramenta de Treinamento de Personagens Consistentes5"""6 7import os8import sys9 10# Adicionar o diretório src ao path11sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))12 13# Importar a aplicação Flask14from main import app15 16# Configurações específicas para Hugging Face Spaces17app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024  # 500MB18app.config['UPLOAD_FOLDER'] = '/tmp/lora_uploads'19app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY', 'lora-trainer-secret-key-2024')20 21# Configurar CORS para Hugging Face22from flask_cors import CORS23try:24    CORS(app, origins=["*"])25except ImportError:26    # Se flask-cors não estiver disponível, configurar manualmente27    @app.after_request28    def after_request(response):29        response.headers.add('Access-Control-Allow-Origin', '*')30        response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')31        response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')32        return response33 34if __name__ == "__main__":35    # Porta padrão do Hugging Face Spaces36    port = int(os.environ.get("PORT", 7860))37    38    # Executar aplicação39    app.run(40        host="0.0.0.0",41        port=port,42        debug=False  # Desabilitar debug em produção43    )44 45