Update app.py

app.py

@@ -1,101 +1,93 @@
 #!/usr/bin/env python3
 """
 Dr. Aldo Henrique - Chatbot com RAG
-Arquivo principal que inicializa o sistema
+Arquivo principal que inicializa o sistema e lança a interface Gradio.
 """
 
 import os
-from huggingface_hub import hf_hub_download
-from ai_logic import load_vector_store
-from interface import configurar_interface
-
+import gradio as gr # Importa Gradio para verificar o tipo do objeto retornado
+from interface import configurar_interface # Importa a função que configura a interface
 
 # 🔑 Token do Hugging Face vindo das variáveis de ambiente
+# É crucial que este token esteja definido para que os modelos funcionem.
 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"""
+    """
+    Função principal que gerencia o fluxo de inicialização do chatbot:
+    1. Verifica a presença do token Hugging Face.
+    2. Tenta configurar a interface Gradio (que internamente testa os modelos).
+    3. Lança a interface completa ou uma página de erro, dependendo do resultado.
+    """
     print("🚀 Iniciando Dr. Aldo Henrique com RAG...")
 
-    # Verificar se o token HF está configurado
+    # --- 1. 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]}...")
-
-    # 🔍 Testar acesso aos modelos (opcional, apenas informativo)
-    testar_acesso_modelos()
-
-    # ⚙️ Carrega ou constrói o vector store
-    load_vector_store()
-
-    # 🌐 Configura a interface
-    interface = configurar_interface()
-
-    print("🌐 Interface pronta!")
-
-    # 🚀 Lançar a aplicação
-    interface.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        max_threads=8,
-        show_error=True
-    )
+        print("❌ Erro: Token HF_TOKEN não encontrado nas variáveis de ambiente.")
+        print("A interface não será carregada. Por favor, defina a variável de ambiente HF_TOKEN.")
+        # Se o token não está presente, exibe uma página de erro simples
+        error_app_no_token = gr.Blocks()
+        with error_app_no_token:
+            gr.Markdown("<h1>Erro: Token HF_TOKEN não encontrado.</h1><p>Por favor, defina a variável de ambiente **HF_TOKEN** para iniciar o sistema.</p>")
+        error_app_no_token.launch(
+            server_name="0.0.0.0", # Permite acesso externo
+            server_port=7860,      # Porta padrão para Gradio
+            share=False,           # Não compartilha link público
+            show_error=True        # Mostra erros na interface
+        )
+        return # Termina a execução se o token estiver faltando
+
+    print(f"🔑 Token HF encontrado: {HF_TOKEN[:8]}...") # Exibe parte do token para confirmação
+
+    # --- 2. Configurar a interface ---
+    # A função `configurar_interface()` (de interface.py) internamente:
+    # - Chama `ai_logic.inicializar_sistema()` para testar e carregar modelos.
+    # - Retorna um objeto `gr.Blocks` (interface completa) se houver modelos suficientes,
+    #   OU um objeto `gr.HTML` (página de erro) caso contrário.
+    app_to_launch = configurar_interface()
+
+    print("🌐 Tentando lançar a interface...")
+
+    # --- 3. Lançar a aplicação com base no que foi retornado ---
+    # Verifica se o objeto retornado é uma interface Gradio válida
+    if isinstance(app_to_launch, (gr.Blocks, gr.Interface)):
+        # Se for um `gr.Blocks` ou `gr.Interface`, lança a aplicação completa
+        app_to_launch.launch(
+            server_name="0.0.0.0", # Permite acesso externo
+            server_port=7860,      # Porta padrão para Gradio
+            share=False,           # Não compartilha link público
+            max_threads=8,         # Otimização para threads
+            show_error=True        # Mostra erros na interface
+        )
+    elif isinstance(app_to_launch, gr.HTML):
+        # Se for um `gr.HTML`, significa que a inicialização de modelos falhou.
+        # Precisamos envolver esse HTML em um `gr.Blocks` para poder lançá-lo.
+        print("Não foi possível carregar a interface completa devido a modelos insuficientes ou outros erros de inicialização.")
+        print("Exibindo página de erro. Por favor, verifique os logs para mais detalhes.")
+        error_page_block = gr.Blocks()
+        with error_page_block:
+            # Adiciona o componente HTML de erro diretamente ao Blocks
+            error_page_block.add_component(app_to_launch) 
+            # Alternativamente, para HTML simples: gr.HTML(app_to_launch.value) se `app_to_launch` fosse apenas a string.
+            # Mas como `app_to_launch` JÁ É um gr.HTML component, usamos add_component ou simplesmente o colocamos no `with` block
+        error_page_block.launch(
+            server_name="0.0.0.0",
+            server_port=7860, # Usa a porta padrão; se já estiver em uso, Gradio tentará a próxima disponível.
+            share=False,
+            show_error=True
+        )
+    else:
+        # Caso um tipo de objeto inesperado seja retornado
+        print("Erro inesperado: configurar_interface retornou um tipo de objeto desconhecido.")
+        unexpected_error_app = gr.Blocks()
+        with unexpected_error_app:
+            gr.Markdown("<h1>Erro Inesperado</h1><p>Ocorreu um erro desconhecido ao configurar a interface. Por favor, entre em contato com o suporte.</p>")
+        unexpected_error_app.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False,
+            show_error=True
+        )
 
 
 if __name__ == "__main__":