Update app.py

app.py

@@ -1,103 +1,74 @@
-# -*- coding: utf-8 -*-
-"""Imovel.ipynb
-Automatically generated by Colab.
-Original file is located at
-    https://colab.research.google.com/drive/1DZAQZfrXvTy358Vg2qfUg_ge6b3kcgBy
-"""
-
-# app.py
 import pandas as pd
 import requests
 import io
 import itertools
 from collections import defaultdict
 import gradio as gr
-import time
 
-# 🔧 CONFIGURAÇÕES GLOBAIS
-# Certifique-se de que este FILE_ID esteja acessível publicamente ou para "Qualquer pessoa com o link"
-FILE_ID = "1XvNz7DlWioT64uNdcfBSL8Ui2nRSi9gT"
+# 🔧 CONFIGURAÇÕES
+FILE_ID = "1XvNz7DlWioT64uNdcfBSL8Ui2nRSi9gT"  # Planilha de Imóvel
 URL = f"https://drive.google.com/uc?export=download&id={FILE_ID}"
-
-# Variável global para armazenar o DataFrame
 df = None
 
+# --- Funções ---
 def carregar_planilha():
-    """Carrega e prepara a planilha para análise."""
     global df
     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(" ", "_")
-        df['administradora'] = df['administradora'].str.strip().str.lower()
+        df['administradora'] = df['administradora'].astype(str).str.strip().str.lower()
 
+        num_cols = ['valor_credito', 'valor_entrada', 'valor_parcela', 'saldo_devedor', 'qtd_parcelas']
+        for col in num_cols:
+            if col in df.columns:
+                df[col] = pd.to_numeric(df[col], errors='coerce').fillna(0)
 
-        # 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
+            raise ValueError("Colunas essenciais ausentes.")
         df.drop_duplicates(inplace=True)
-
-        print("✅ Planilha carregada com sucesso")
         return True
-
     except Exception as e:
-        print(f"❌ Erro ao carregar planilha: {str(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 dataframe.iterrows():
         grupos[row['administradora']].append(row.to_dict())
     return grupos
 
 def buscar_combinacoes(credito_desejado, entrada_desejada, parcela_desejada, administradoras=None):
-    """Busca combinações de cotas dentro dos parâmetros especificados."""
-    # Garante que o DataFrame está carregado
+    global df
     if df is None:
-        # Tenta carregar se ainda não foi carregado
         if not carregar_planilha():
-            return "❌ Erro: Planilha não carregada. Por favor, tente novamente."
+            return "❌ Erro ao carregar planilha."
 
-    # Filtra por tipo de bem e administradoras
     df_filtrado = df[df['tipo_do_bem'].str.lower() == 'imovel']
     if administradoras:
-        # Certifica-se de que administradoras é uma lista (Gradio retorna tupla)
         if isinstance(administradoras, tuple):
             administradoras = list(administradoras)
         administradoras = [a.strip().lower() for a in administradoras]
         df_filtrado = df_filtrado[df_filtrado['administradora'].isin(administradoras)]
 
     if df_filtrado.empty:
-        return "🚫 Nenhuma cota encontrada com os filtros aplicados. Tente ajustar os parâmetros."
+        return "🚫 Nenhuma cota encontrada para imóvel com os filtros fornecidos."
 
-    # Define faixas de tolerância
     credito_min, credito_max = credito_desejado * 0.93, credito_desejado * 1.07
     entrada_min, entrada_max = entrada_desejada * 0.93, entrada_desejada * 1.07
-    parcela_min, parcela_max = parcela_desejada * 0.75, parcela_desejada * 1.25
+    parcela_min, parcela_max = parcela_desejada * 0.5, parcela_desejada * 1.5
 
     grupos = agrupar_por_administradora(df_filtrado)
     resultados = []
 
     for adm, cotas in grupos.items():
-        for r in range(1, min(3, len(cotas)) + 1):  # Máximo 3 cotas por combinação
+        for r in range(1, min(3, len(cotas)) + 1):
             for combinacao in itertools.combinations(cotas, r):
                 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)  # aplica os 8,5% sobre o crédito
+                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
@@ -106,25 +77,49 @@ def buscar_combinacoes(credito_desejado, entrada_desejada, parcela_desejada, adm
 
                     total_saldo = sum(c['saldo_devedor'] for c in combinacao)
                     valor_final = round(total_entrada + total_saldo, 2)
-                    diferenca = abs(valor_final - total_credito) # Diferença do valor final pelo crédito (para ordenação)
-
+                    diferenca = abs(valor_final - total_credito)
                     resultados.append((adm, combinacao, diferenca))
 
     if not resultados:
-        return "🚫 Nenhuma combinação adequada encontrada com os parâmetros fornecidos."
-
-    resultados.sort(key=lambda x: x[2]) # Ordena pela diferença
+        return "🚫 Nenhuma combinação encontrada."
 
+    resultados.sort(key=lambda x: x[2])
     return formatar_resultados(resultados)
 
 def formatar_moeda(valor):
     return f"R$ {valor:,.2f}".replace(",", "v").replace(".", ",").replace("v", ".")
 
 def formatar_resultados(resultados):
-    """Formata os resultados para exibição."""
     resposta = ""
-    # Limita a 15 resultados para evitar saídas muito longas
-    for idx, (adm, combinacao, _) in enumerate(resultados[:15], 1):
+    grouped_by_adm = defaultdict(list)
+    for adm, combinacao, diferenca in resultados:
+        grouped_by_adm[adm].append((combinacao, diferenca))
+    for adm in grouped_by_adm:
+        grouped_by_adm[adm].sort(key=lambda x: x[1])
+
+    final_resultados = []
+    adm_indices = {adm: 0 for adm in grouped_by_adm.keys()}
+    active_adms = list(grouped_by_adm.keys())
+    current_adm_idx = 0
+
+    while len(final_resultados) < 15 and active_adms:
+        adm_da_vez = active_adms[current_adm_idx]
+        idx_na_adm = adm_indices[adm_da_vez]
+        if idx_na_adm < len(grouped_by_adm[adm_da_vez]):
+            combinacao_data, _ = grouped_by_adm[adm_da_vez][idx_na_adm]
+            final_resultados.append((adm_da_vez, combinacao_data))
+            adm_indices[adm_da_vez] += 1
+        else:
+            active_adms.pop(current_adm_idx)
+            if current_adm_idx >= len(active_adms) and active_adms:
+                current_adm_idx = 0
+
+        if active_adms:
+            current_adm_idx = (current_adm_idx + 1) % len(active_adms)
+        else:
+            break
+
+    for idx, (adm, combinacao) in enumerate(final_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)
@@ -132,7 +127,6 @@ def formatar_resultados(resultados):
         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
@@ -156,93 +150,22 @@ def formatar_resultados(resultados):
 """
     return resposta
 
-# Carrega a planilha uma vez quando o aplicativo inicia
+# Inicializa o DataFrame ao carregar
 carregar_planilha()
 
 def ajustar_e_buscar(credito_desejado, entrada_desejada, parcela_desejada, administradoras):
     entrada_calculada = credito_desejado * 0.085 + entrada_desejada
     return buscar_combinacoes(credito_desejado, entrada_calculada, parcela_desejada, administradoras)
 
-# Prepara a lista de administradoras para o Gradio
 def get_administradoras_options():
     if df is not None:
         adm_list = df[df['tipo_do_bem'].str.lower() == 'imovel']['administradora'].unique()
-        return sorted(adm_list)
+        return sorted(adm_list.tolist())
     return []
 
-# Criando a interface Gradio
-with gr.Blocks(title="Sistema de Busca de Consórcio Imóvel") 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;
-        }
-    
-        /* Inputs Numéricos: Crédito, Entrada, Parcela */
-        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;
-        }
-    
-        /* Inputs de texto e áreas de texto (como a saída e código) */
-        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 Imóvel")
-    gr.Markdown("Insira os valores desejados para crédito, entrada e parcela, e filtre por administradora.")
+# Interface Gradio
+with gr.Blocks(title="Sistema de Consórcio de Imóvel") as demo:
+    gr.Markdown("# 🏠 Sistema de Busca de Combinações de Consórcio (Imóvel)")
 
     with gr.Row():
         credito_input = gr.Number(label="Crédito Desejado (R$)", value=300000, step=1000)
@@ -250,24 +173,20 @@ with gr.Blocks(title="Sistema de Busca de Consórcio Imóvel") as demo:
         parcela_input = gr.Number(label="Parcela Desejada (R$)", value=1500, step=100)
 
     administradoras_multiselect = gr.CheckboxGroup(
-        label="Filtrar por Administradoras",
+        label="Filtrar Administradoras",
         choices=get_administradoras_options(),
-        value=get_administradoras_options() # Seleciona todas por padrão
+        value=get_administradoras_options()
     )
 
-    submit_btn = gr.Button("🔍 Buscar Combinações", variant="primary")
+    buscar_btn = gr.Button("🔍 Buscar Combinações")
+    resultados_output = gr.Textbox(label="Resultados", lines=25, interactive=False)
 
-    output_text = gr.Textbox(label="Resultados da Busca", lines=20, max_lines=40, interactive=False, placeholder="Os resultados aparecerão aqui...")
-
-    submit_btn.click(
+    buscar_btn.click(
         fn=ajustar_e_buscar,
         inputs=[credito_input, entrada_input, parcela_input, administradoras_multiselect],
-        outputs=output_text
+        outputs=resultados_output
     )
 
-    gr.Markdown("---")
-    gr.Markdown("Desenvolvido para auxiliar na busca por cotas de consórcio ideais.")
-
-# Inicia o aplicativo Gradio
+# Não lançar no Colab
 if __name__ == "__main__":
-    demo.launch() # Você não precisa disso no Colab, mas sim no arquivo que você fará o deploy
+    demo.launch()