Update ai_logic.py

ai_logic.py

@@ -356,51 +356,60 @@ api_client = HuggingFaceAPIClient(HF_TOKEN)
 def test_and_update_models() -> int:
     """
     Testa a disponibilidade dos novos modelos e atualiza a lista MODELS.
-    Retorna o número de modelos disponíveis (incluindo os iniciais).
+    Garante que o DEFAULT_MODEL seja sempre o primeiro da lista.
+    Retorna o número de modelos disponíveis.
     """
     print("Testando disponibilidade dos novos modelos...")
     print(f"Token HF disponível: {'Sim' if HF_TOKEN else 'Não'}")
     print("-" * 60)
     
-    initial_model_count = len(MODELS)
-    available_models_during_test = []
-    unavailable_models = []
-
+    # Cria um dicionário temporário para os modelos disponíveis
+    temp_models = {}
+    
+    # Primeiro verifica o modelo padrão
+    default_label, default_name = "Mistral 7B (Mais acertivo)", "mistralai/Mistral-7B-Instruct-v0.3"
+    is_available, message = api_client.test_model_availability(default_name)
+    
+    if is_available:
+        temp_models[default_label] = default_name
+        print(f"✓ {default_label} (DEFAULT MODEL)")
+    else:
+        print(f"✗ {default_label} - {message} (MODELO PADRÃO INDISPONÍVEL)")
+    
+    # Depois verifica os outros modelos
     for model_label, model_name in NEW_MODELS_TO_TEST:
+        # Pula o modelo padrão se já foi testado
+        if model_label == default_label and model_name == default_name:
+            continue
+            
         is_available, message = api_client.test_model_availability(model_name)
         
         if is_available:
-            if model_label not in MODELS:
-                MODELS[model_label] = model_name
-            available_models_during_test.append((model_label, model_name, message))
+            temp_models[model_label] = model_name
             print(f"✓ {model_label}")
         else:
-            unavailable_models.append((model_label, model_name, message))
             print(f"✗ {model_label} - {message}")
             
         time.sleep(1)
-    DEFAULT_MODEL = MODELS[0]
+    
+    # Atualiza MODELS garantindo que o padrão seja o primeiro
+    global MODELS
+    MODELS.clear()
+    
+    # Adiciona primeiro o modelo padrão (se disponível)
+    if default_label in temp_models:
+        MODELS[default_label] = temp_models.pop(default_label)
+    
+    # Adiciona os demais modelos
+    MODELS.update(temp_models)
+    
     print("\n" + "=" * 60)
-    print("RESULTADOS DA VALIDAÇÃO:")
+    print("MODELOS DISPONÍVEIS (ORDEM):")
     print("=" * 60)
+    for i, (label, name) in enumerate(MODELS.items(), 1):
+        print(f"{i}. {label}")
     
-    if available_models_during_test:
-        print(f"\n✓ MODELOS DISPONÍVEIS (novos e testados: {len(available_models_during_test)}):")
-        for label, name, msg in available_models_during_test:
-            print(f"  - {label}")
-            print(f"    {name}")
-            print(f"    Status: {msg}")
-            print()
-            
-    if unavailable_models:
-        print(f"\n✗ MODELOS NÃO DISPONÍVEIS ({len(unavailable_models)}):")
-        for label, name, msg in unavailable_models:
-            print(f"  - {label}")
-            print(f"    {name}")
-            print(f"    Motivo: {msg}")
-            print()
-            
-    print(f"TOTAL DE MODELOS ATUALMENTE DISPONÍVEIS: {len(MODELS)}")
+    print(f"\nTOTAL DE MODELOS DISPONÍVEIS: {len(MODELS)}")
     print("=" * 60)
     
     save_updated_models()