import gradio as gr

# Function to evaluate slogans
def evaluate_slogans(slogans_X, slogans_Y):
    wins_X, wins_Y = 0, 0
    for (slogan_X, score_X), (slogan_Y, score_Y) in zip(slogans_X, slogans_Y):
        if score_X > score_Y:
            wins_X += 1
        else:
            wins_Y += 1
    success_rate_X = (wins_X / len(slogans_X)) * 100
    success_rate_Y = (wins_Y / len(slogans_Y)) * 100
    return f"Model X Success Rate: {success_rate_X:.2f}%", f"Model Y Success Rate: {success_rate_Y:.2f}%"

# Gradio interface
def gradio_interface(slogans_X_text, slogans_Y_text):
    # Convert input text to lists
    slogans_X = [tuple(item.split(",")) for item in slogans_X_text.strip().split("\n")]
    slogans_Y = [tuple(item.split(",")) for item in slogans_Y_text.strip().split("\n")]
    
    # Convert scores to float
    slogans_X = [(slogan, float(score)) for slogan, score in slogans_X]
    slogans_Y = [(slogan, float(score)) for slogan, score in slogans_Y]
    
    return evaluate_slogans(slogans_X, slogans_Y)

# Launch Gradio app
iface = gr.Interface(
    fn=gradio_interface,
    inputs=[
        gr.Textbox(label="Slogans from Model X (format: slogan,score per line)"),
        gr.Textbox(label="Slogans from Model Y (format: slogan,score per line)")
    ],
    outputs=[
        gr.Textbox(label="Model X Success Rate"),
        gr.Textbox(label="Model Y Success Rate")
    ],
    title="Gym Campaign Slogan Comparator",
    description="Compare the effectiveness of slogans generated by two models for a gym campaign."
)

iface.launch()