import gradio as gr
from PIL import Image

# Función para calcular la actualización de Elo
def elo_update(winner_elo, loser_elo, k=32):
    expected_score_winner = 1 / (1 + 10 ** ((loser_elo - winner_elo) / 400))
    new_winner_elo = winner_elo + k * (1 - expected_score_winner)
    new_loser_elo = loser_elo + k * (0 - (1 - expected_score_winner))
    return new_winner_elo, new_loser_elo

# 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("dalle3_image.png")
    midjourney_image = Image.open("midjourney_image.png")
    flux_image = Image.open("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

    # Primera elección
    if first == "DALL-E 3":
        elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"])
        elo_scores["DALL-E 3"], elo_scores["Flux"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Flux"])
    elif first == "Midjourney 6.1":
        elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"])
        elo_scores["Midjourney 6.1"], elo_scores["Flux"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["Flux"])
    elif first == "Flux":
        elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Flux"], elo_scores["DALL-E 3"])
        elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["Flux"], elo_scores["Midjourney 6.1"])

    # Segunda elección
    if second == "DALL-E 3":
        elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"])
        elo_scores["DALL-E 3"], elo_scores["Flux"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Flux"])
    elif second == "Midjourney 6.1":
        elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"])
        elo_scores["Midjourney 6.1"], elo_scores["Flux"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["Flux"])
    elif second == "Flux":
        elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Flux"], elo_scores["DALL-E 3"])
        elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["Flux"], elo_scores["Midjourney 6.1"])

    return f"DALL-E 3: {round(elo_scores['DALL-E 3'])}, Midjourney 6.1: {round(elo_scores['Midjourney 6.1'])}, Flux: {round(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()