app.py

#!/usr/bin/env python3
"""
Arquivo principal para deploy no Hugging Face Spaces
LoRA Trainer - Ferramenta de Treinamento de Personagens Consistentes
"""

import os
import sys

# Adicionar o diretório src ao path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

# Importar a aplicação Flask
from main import app

# Configurações específicas para Hugging Face Spaces
app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024  # 500MB
app.config['UPLOAD_FOLDER'] = '/tmp/lora_uploads'
app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY', 'lora-trainer-secret-key-2024')

# Configurar CORS para Hugging Face
from flask_cors import CORS
try:
    CORS(app, origins=["*"])
except ImportError:
    # Se flask-cors não estiver disponível, configurar manualmente
    @app.after_request
    def after_request(response):
        response.headers.add('Access-Control-Allow-Origin', '*')
        response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
        response.headers.add('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE,OPTIONS')
        return response

if __name__ == "__main__":
    # Porta padrão do Hugging Face Spaces
    port = int(os.environ.get("PORT", 7860))
    
    # Executar aplicação
    app.run(
        host="0.0.0.0",
        port=port,
        debug=False  # Desabilitar debug em produção
    )