Update app.py

app.py

@@ -1,131 +1,444 @@
-# app.py - BACKEND SIMPLIFICADO PARA TESTE
-import os
-os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
-os.environ["GRADIO_QUEUE_ENABLED"] = "False"
-
-import gradio as gr
+# app_auto.py - VERSÃO SIMILAR AO CÓDIGO DE IMÓVEIS
 import pandas as pd
 import requests
 import io
+import itertools
 from collections import defaultdict
+import gradio as gr
+import time
+import math
 
-# DADOS DE TESTE PARA INÍCIO
-def criar_dados_teste():
-    """Cria dados de teste enquanto não carrega a planilha"""
-    dados = []
-    administradoras = ["bradesco", "santander", "itau", "caixa", "banco do brasil"]
-    
-    for i in range(50):
-        dados.append({
-            "administradora": administradoras[i % len(administradoras)],
-            "valor_credito": 20000 + (i * 5000),
-            "valor_entrada": 2000 + (i * 500),
-            "valor_parcela": 300 + (i * 50),
-            "qtd_parcelas": 60,
-            "saldo_devedor": 18000 + (i * 4500),
-            "codigo": f"TEST{i:03d}",
-            "tipo_do_bem": "auto"
-        })
-    
-    return pd.DataFrame(dados)
+# 🔧 CONFIGURAÇÕES GLOBAIS
+FILE_ID = "1EoH8V3Fyls0mNa95VcBN6ZL1B552ty3X"  # ID da planilha de auto
+URL = f"https://drive.google.com/uc?export=download&id={FILE_ID}"
+
+# Variáveis globais para armazenamento
+df = None
+df_auto = None
+grupos_auto = None
+planilha_carregada = False
+
+def carregar_planilha():
+    """Carrega e prepara a planilha para análise."""
+    global df, df_auto, grupos_auto, planilha_carregada
+
+    if planilha_carregada:
+        return True
+
+    try:
+        print("⏳ Carregando planilha...")
+        response = requests.get(URL)
+        response.raise_for_status()
+        data = io.BytesIO(response.content)
+        df = pd.read_excel(data)
+
+        # Padroniza nomes das colunas
+        df.columns = df.columns.str.strip().str.lower().str.replace(" ", "_")
+
+        # Converte colunas numéricas
+        colunas_numericas = ['valor_credito', 'valor_entrada', 'valor_parcela', 'saldo_devedor', 'qtd_parcelas']
+        for coluna in colunas_numericas:
+            if coluna in df.columns:
+                df[coluna] = pd.to_numeric(df[coluna], errors='coerce').fillna(0)
+
+        # Verifica colunas essenciais
+        if 'tipo_do_bem' not in df.columns or 'administradora' not in df.columns:
+            raise ValueError("Colunas essenciais não encontradas")
+
+        # Remove duplicatas
+        df.drop_duplicates(inplace=True)
 
-# Variáveis globais
-df = criar_dados_teste()
-planilha_carregada = True
+        # Pré-filtra e pré-processa dados de auto
+        df_auto = df[df['tipo_do_bem'].str.lower() == 'auto'].copy()
+        grupos_auto = agrupar_por_administradora(df_auto)
 
-def agrupar_por_administradora(df):
+        planilha_carregada = True
+        print("✅ Planilha carregada e pré-processada com sucesso")
+        print(f"✅ Total de cotas de auto: {len(df_auto)}")
+        print(f"✅ Administradoras disponíveis: {list(grupos_auto.keys())}")
+        return True
+
+    except Exception as e:
+        print(f"❌ Erro ao carregar planilha: {str(e)}")
+        return False
+
+def agrupar_por_administradora(dataframe):
+    """Agrupa cotas por administradora."""
     grupos = defaultdict(list)
-    for _, row in df.iterrows():
+    for _, row in dataframe.iterrows():
         grupos[row['administradora']].append(row.to_dict())
     return grupos
 
-grupos_auto = agrupar_por_administradora(df[df['tipo_do_bem'] == 'auto'])
+def calcular_tempo_parcelas(combinacao):
+    """Calcula o tempo de parcelas baseado nas cotas da combinação."""
+    if not combinacao:
+        return ""
+
+    try:
+        cotas_ordenadas = sorted(combinacao, key=lambda x: int(x.get('qtd_parcelas', 0)))
+        prazos = sorted(set(int(c.get('qtd_parcelas', 0)) for c in combinacao))
+
+        if not prazos or prazos[0] <= 0:
+            return ""
+
+        resultado = []
+        prazo_anterior = 0
+
+        for i, prazo_atual in enumerate(prazos):
+            cotas_ativas = [c for c in cotas_ordenadas if int(c.get('qtd_parcelas', 0)) >= prazo_anterior + 1]
+            valor_parcela_total = sum(c.get('valor_parcela', 0) for c in cotas_ativas)
+
+            inicio = prazo_anterior + 1
+            fim = prazo_atual
+
+            if inicio <= fim and valor_parcela_total > 0:
+                resultado.append(f"{inicio}x à {fim}x: {formatar_moeda(valor_parcela_total)}/mês")
+
+            prazo_anterior = prazo_atual
+
+        if prazos and prazo_anterior < max(prazos):
+            cotas_ativas_finais = [c for c in cotas_ordenadas if int(c.get('qtd_parcelas', 0)) > prazo_anterior]
+            if cotas_ativas_finais:
+                valor_parcela_final = sum(c.get('valor_parcela', 0) for c in cotas_ativas_finais)
+                inicio_final = prazo_anterior + 1
+                fim_final = max(prazos)
+                if valor_parcela_final > 0:
+                    resultado.append(f"{inicio_final}x à {fim_final}x: {formatar_moeda(valor_parcela_final)}/mês")
+
+        return "\n".join(resultado) if resultado else ""
+
+    except Exception as e:
+        print(f"❌ Erro ao calcular tempo de parcelas: {str(e)}")
+        return ""
+
+def buscar_combinacoes_otimizado(credito_desejado, entrada_desejada, parcela_desejada, administradoras=None, max_combinacoes=15):
+    """Busca combinações de cotas dentro dos parâmetros especificados de forma otimizada."""
+    global df_auto, grupos_auto
+
+    if not planilha_carregada:
+        if not carregar_planilha():
+            return "❌ Erro: Planilha não carregada. Por favor, tente novamente."
+
+    # Filtra administradoras se especificado
+    if administradoras and len(administradoras) > 0:
+        grupos_filtrados = {adm: cotas for adm, cotas in grupos_auto.items() if adm in administradoras}
+        print(f"🔍 Filtrando por administradoras: {administradoras}")
+    else:
+        grupos_filtrados = grupos_auto
+        print("🔍 Buscando em todas as administradoras")
+
+    if not grupos_filtrados:
+        return "🚫 Nenhuma cota encontrada com os filtros aplicados. Tente ajustar os parâmetros."
+
+    print(f"🔍 Total de administradoras no filtro: {len(grupos_filtrados)}")
+
+    # TOLERÂNCIAS (idênticas ao código de imóveis)
+    if credito_desejado > 10000000:
+        tolerancia_credito = 0.50
+        tolerancia_entrada = 0.50
+        tolerancia_parcela = 1.00
+        max_cotas_combinacao = 25
+    else:
+        tolerancia_credito = 0.25
+        tolerancia_entrada = 0.25
+        tolerancia_parcela = 0.50
+        max_cotas_combinacao = 15
+
+    credito_min = credito_desejado * (1 - tolerancia_credito)
+    credito_max = credito_desejado * (1 + tolerancia_credito)
+    entrada_min = entrada_desejada * (1 - tolerancia_entrada)
+    entrada_max = entrada_desejada * (1 + tolerancia_entrada)
+    parcela_min = parcela_desejada * (1 - tolerancia_parcela)
+    parcela_max = parcela_desejada * (1 + tolerancia_parcela)
+
+    print(f"💰 Valores desejados: Crédito: R$ {credito_desejado:,.2f}, Entrada: R$ {entrada_desejada:,.2f}, Parcela: R$ {parcela_desejada:,.2f}")
 
-def buscar_combinacoes_simples(credito_desejado, entrada_desejada, parcela_desejada, administradoras, max_combinacoes=5):
-    """Busca simplificada para teste"""
     resultados = []
-    
-    for adm in administradoras:
-        if adm in grupos_auto:
-            for cota in grupos_auto[adm][:10]:  # Limita a 10 cotas por admin
-                credito = cota['valor_credito']
-                entrada = cota['valor_entrada']
-                parcela = cota['valor_parcela']
-                
-                # Tolerâncias simples
-                if (0.8 * credito_desejado <= credito <= 1.2 * credito_desejado and
-                    0.8 * entrada_desejada <= entrada <= 1.2 * entrada_desejada and
-                    0.8 * parcela_desejada <= parcela <= 1.2 * parcela_desejada):
-                    
-                    resultados.append({
-                        "administradora": adm,
-                        "credito": credito,
-                        "entrada": entrada,
-                        "parcela": parcela,
-                        "prazo": cota['qtd_parcelas'],
-                        "saldo_devedor": cota['saldo_devedor'],
-                        "codigo": cota['codigo']
-                    })
-    
-    # Limita resultados
-    return resultados[:max_combinacoes]
+    total_combinacoes_testadas = 0
+
+    for adm, cotas in grupos_filtrados.items():
+        print(f"🔍 Analisando administradora: {adm} ({len(cotas)} cotas)")
+
+        cotas_para_combinar = cotas
+
+        for r in range(1, min(max_cotas_combinacao + 1, len(cotas_para_combinar) + 1)):
+            if credito_desejado > 10000000:
+                max_combinacoes_por_adm = 7000
+            else:
+                max_combinacoes_por_adm = 4000
+            combinacoes_geradas = 0
+
+            for combinacao in itertools.combinations(cotas_para_combinar, r):
+                if combinacoes_geradas >= max_combinacoes_por_adm:
+                    break
+
+                total_combinacoes_testadas += 1
+                combinacoes_geradas += 1
+
+                total_credito = sum(c['valor_credito'] for c in combinacao)
+                total_entrada_base = sum(c['valor_entrada'] for c in combinacao)
+                total_entrada = total_entrada_base + (total_credito * 0.085)
+                total_parcelas = sum(c['valor_parcela'] for c in combinacao)
+
+                if (credito_min <= total_credito <= credito_max and
+                    entrada_min <= total_entrada <= entrada_max and
+                    parcela_min <= total_parcelas <= parcela_max):
+
+                    total_saldo = sum(c['saldo_devedor'] for c in combinacao)
+                    valor_final = round(total_entrada + total_saldo, 2)
+                    diferenca = abs(valor_final - total_credito)
+
+                    resultados.append((adm, combinacao, diferenca))
+
+                    if len(resultados) >= max_combinacoes * 3:
+                        break
+
+            if len(resultados) >= max_combinacoes * 3:
+                break
+
+        if len(resultados) >= max_combinacoes * 3:
+            break
+
+    print(f"🔍 Total de combinações testadas: {total_combinacoes_testadas}")
+    print(f"🔍 Combinações válidas encontradas: {len(resultados)}")
+
+    if not resultados:
+        # Busca ampla com tolerâncias maiores
+        print("🔍 Nenhuma combinação encontrada. Tentando busca ampla...")
+        return buscar_combinacoes_amplo(credito_desejado, entrada_desejada, parcela_desejada, administradoras, max_combinacoes)
+
+    resultados.sort(key=lambda x: x[2])
+    return formatar_resultados(resultados[:max_combinacoes])
+
+def buscar_combinacoes_amplo(credito_desejado, entrada_desejada, parcela_desejada, administradoras=None, max_combinacoes=15):
+    """Busca combinações com tolerâncias mais amplas para valores altos."""
+    global grupos_auto
+
+    if administradoras and len(administradoras) > 0:
+        grupos_filtrados = {adm: cotas for adm, cotas in grupos_auto.items() if adm in administradoras}
+    else:
+        grupos_filtrados = grupos_auto
+
+    if credito_desejado > 10000000:
+        tolerancia_credito = 0.50
+        tolerancia_entrada = 0.50
+        tolerancia_parcela = 1.00
+        max_cotas_combinacao = 25
+    else:
+        tolerancia_credito = 0.25
+        tolerancia_entrada = 0.25
+        tolerancia_parcela = 0.50
+        max_cotas_combinacao = 15
+
+    credito_min = credito_desejado * (1 - tolerancia_credito)
+    credito_max = credito_desejado * (1 + tolerancia_credito)
+    entrada_min = entrada_desejada * (1 - tolerancia_entrada)
+    entrada_max = entrada_desejada * (1 + tolerancia_entrada)
+    parcela_min = parcela_desejada * (1 - tolerancia_parcela)
+    parcela_max = parcela_desejada * (1 + tolerancia_parcela)
+
+    resultados = []
+    total_combinacoes_testadas = 0
+
+    for adm, cotas in grupos_filtrados.items():
+        print(f"🔍 BUSCA AMPLA - Analisando administradora: {adm}")
+
+        for r in range(1, min(max_cotas_combinacao + 1, len(cotas) + 1)):
+            combinacoes_geradas = 0
+            max_combinacoes_por_adm = 7000
+
+            for combinacao in itertools.combinations(cotas, r):
+                if combinacoes_geradas >= max_combinacoes_por_adm:
+                    break
+
+                total_combinacoes_testadas += 1
+                combinacoes_geradas += 1
+
+                total_credito = sum(c['valor_credito'] for c in combinacao)
+                total_entrada_base = sum(c['valor_entrada'] for c in combinacao)
+                total_entrada = total_entrada_base + (total_credito * 0.085)
+                total_parcelas = sum(c['valor_parcela'] for c in combinacao)
+
+                if (credito_min <= total_credito <= credito_max and
+                    entrada_min <= total_entrada <= entrada_max and
+                    parcela_min <= total_parcelas <= parcela_max):
+
+                    total_saldo = sum(c['saldo_devedor'] for c in combinacao)
+                    valor_final = round(total_entrada + total_saldo, 2)
+                    diferenca = abs(valor_final - total_credito)
+
+                    resultados.append((adm, combinacao, diferenca))
+
+                    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 formatar_moeda(valor):
+    return f"R$ {valor:,.2f}".replace(",", "v").replace(".", ",").replace("v", ".")
 
 def formatar_resultados(resultados):
+    """Formata os resultados para exibição incluindo tempo de parcelas."""
     if not resultados:
-        return "Nenhuma combinação encontrada com os critérios informados."
-    
+        return "🚫 Nenhuma combinação adequada encontrada com os parâmetros fornecidos."
+
     resposta = ""
-    for i, r in enumerate(resultados, 1):
-        resposta += f"""
---- COMBINAÇÃO {i} ---
-🏢 Administradora: {r['administradora']}
-🔢 Código: {r['codigo']}
-💰 Crédito Total: R$ {r['credito']:,.2f}
-💵 Entrada: R$ {r['entrada']:,.2f}
-💲 Parcela: R$ {r['parcela']:,.2f}
-⏳ Prazo: {r['prazo']} meses
-📉 Saldo Devedor: R$ {r['saldo_devedor']:,.2f}
+    for idx, (adm, combinacao, _) in enumerate(resultados, 1):
+        codigos = [str(c.get('codigo', 'N/A')) for c in combinacao]
+        total_credito = sum(c['valor_credito'] for c in combinacao)
+        total_entrada_base = sum(c['valor_entrada'] for c in combinacao)
+        total_entrada = total_entrada_base + (total_credito * 0.085)
+        total_saldo = sum(c['saldo_devedor'] for c in combinacao)
+        total_parcelas = sum(c['valor_parcela'] for c in combinacao)
+        maior_prazo = max(int(c.get('qtd_parcelas', 0)) for c in combinacao)
+
+        valor_final = round(total_entrada + total_saldo, 2)
+        credito_real = round(total_credito - total_entrada, 2)
+        taxa_total = total_saldo - credito_real
+        taxa_percentual = (taxa_total / credito_real) * 100 if credito_real > 0 else 0
+        taxa_mensal = taxa_percentual / maior_prazo if maior_prazo > 0 else 0
+
+        tempo_parcelas = calcular_tempo_parcelas(combinacao)
 
+        resposta += f"""
+--- COMBINAÇÃO {idx} ---
+🏢 Administradora: {adm}
+🔢 Códigos: {', '.join(codigos)} ({len(combinacao)} cotas)
+💰 Crédito Total: {formatar_moeda(total_credito)}
+💵 Entrada (+comissão): {formatar_moeda(total_entrada)}
+📉 Saldo Devedor: {formatar_moeda(total_saldo)}
+💲 Parcela: {formatar_moeda(total_parcelas)}
+⏳ Prazo: {maior_prazo} meses
+🔚 Valor Final: {formatar_moeda(valor_final)}
+💠 Crédito Líquido: {formatar_moeda(credito_real)}
+📊 Taxa Total: {taxa_percentual:.2f}% ({formatar_moeda(taxa_total)})
+🗓️ Taxa Mensal: {taxa_mensal:.2f}%
+🗓️ Taxa Anual: {taxa_mensal * 12:.2f}%
+⏰ TEMPO DE PARCELAS:
+{tempo_parcelas if tempo_parcelas else "Não foi possível calcular o tempo de parcelas"}
 """
     return resposta
 
+# Carrega a planilha uma vez quando o aplicativo inicia
+carregar_planilha()
+
 def ajustar_e_buscar(credito_desejado, entrada_desejada, parcela_desejada, administradoras):
-    print(f"Recebido: Crédito={credito_desejado}, Entrada={entrada_desejada}, Parcela={parcela_desejada}, Admin={administradoras}")
+    return buscar_combinacoes_otimizado(
+        credito_desejado, entrada_desejada, parcela_desejada, administradoras, max_combinacoes=15
+    )
+
+def get_administradoras_options():
+    if df is not None and planilha_carregada:
+        adm_list = df[df['tipo_do_bem'].str.lower() == 'auto']['administradora'].unique()
+        return sorted(adm_list)
+    return []
+
+# Interface Gradio (idêntica à do imóvel, apenas com título alterado)
+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;
+        }
     
-    if not planilha_carregada:
-        return "Erro: Dados não carregados"
+        body {
+            background-color: #ff8f00 !important;
+            color: black !important;
+        }
     
-    resultados = buscar_combinacoes_simples(
-        credito_desejado, entrada_desejada, parcela_desejada, 
-        [a.lower() for a in administradoras], max_combinacoes=5
-    )
+        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;
+        }
     
-    return formatar_resultados(resultados)
-
-# Interface
-with gr.Blocks(title="Consórcio Auto - TESTE") as demo:
-    gr.Markdown("# 🚗 Busca de Combinações (MODO TESTE)")
+        input[type="text"], textarea {
+            background-color: white !important;
+            border: 2px solid black !important;
+            border-radius: 12px !important;
+            color: black !important;
+            padding: 10px !important;
+        }
     
-    with gr.Row():
-        credito = gr.Number(label="Crédito Desejado (R$)", value=30000)
-        entrada = gr.Number(label="Entrada Desejada (R$)", value=3000)
-        parcela = gr.Number(label="Parcela Desejada (R$)", value=500)
+        input::placeholder, textarea::placeholder {
+            color: #333 !important;
+            opacity: 0.7;
+        }
     
-    administradoras = gr.CheckboxGroup(
-        label="Administradoras",
-        choices=["bradesco", "santander", "itau", "caixa", "banco do brasil"],
-        value=["bradesco"]
-    )
+        .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;
+        }
     
-    buscar_btn = gr.Button("Buscar")
-    resultado = gr.Textbox(label="Resultados", lines=20)
+        .gr-button:hover {
+            background-color: #fff3e0 !important;
+            color: black !important;
+        }
     
-    buscar_btn.click(
-        ajustar_e_buscar,
-        inputs=[credito, entrada, parcela, administradoras],
-        outputs=resultado,
-        api_name="ajustar_e_buscar"
+        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()