Update app.py

app.py

@@ -7,6 +7,8 @@ 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}"
@@ -83,7 +85,77 @@ def calcular_tempo_parcelas(combinacao):
         print(f"Erro tempo: {e}")
         return ""
 
-def buscar_combinacoes(credito, entrada, parcela, adm_selecionadas):
+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):
     if not planilha_carregada:
         if not carregar_planilha():
             return "❌ Erro: planilha não carregada."
@@ -134,9 +206,11 @@ def buscar_combinacoes(credito, entrada, parcela, adm_selecionadas):
             break
 
     if not resultados:
-        return "🚫 Nenhuma combinação encontrada."
+        print("🔍 Nenhuma combinação encontrada. Tentando busca ampla...")
+        return buscar_combinacoes_amplo(credito, entrada, parcela, adm_selecionadas, max_combinacoes)
+
     resultados.sort(key=lambda x: x[2])
-    return formatar_resultados(resultados[:15])
+    return formatar_resultados(resultados[:max_combinacoes])
 
 def formatar_resultados(resultados):
     if not resultados:
@@ -178,25 +252,98 @@ def formatar_resultados(resultados):
 # Carrega a planilha antes de iniciar
 carregar_planilha()
 
-# 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."
-)
+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.")
 
 if __name__ == "__main__":
-    demo.launch()
-#
+    demo.launch()