Update app.py

app.py

@@ -1,8 +1,6 @@
 # -*- coding: utf-8 -*-
 """Imovel.ipynb
-
 Automatically generated by Colab.
-
 Original file is located at
     https://colab.research.google.com/drive/1DZAQZfrXvTy358Vg2qfUg_ge6b3kcgBy
 """
@@ -77,10 +75,11 @@ def buscar_combinacoes(credito_desejado, entrada_desejada, parcela_desejada, adm
     # 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'].str.lower().isin(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."
@@ -121,43 +120,11 @@ def buscar_combinacoes(credito_desejado, entrada_desejada, parcela_desejada, adm
 def formatar_moeda(valor):
     return f"R$ {valor:,.2f}".replace(",", "v").replace(".", ",").replace("v", ".")
 
-from collections import defaultdict
-
 def formatar_resultados(resultados):
-    """Formata os resultados para exibição com diversidade entre administradoras."""
+    """Formata os resultados para exibição."""
     resposta = ""
-    
-    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_diversificados = []
-    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_diversificados) < 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_diversificados.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_diversificados, 1):
+    # Limita a 15 resultados para evitar saídas muito longas
+    for idx, (adm, combinacao, _) in enumerate(resultados[:15], 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)
@@ -199,8 +166,8 @@ def ajustar_e_buscar(credito_desejado, entrada_desejada, parcela_desejada, admin
 # Prepara a lista de administradoras para o Gradio
 def get_administradoras_options():
     if df is not None:
-        adms = df[df['tipo_do_bem'].str.lower() == 'imovel']['administradora'].unique()
-        return sorted([adm.title() for adm in adms])  # Mostra os nomes com letras maiúsculas
+        adm_list = df[df['tipo_do_bem'].str.lower() == 'imovel']['administradora'].unique()
+        return sorted(adm_list)
     return []
 
 # Criando a interface Gradio