import gradio as gr
from PIL import Image
from elo import Elo

# Inicializa el sistema de puntuación Elo
elo_system = Elo(k=32)

# Inicializa puntuaciones Elo
elo_scores = {
    "DALL-E 3": 1000,
    "Midjourney 6.1": 1000,
    "Flux": 1000,
}

# Cargar imágenes predefinidas
def load_images():
    # Asegúrate de que los archivos de imágenes coincidan con estos nombres
    dalle3_image = Image.open("images/dalle3_image.png")
    midjourney_image = Image.open("images/midjourney_image.png")
    flux_image = Image.open("images/flux_image.png")
    return dalle3_image, midjourney_image, flux_image

# Actualizar el puntaje Elo basado en la votación
def update_elo(first, second, third):
    global elo_scores

    if first == "DALL-E 3":
        elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"] = elo_system.rate_match(elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"])
        elo_scores["DALL-E 3"], elo_scores["Flux"] = elo_system.rate_match(elo_scores["DALL-E 3"], elo_scores["Flux"])
    elif first == "Midjourney 6.1":
        elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"] = elo_system.rate_match(elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"])
        elo_scores["Midjourney 6.1"], elo_scores["Flux"] = elo_system.rate_match(elo_scores["Midjourney 6.1"], elo_scores["Flux"])
    elif first == "Flux":
        elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_system.rate_match(elo_scores["Flux"], elo_scores["DALL-E 3"])
        elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_system.rate_match(elo_scores["Flux"], elo_scores["Midjourney 6.1"])

    return f"DALL-E 3: {elo_scores['DALL-E 3']}, Midjourney 6.1: {elo_scores['Midjourney 6.1']}, Flux: {elo_scores['Flux']}"

# Interfaz Gradio
def vote(first, second, third):
    updated_scores = update_elo(first, second, third)
    dalle3_image, midjourney_image, flux_image = load_images()
    return dalle3_image, midjourney_image, flux_image, updated_scores

with gr.Blocks() as demo:
    gr.Markdown("# Akelarre Generativo (DALL-E 3 vs MJ 6.1 vs Flux 1)")
    gr.Markdown("### Vota por la imagen generada por IA que prefieras. La puntuación Elo se actualizará en tiempo real.")

    dalle3_image = gr.Image(label="DALL-E 3", interactive=False)
    midjourney_image = gr.Image(label="Midjourney 6.1", interactive=False)
    flux_image = gr.Image(label="Flux", interactive=False)

    with gr.Row():
        first_choice = gr.Dropdown(["DALL-E 3", "Midjourney 6.1", "Flux"], label="Primera Favorita")
        second_choice = gr.Dropdown(["DALL-E 3", "Midjourney 6.1", "Flux"], label="Segunda Favorita")
        third_choice = gr.Dropdown(["DALL-E 3", "Midjourney 6.1", "Flux"], label="Tercera Favorita")
    
    elo_scores_text = gr.Textbox(label="Puntuación Elo")

    vote_button = gr.Button("Votar y Actualizar")

    vote_button.click(vote, inputs=[first_choice, second_choice, third_choice], outputs=[dalle3_image, midjourney_image, flux_image, elo_scores_text])

demo.launch()