Update app.py

app.py

@@ -7,8 +7,6 @@ import io
 import itertools
 from collections import defaultdict
 import gradio as gr
-import time  # Adicionado para compatibilidade com o código de imóveis
-import math  # Adicionado para compatibilidade com o código de imóveis
 
 FILE_ID = "1EoH8V3Fyls0mNa95VcBN6ZL1B552ty3X"
 URL = f"https://drive.google.com/uc?export=download&id={FILE_ID}"
@@ -85,77 +83,7 @@ def calcular_tempo_parcelas(combinacao):
         print(f"Erro tempo: {e}")
         return ""
 
-def buscar_combinacoes_amplo(credito, entrada, parcela, adm_selecionadas, max_combinacoes=15):
-    """Busca combinações com tolerâncias mais amplas para valores altos."""
-    global grupos_auto
-    if not planilha_carregada:
-        if not carregar_planilha():
-            return "❌ Erro: planilha não carregada."
-
-    if adm_selecionadas:
-        grupos = {adm: cotas for adm, cotas in grupos_auto.items() if adm in adm_selecionadas}
-    else:
-        grupos = grupos_auto
-    if not grupos:
-        return "🚫 Nenhuma cota encontrada com os filtros."
-
-    # Tolerâncias mais amplas para valores altos
-    if credito > 5000000:
-        tol_cred, tol_ent, tol_par = 0.50, 0.50, 1.00
-        max_cotas = 25
-    elif credito > 100000:
-        tol_cred, tol_ent, tol_par = 0.25, 0.25, 0.50
-        max_cotas = 15
-    else:
-        tol_cred, tol_ent, tol_par = 0.15, 0.15, 0.30
-        max_cotas = 15
-
-    cred_min = credito * (1 - tol_cred)
-    cred_max = credito * (1 + tol_cred)
-    ent_min = entrada * (1 - tol_ent)
-    ent_max = entrada * (1 + tol_ent)
-    parc_min = parcela * (1 - tol_par)
-    parc_max = parcela * (1 + tol_par)
-
-    resultados = []
-    total_combinacoes_testadas = 0
-
-    for adm, cotas in grupos.items():
-        print(f"🔍 BUSCA AMPLA - Analisando administradora: {adm} ({len(cotas)} cotas)")
-        for r in range(1, min(max_cotas + 1, len(cotas) + 1)):
-            max_combinacoes_por_adm = 7000 if credito > 5000000 else 4000
-            combinacoes_geradas = 0
-            for comb in itertools.combinations(cotas, r):
-                if combinacoes_geradas >= max_combinacoes_por_adm:
-                    break
-                total_combinacoes_testadas += 1
-                combinacoes_geradas += 1
-                total_cred = sum(c['valor_credito'] for c in comb)
-                total_ent_base = sum(c['valor_entrada'] for c in comb)
-                total_ent = total_ent_base + total_cred * 0.085
-                total_parc = sum(c['valor_parcela'] for c in comb)
-                if (cred_min <= total_cred <= cred_max and
-                    ent_min <= total_ent <= ent_max and
-                    parc_min <= total_parc <= parc_max):
-                    total_saldo = sum(c['saldo_devedor'] for c in comb)
-                    valor_final = total_ent + total_saldo
-                    diff = abs(valor_final - total_cred)
-                    resultados.append((adm, comb, diff))
-                    if len(resultados) >= max_combinacoes * 2:
-                        break
-            if len(resultados) >= max_combinacoes * 2:
-                break
-        if len(resultados) >= max_combinacoes * 2:
-            break
-
-    print(f"🔍 BUSCA AMPLA: Combinações válidas encontradas: {len(resultados)}")
-    if not resultados:
-        return "🚫 Nenhuma combinação adequada encontrada com os parâmetros fornecidos."
-
-    resultados.sort(key=lambda x: x[2])
-    return formatar_resultados(resultados[:max_combinacoes])
-
-def buscar_combinacoes(credito, entrada, parcela, adm_selecionadas, max_combinacoes=15):
+def buscar_combinacoes(credito, entrada, parcela, adm_selecionadas):
     if not planilha_carregada:
         if not carregar_planilha():
             return "❌ Erro: planilha não carregada."
@@ -206,11 +134,9 @@ def buscar_combinacoes(credito, entrada, parcela, adm_selecionadas, max_combinac
             break
 
     if not resultados:
-        print("🔍 Nenhuma combinação encontrada. Tentando busca ampla...")
-        return buscar_combinacoes_amplo(credito, entrada, parcela, adm_selecionadas, max_combinacoes)
-
+        return "🚫 Nenhuma combinação encontrada."
     resultados.sort(key=lambda x: x[2])
-    return formatar_resultados(resultados[:max_combinacoes])
+    return formatar_resultados(resultados[:15])
 
 def formatar_resultados(resultados):
     if not resultados:
@@ -252,98 +178,24 @@ def formatar_resultados(resultados):
 # Carrega a planilha antes de iniciar
 carregar_planilha()
 
-def get_administradoras_options():
-    """Retorna a lista de administradoras disponíveis para o dropdown."""
-    return ADMIN_OPCOES
-
-def ajustar_e_buscar(credito_desejado, entrada_desejada, parcela_desejada, administradoras):
-    """Função wrapper para chamar a busca com os parâmetros ajustados."""
-    return buscar_combinacoes(credito_desejado, entrada_desejada, parcela_desejada, administradoras, max_combinacoes=15)
-
-# Interface usando gr.Blocks para garantir carregamento correto (inspirado no código de imóveis)
-with gr.Blocks(title="Sistema de Busca de Consórcio Auto") as demo:
-    gr.HTML("""
-    <link href="https://fonts.googleapis.com/css2?family=Poppins&display=swap" rel="stylesheet">
-    <style>
-        * {
-            font-family: 'Poppins', sans-serif !important;
-        }
-        body {
-            background-color: #ff8f00 !important;
-            color: black !important;
-        }
-        input[type="number"] {
-            background-color: white !important;
-            border: 2px solid #000 !important;
-            border-radius: 12px !important;
-            color: black !important;
-            padding: 10px !important;
-            font-weight: 500;
-        }
-        input[type="text"], textarea {
-            background-color: white !important;
-            border: 2px solid black !important;
-            border-radius: 12px !important;
-            color: black !important;
-            padding: 10px !important;
-        }
-        input::placeholder, textarea::placeholder {
-            color: #333 !important;
-            opacity: 0.7;
-        }
-        .gr-checkbox label {
-            display: flex;
-            align-items: center;
-            gap: 8px;
-            font-weight: 600;
-        }
-        input[type="checkbox"] {
-            accent-color: black !important;
-            width: 18px;
-            height: 18px;
-        }
-        .gr-button {
-            font-weight: bold;
-            border-radius: 10px !important;
-            background-color: white !important;
-            color: black !important;
-            border: 2px solid black !important;
-        }
-        .gr-button:hover {
-            background-color: #fff3e0 !important;
-            color: black !important;
-        }
-        label {
-            font-weight: 600 !important;
-        }
-    </style>
-    """)
-
-    gr.Markdown("# 🚗 Busca de Combinações de Consórcio Auto")
-    gr.Markdown("Insira os valores desejados para crédito, entrada e parcela, e filtre por administradora.")
-
-    with gr.Row():
-        credito_input = gr.Number(label="Crédito Desejado (R$)", value=30000, step=1000)
-        entrada_input = gr.Number(label="Entrada Desejada (R$)", value=3000, step=100)
-        parcela_input = gr.Number(label="Parcela Desejada (R$)", value=500, step=10)
-
-    administradoras_multiselect = gr.CheckboxGroup(
-        label="Filtrar por Administradoras",
-        choices=get_administradoras_options(),
-        value=get_administradoras_options()  # Seleciona todas por padrão
-    )
-
-    submit_btn = gr.Button("🔍 Buscar Combinações", variant="primary")
-    output_text = gr.Textbox(label="Resultados da Busca", lines=20, max_lines=40, interactive=False, placeholder="Os resultados aparecerão aqui...")
-
-    submit_btn.click(
-        fn=ajustar_e_buscar,
-        inputs=[credito_input, entrada_input, parcela_input, administradoras_multiselect],
-        outputs=output_text
-    )
-
-    gr.Markdown("---")
-    gr.Markdown("Desenvolvido para auxiliar na busca por cotas de consórcio ideais.")
+# Interface usando gr.Interface (API automática em /api/predict)
+demo = gr.Interface(
+    fn=buscar_combinacoes,
+    inputs=[
+        gr.Number(label="Crédito Desejado (R$)", value=30000, step=1000),
+        gr.Number(label="Entrada Desejada (R$)", value=3000, step=100),
+        gr.Number(label="Parcela Desejada (R$)", value=500, step=10),
+        gr.Dropdown(
+            choices=ADMIN_OPCOES,
+            label="Filtrar por Administradoras",
+            multiselect=True,
+            value=ADMIN_OPCOES
+        )
+    ],
+    outputs=gr.Textbox(label="Resultados da Busca", lines=20, max_lines=40),
+    title="🚗 Busca de Combinações de Consórcio Auto",
+    description="Insira os valores desejados para crédito, entrada e parcela, e filtre por administradora."
+)
 
 if __name__ == "__main__":
     demo.launch()