Update app.py

app.py

@@ -5,29 +5,90 @@ Arquivo principal que inicializa o sistema
 """
 
 import os
-from ai_logic import load_vector_store, HF_TOKEN
+from huggingface_hub import hf_hub_download
+from ai_logic import load_vector_store
 from interface import configurar_interface
 
+
+# 🔑 Token do Hugging Face vindo das variáveis de ambiente
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# Lista de modelos para testar acesso
+MODELS = [
+    {"name": "Mistral-7B-Instruct-v0.3", "repo_id": "mistralai/Mistral-7B-Instruct-v0.3"},
+    {"name": "Mistral-7B-v0.1", "repo_id": "mistralai/Mistral-7B-v0.1"},
+    {"name": "Mixtral-8x7B-Instruct-v0.1", "repo_id": "mistralai/Mixtral-8x7B-Instruct-v0.1"},
+    {"name": "Phi-3-mini-4k-instruct", "repo_id": "microsoft/Phi-3-mini-4k-instruct"},
+    {"name": "Phi-3-small-8k-instruct", "repo_id": "microsoft/Phi-3-small-8k-instruct"},
+    {"name": "DeepSeek-R1-Distill-Qwen-7B", "repo_id": "unsloth/DeepSeek-R1-Distill-Qwen-7B"},
+    {"name": "DeepSeek-Coder-V2-Lite", "repo_id": "DeepSeek/DeepSeek-Coder-V2-Lite"},
+    {"name": "Gemma-2-9B", "repo_id": "google/gemma-2-9b"},
+    {"name": "Gemma-3-4B", "repo_id": "google/gemma-3-4b"},
+    {"name": "LLaMA-3.1-8B-Instruct", "repo_id": "meta-llama/Llama-3.1-8B-Instruct"},
+    {"name": "Qwen2-7B-Instruct", "repo_id": "Qwen/Qwen2-7B-Instruct"},
+    {"name": "Qwen2.5-7B-Instruct", "repo_id": "Qwen/Qwen2.5-7B-Instruct"},
+    {"name": "Zephyr-7B-Beta", "repo_id": "HuggingFaceH4/zephyr-7b-beta"},
+    {"name": "OpenHermes-2.5-Mistral-7B", "repo_id": "teknium/OpenHermes-2.5-Mistral-7B"},
+    {"name": "Falcon-7B-Instruct", "repo_id": "tiiuae/falcon-7b-instruct"},
+    {"name": "Yi-6B", "repo_id": "01-ai/Yi-6B"},
+    {"name": "XGen-7B-Instruct", "repo_id": "Salesforce/xgen-7b-8k-instruct"},
+    {"name": "StableLM-3B-4e1t", "repo_id": "stabilityai/stablelm-3b-4e1t"},
+    {"name": "Mistral-NeMo", "repo_id": "mistralai/Mixtral-NeMo"},
+    {"name": "Mathstral-7B", "repo_id": "mistralai/Mathstral-7B"}
+]
+
+
+def testar_acesso_modelos():
+    """Testa o acesso a cada modelo da lista."""
+    if not HF_TOKEN:
+        print("⚠️ Token Hugging Face não encontrado. Pulando teste de acesso aos modelos.\n")
+        return
+
+    print("\n🔍 Testando acesso aos modelos Hugging Face...\n")
+    resultados = []
+    for modelo in MODELS:
+        nome = modelo["name"]
+        repo_id = modelo["repo_id"]
+        try:
+            hf_hub_download(
+                repo_id=repo_id,
+                filename="config.json",
+                token=HF_TOKEN
+            )
+            resultados.append(f"✅ {nome}: Acesso OK")
+        except Exception as e:
+            resultados.append(f"❌ {nome}: Sem acesso ({str(e)})")
+    
+    print("🗒️  Resultado dos testes de acesso:")
+    print("------------------------------------------------")
+    for r in resultados:
+        print(r)
+    print("------------------------------------------------\n")
+
+
 def main():
     """Função principal que inicializa o sistema"""
     print("🚀 Iniciando Dr. Aldo Henrique com RAG...")
-    
+
     # Verificar se o token HF está configurado
     if not HF_TOKEN:
         print("❌ Erro: Token HF_TOKEN não encontrado nas variáveis de ambiente")
         return
-    
+
     print(f"🔑 Token HF encontrado: {HF_TOKEN[:8]}...")
-    
-    # Carrega ou constrói o vector store na inicialização
+
+    # 🔍 Testar acesso aos modelos (opcional, apenas informativo)
+    testar_acesso_modelos()
+
+    # ⚙️ Carrega ou constrói o vector store
     load_vector_store()
-    
-    # Configurar a interface
+
+    # 🌐 Configura a interface
     interface = configurar_interface()
-    
+
     print("🌐 Interface pronta!")
-    
-    # Lançar a aplicação
+
+    # 🚀 Lançar a aplicação
     interface.launch(
         server_name="0.0.0.0",
         server_port=7860,
@@ -36,5 +97,6 @@ def main():
         show_error=True
     )
 
+
 if __name__ == "__main__":
-    main()
+    main()