import gradio as gr
import joblib
import numpy as np

# Carregar o modelo e o modelo de calibração serializados
model, calibration_model = joblib.load("tiebreak_calibrated_model.pkl")

# Função para realizar a previsão
def predict_tiebreak(odds1_input, odds2_input):
    # Converter vírgulas para pontos caso necessário
    odds1_input = str(odds1_input).replace(',', '.')
    odds2_input = str(odds2_input).replace(',', '.')

    # Converter para float
    odds1 = float(odds1_input)
    odds2 = float(odds2_input)
    
    # Identificar a menor e maior odd (não precisamos assumir qual é qual)
    win_odds = min(odds1, odds2)
    loser_odds = max(odds1, odds2)
    
    # Calculando as features
    odds_ratio = win_odds / loser_odds
    log_odds_w = np.log(win_odds)
    log_odds_l = np.log(loser_odds)
    prob_w = 1 / win_odds
    prob_l = 1 / loser_odds
    odds_spread = abs(loser_odds - win_odds)

    # Criando o vetor de features para o modelo
    features = np.array([[odds_ratio, log_odds_w, log_odds_l, prob_w, prob_l, odds_spread]])

    # Realizando a previsão com o modelo original
    raw_prob = model.predict_proba(features)[0, 1]  # Probabilidade da classe 1 (menos de 1.5 tiebreaks)

    # Calibrando a probabilidade com o modelo de calibração
    calibrated_prob = calibration_model.predict_proba([[raw_prob]])[0, 1]
    
    # Calculando a odds mínima
    odds_minima = 1 / calibrated_prob
    
    # Formatando a probabilidade para percentual com duas casas decimais
    prob_percent = f"{round(calibrated_prob * 100, 2)}%"
    
    # Retornando os valores
    return prob_percent, round(odds_minima, 2)

# Interface Gradio
inputs = [gr.Number(label="Odds 1"), gr.Number(label="Odds 2")]
outputs = [gr.Textbox(label="Probabilidade de menos de 1.5 Tiebreaks"), gr.Textbox(label="Odds Mínima")]

# Criação da interface
gr.Interface(fn=predict_tiebreak, inputs=inputs, outputs=outputs, title="Previsão de Tiebreaks",
             description="Insira as odds para prever a probabilidade de haver menos de 1.5 tiebreaks e calcular as odds mínimas.").launch()