Update app.py

app.py

@@ -4,12 +4,6 @@ import gradio as gr
 def obter_ultimo_sorteio(input_numeros):
     """
     Recebe os números do último sorteio fornecidos pelo usuário.
-
-    Args:
-        input_numeros (str): String com 6 números separados por vírgula.
-
-    Returns:
-        list: Lista de números inteiros.
     """
     try:
         numeros = [int(x.strip()) for x in input_numeros.split(',') if x.strip().isdigit()]
@@ -20,22 +14,23 @@ def obter_ultimo_sorteio(input_numeros):
     except Exception as e:
         return str(e)
 
-def gerar_palpites(ultimo_sorteio, quantidade_numeros, quantidade_palpites, logica):
+def gerar_palpites(ultimo_sorteio, quantidade_numeros, quantidade_palpites, logica=None):
     """
     Gera palpites para a Mega-Sena com base na lógica escolhida.
-
-    Args:
-        ultimo_sorteio (list): Números do último sorteio.
-        quantidade_numeros (int): Quantidade de números por palpite (6 a 9).
-        quantidade_palpites (int): Quantidade de palpites a serem gerados (1 a 10).
-        logica (int): Lógica a ser usada (1, 2 ou 3).
-
-    Returns:
-        list: Lista de palpites gerados.
+    Se a lógica não for informada, a lógica automática será aplicada.
     """
     palpites = []
     todos_os_numeros = list(range(1, 61))
 
+    # Lógica automática: Escolher lógica com base em quantidade de palpites e números por palpite
+    if logica is None:
+        if quantidade_palpites > 5:
+            logica = 3  # Lógica 3: Distribuição alternada se houver muitos palpites
+        elif quantidade_numeros > 6:
+            logica = 2  # Lógica 2: Combina frequentes e atrasados se a quantidade de números for maior
+        else:
+            logica = 1  # Lógica 1: Priorizar números frequentes
+
     for _ in range(quantidade_palpites):
         if logica == 1:  # Lógica 1: Priorizar números frequentes
             mais_frequentes = [n for n in ultimo_sorteio[:3]]
@@ -61,51 +56,30 @@ def gerar_palpites(ultimo_sorteio, quantidade_numeros, quantidade_palpites, logi
 
     return palpites
 
-def interface(ultimo_sorteio, quantidade_numeros, quantidade_palpites, logica):
+def interface(ultimo_sorteio, quantidade_numeros, quantidade_palpites):
     """
-    Função que integra as partes do programa e retorna os palpites gerados.
-
-    Args:
-        ultimo_sorteio (str): Números do último sorteio.
-        quantidade_numeros (int): Quantidade de números por palpite.
-        quantidade_palpites (int): Quantidade de palpites.
-        logica (str): Lógica escolhida como string.
-
-    Returns:
-        list: Palpites gerados ou mensagem de erro.
+    Função que integra as partes do programa e retorna os palpites gerados com lógica automática.
     """
-    try:
-        logica = int(logica)  # Converte o valor de string para inteiro
-    except ValueError:
-        return "Lógica inválida. Escolha entre 1, 2 ou 3."
-
     numeros = obter_ultimo_sorteio(ultimo_sorteio)
     if isinstance(numeros, str):
         return numeros  # Mensagem de erro
-    palpites = gerar_palpites(numeros, quantidade_numeros, quantidade_palpites, logica)
+    palpites = gerar_palpites(numeros, quantidade_numeros, quantidade_palpites)
     return palpites
 
-descricao_logicas = {
-    "1": "Lógica 1: Prioriza os números mais frequentes nos sorteios anteriores.",
-    "2": "Lógica 2: Combina números frequentes com os que não saíram recentemente.",
-    "3": "Lógica 3: Distribui os números alternadamente entre faixas." 
-}
-
 with gr.Blocks() as app:
     gr.Markdown("""
     # Gerador de Palpites para a Mega-Sena
-    Escolha a lógica desejada, insira os números do último sorteio e defina as preferências para gerar seus palpites.
+    Escolha a quantidade de números e palpites e insira os números do último sorteio. A lógica será escolhida automaticamente.
     """)
 
     with gr.Row():
         ultimo_sorteio = gr.Textbox(label="Último Sorteio (6 números separados por vírgula)")
         quantidade_numeros = gr.Slider(6, 9, step=1, label="Quantidade de Números por Palpite")
         quantidade_palpites = gr.Slider(1, 10, step=1, label="Quantidade de Palpites")
-        logica = gr.Dropdown(["1", "2", "3"], label="Escolha a Lógica", info=descricao_logicas)
 
     palpites_output = gr.Textbox(label="Palpites Gerados", interactive=False)
 
     gerar_palpites_btn = gr.Button("Gerar Palpites")
-    gerar_palpites_btn.click(interface, inputs=[ultimo_sorteio, quantidade_numeros, quantidade_palpites, logica], outputs=[palpites_output])
+    gerar_palpites_btn.click(interface, inputs=[ultimo_sorteio, quantidade_numeros, quantidade_palpites], outputs=[palpites_output])
 
 app.launch(share=True)