Update app.py

app.py

@@ -5,11 +5,10 @@ Arquivo principal que inicializa o sistema e lança a interface Gradio.
 """
 
 import os
-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
+import gradio as gr
+from interface import configurar_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")
 
 def main():
@@ -25,68 +24,57 @@ def main():
     if not HF_TOKEN:
         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
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False,
+            show_error=True
         )
-        return # Termina a execução se o token estiver faltando
+        return
 
-    print(f"🔑 Token HF encontrado: {HF_TOKEN[:8]}...") # Exibe parte do token para confirmação
+    print(f"🔑 Token HF encontrado: {HF_TOKEN[:8]}...")
 
     # --- 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()
+    try:
+        app_to_launch = configurar_interface()
+        print("🌐 Interface configurada com sucesso. Lançando...")
 
-    print("🌐 Tentando lançar a interface...")
+        # --- 3. Lançar a aplicação ---
+        if isinstance(app_to_launch, (gr.Blocks, gr.Interface)):
+            app_to_launch.launch(
+                server_name="0.0.0.0",
+                server_port=7860,
+                share=False,
+                max_threads=8,
+                show_error=True
+            )
+        else:
+            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
+            )
 
-    # --- 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
-            app_to_launch # CORREÇÃO: Coloque o componente HTML diretamente no bloco 'with'
-        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(
+    except RuntimeError as e:
+        # Captura a exceção lançada por configurar_interface() se menos de 3 modelos estiverem disponíveis
+        print(f"❌ Erro ao configurar a interface: {str(e)}")
+        error_app = gr.Blocks()
+        with error_app:
+            gr.Markdown(f"<h1>Erro de Inicialização</h1><p>{str(e)}</p><p>Por favor, verifique a conexão com o Hugging Face e o token de acesso, ou consulte os logs para mais detalhes.</p>")
+        error_app.launch(
             server_name="0.0.0.0",
             server_port=7860,
             share=False,
             show_error=True
         )
 
-
 if __name__ == "__main__":
     main()