Update app.py

app.py

@@ -25,11 +25,10 @@ df = None
 df_auto = None
 grupos_auto = None
 planilha_carregada = False
-ADMINISTRADORAS_LISTA = []  # Lista global de administradoras
 
 def carregar_planilha():
     """Carrega e prepara a planilha para análise."""
-    global df, df_auto, grupos_auto, planilha_carregada, ADMINISTRADORAS_LISTA
+    global df, df_auto, grupos_auto, planilha_carregada
     
     # Se já carregou, não precisa carregar novamente
     if planilha_carregada:
@@ -68,7 +67,7 @@ def carregar_planilha():
 
         # Padroniza os nomes das administradoras
         if 'administradora' in df.columns:
-            df['administradora'] = df['administradora'].astype(str).str.strip()
+            df['administradora'] = df['administradora'].astype(str).str.strip().str.lower()
             print("✅ Administradoras padronizadas")
         else:
             print("⚠️ Coluna 'administradora' não encontrada")
@@ -113,13 +112,10 @@ def carregar_planilha():
         for adm in list(grupos_auto.keys()):
             grupos_auto[adm] = grupos_auto[adm][:MAX_COTAS_POR_ADMIN]
 
-        # CARREGA A LISTA DE ADMINISTRADORAS PARA A API
-        ADMINISTRADORAS_LISTA = obter_lista_administradoras()
-        
         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: {ADMINISTRADORAS_LISTA}")
+        print(f"✅ Administradoras disponíveis: {list(grupos_auto.keys())}")
         return True
 
     except Exception as e:
@@ -128,50 +124,6 @@ def carregar_planilha():
         traceback.print_exc()
         return False
 
-def obter_lista_administradoras():
-    """Obtém a lista de administradoras da planilha."""
-    global df
-    
-    if df is None:
-        print("⚠️ DataFrame não carregado, retornando lista padrão")
-        return ["Ademicon", "Banco do Brasil", "Bancorbrás", "Banrisul", "Bradesco", 
-                "Caixa CNP", "Caixa XS5", "Canopus", "Embracon", "HS", 
-                "HS Consórcio", "Itaú"]
-    
-    try:
-        # Busca por variações do termo 'auto'
-        termos_auto = ['auto', 'automóvel', 'carro', 'veículo', 'automovel']
-        mascara_auto = df['tipo_do_bem'].astype(str).str.lower().isin([t.lower() for t in termos_auto])
-        
-        if mascara_auto.any():
-            df_filtrado = df[mascara_auto]
-        else:
-            df_filtrado = df
-            
-        # Verifica se tem a coluna administradora
-        if 'administradora' in df_filtrado.columns:
-            adm_list = df_filtrado['administradora'].dropna().unique()
-            # Remove strings vazias e mantém o formato original
-            adm_list = [str(adm).strip() for adm in adm_list if str(adm).strip()]
-            adm_list = sorted(list(set(adm_list)))  # Remove duplicatas e ordena
-            
-            if adm_list:
-                print(f"✅ Lista de administradoras carregada: {len(adm_list)} opções")
-                return adm_list
-            else:
-                print("⚠️ Nenhuma administradora encontrada na planilha")
-                return ["Ademicon", "Banco do Brasil", "Bancorbrás", "Banrisul", "Bradesco", 
-                        "Caixa CNP", "Caixa XS5", "Canopus", "Embracon", "HS", 
-                        "HS Consórcio", "Itaú"]
-                
-    except Exception as e:
-        print(f"❌ Erro ao obter lista de administradoras: {str(e)}")
-    
-    # Fallback padrão
-    return ["Ademicon", "Banco do Brasil", "Bancorbrás", "Banrisul", "Bradesco", 
-            "Caixa CNP", "Caixa XS5", "Canopus", "Embracon", "HS", 
-            "HS Consórcio", "Itaú"]
-
 def agrupar_por_administradora(dataframe):
     """Agrupa cotas por administradora."""
     grupos = defaultdict(list)
@@ -250,7 +202,7 @@ def buscar_combinacoes_otimizado(credito_desejado, entrada_desejada, parcela_des
 
     # Filtra administradoras se especificado
     if administradoras and len(administradoras) > 0:
-        administradoras = [a.strip() for a in administradoras]
+        administradoras = [a.strip().lower() for a in administradoras]
         grupos_filtrados = {adm: cotas for adm, cotas in grupos_auto.items() if adm in administradoras}
         print(f"🔍 Filtrando por administradoras: {administradoras}")
     else:
@@ -378,7 +330,7 @@ def buscar_combinacoes_amplo(credito_desejado, entrada_desejada, parcela_desejad
     
     # Filtra administradoras se especificado
     if administradoras and len(administradoras) > 0:
-        administradoras = [a.strip() for a in administradoras]
+        administradoras = [a.strip().lower() for a in administradoras]
         grupos_filtrados = {adm: cotas for adm, cotas in grupos_auto.items() if adm in administradoras}
     else:
         grupos_filtrados = grupos_auto
@@ -548,6 +500,14 @@ def ajustar_e_buscar(credito_desejado, entrada_desejada, parcela_desejada, admin
         traceback.print_exc()
         return f"❌ Erro interno do servidor: {str(e)}"
 
+# Prepara a lista de administradoras para o Gradio
+def get_administradoras_options():
+    if df is not None and planilha_carregada:
+        if 'administradora' in df.columns and 'tipo_do_bem' in df.columns:
+            adm_list = df[df['tipo_do_bem'].astype(str).str.lower() == 'auto']['administradora'].unique()
+            return sorted([str(adm) for adm in adm_list if adm])
+    return ["bradesco", "santander", "itau", "caixa", "banco do brasil"]
+
 # Criando a interface Gradio
 with gr.Blocks(title="Sistema de Busca de Consórcio Auto") as demo:
     gr.HTML("""
@@ -622,33 +582,14 @@ with gr.Blocks(title="Sistema de Busca de Consórcio Auto") as demo:
     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,
-            description="Valor do crédito desejado para o consórcio"
-        )
-        entrada_input = gr.Number(
-            label="Entrada Desejada (R$)", 
-            value=3000, 
-            step=100,
-            description="Valor da entrada desejada"
-        )
-        parcela_input = gr.Number(
-            label="Parcela Desejada (R$)", 
-            value=500, 
-            step=10,
-            description="Valor da parcela mensal desejada"
-        )
+        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)
 
-    # Carrega a lista de administradoras
-    admin_lista = ADMINISTRADORAS_LISTA if ADMINISTRADORAS_LISTA else obter_lista_administradoras()
-    
     administradoras_multiselect = gr.CheckboxGroup(
         label="Filtrar por Administradoras",
-        choices=admin_lista,
-        value=admin_lista[:5] if len(admin_lista) >= 5 else admin_lista,
-        info=f"Total de {len(admin_lista)} administradoras disponíveis"
+        choices=get_administradoras_options(),
+        value=get_administradoras_options()[:3] if len(get_administradoras_options()) >= 3 else get_administradoras_options()
     )
 
     submit_btn = gr.Button("🔍 Buscar Combinações", variant="primary")
@@ -658,43 +599,9 @@ with gr.Blocks(title="Sistema de Busca de Consórcio Auto") as demo:
         lines=20, 
         max_lines=40, 
         interactive=False, 
-        placeholder="Os resultados aparecerão aqui...",
-        show_copy_button=True
+        placeholder="Os resultados aparecerão aqui..."
     )
 
-    # Adiciona documentação da API explicitamente
-    gr.Markdown("""
-    ## 📡 API Documentation
-    
-    ### Endpoint: `/ajustar_e_buscar`
-    
-    **Parameters:**
-    
-    | Parameter | Type | Default | Description |
-    |-----------|------|---------|-------------|
-    | `credito_desejado` | `number` | 30000 | Valor do crédito desejado em R$ |
-    | `entrada_desejada` | `number` | 3000 | Valor da entrada desejada em R$ |
-    | `parcela_desejada` | `number` | 500 | Valor da parcela mensal desejada em R$ |
-    | `administradoras` | `string[]` | %s | Lista de administradoras para filtrar |
-    
-    **Administradoras disponíveis:** %s
-    
-    **Example JavaScript API call:**
-    ```javascript
-    import { Client } from "@gradio/client";
-    
-    const client = await Client.connect("BACKUPMAXCON/combinacoescotas");
-    const result = await client.predict("/ajustar_e_buscar", {
-        credito_desejado: 30000,
-        entrada_desejada: 3000,
-        parcela_desejada: 500,
-        administradoras: ["Ademicon"],
-    });
-    
-    console.log(result.data);
-    ```
-    """ % (str(admin_lista[:3]), ", ".join(admin_lista)))
-
     submit_btn.click(
         fn=ajustar_e_buscar,
         inputs=[credito_input, entrada_input, parcela_input, administradoras_multiselect],
@@ -708,11 +615,9 @@ with gr.Blocks(title="Sistema de Busca de Consórcio Auto") as demo:
 # Inicia o aplicativo Gradio
 if __name__ == "__main__":
     print("🌐 Iniciando servidor Gradio...")
-    print(f"📋 Administradoras carregadas: {ADMINISTRADORAS_LISTA}")
     demo.launch(
         server_name="0.0.0.0", 
         server_port=7860,
         debug=True,
-        show_error=True,
-        share=False
+        show_error=True
     )