Update app.py

app.py

@@ -5,6 +5,9 @@ import numpy as np
 # Carregar o novo modelo serializado
 model = joblib.load("tiebreak_model_v1.pkl")
 
+# Definir o ponto de corte
+best_threshold = 0.9420000000000005
+
 # Função para realizar a previsão
 def predict_tiebreak(odds1_input, odds2_input):
     # Converter vírgulas para pontos caso necessário
@@ -33,16 +36,19 @@ def predict_tiebreak(odds1_input, odds2_input):
     # Calculando a odds mínima
     odds_minima = 1 / raw_prob
     
+    # Decisão de entrar ou não na aposta
+    enter_bet = "Sim" if raw_prob >= best_threshold else "Não"
+    
     # Formatando a probabilidade para percentual com duas casas decimais
     prob_percent = f"{round(raw_prob * 100, 2)}%"
     
     # Retornando os valores
-    return prob_percent, round(odds_minima, 2)
+    return prob_percent, round(odds_minima, 2), enter_bet
 
 # 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")]
+outputs = [gr.Textbox(label="Probabilidade de menos de 1.5 Tiebreaks"), gr.Textbox(label="Odds Mínima"), gr.Textbox(label="Entrar na Aposta?")]
 
 # 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()
+             description="Insira as odds para prever a probabilidade de haver menos de 1.5 tiebreaks, calcular as odds mínimas, e decidir se deve entrar na aposta.").launch()