Update app.py

app.py

@@ -2,176 +2,168 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForQuestionAnswering
 import torch
 import logging
-from typing import Tuple, Dict
 import warnings
 import json
 from datetime import datetime
 import random
+from pathlib import Path
+
+# Configuração do logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
 warnings.filterwarnings('ignore')
 
-# [O código das constantes BIBLICAL_PASSAGES permanece o mesmo]
-# [A classe BiblicalCounselorModel permanece a mesma]
-
-def create_enhanced_interface():
-    qa_model = BiblicalCounselorModel()
-
-    def get_verse_of_day():
-        """Retorna um versículo diário aleatório"""
-        all_verses = []
-        for theme_verses in BIBLICAL_PASSAGES.values():
-            all_verses.extend(theme_verses)
-        verse = random.choice(all_verses)
-        return f"📖 {verse['passagem']}\n\n{verse['texto']}"
-
-    def get_theme_summary(theme: str) -> str:
-        """Retorna um resumo das passagens bíblicas do tema"""
-        if theme in BIBLICAL_PASSAGES:
-            passages = BIBLICAL_PASSAGES[theme]
-            summary = f"📚 Principais passagens sobre {theme.replace('_', ' ').title()}:\n\n"
-            for p in passages:
-                summary += f"• {p['passagem']}: {p['texto'][:100]}...\n\n"
-            return summary
-        return "Tema não encontrado."
-
-    def save_testimony(testimony: str) -> str:
-        """Salva um testemunho compartilhado"""
-        if testimony.strip():
-            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-            with open("testimonies.txt", "a", encoding="utf-8") as f:
-                f.write(f"\n[{timestamp}] {testimony}\n")
-            return "🙏 Obrigado por compartilhar seu testemunho! Que ele possa edificar outros irmãos."
-        return "Por favor, escreva seu testemunho antes de enviar."
-
-    def process_counseling(theme: str, question: str) -> Tuple[str, str]:
+# Banco de passagens bíblicas
+BIBLICAL_PASSAGES = {
+    "casamento": [
+        {
+            "passagem": "Efésios 5:22-33",
+            "texto": "Vós, mulheres, sujeitai-vos a vossos maridos, como ao Senhor; Porque o marido é a cabeça da mulher, como também Cristo é a cabeça da igreja... Assim devem os maridos amar as suas mulheres, como a seus próprios corpos.",
+            "contexto": "Relacionamento conjugal"
+        }
+    ],
+    "criacao_filhos": [
+        {
+            "passagem": "Provérbios 22:6",
+            "texto": "Instrui o menino no caminho em que deve andar, e até quando envelhecer não se desviará dele.",
+            "contexto": "Educação dos filhos"
+        }
+    ],
+    "papel_mulher": [
+        {
+            "passagem": "Provérbios 31:10-31",
+            "texto": "Mulher virtuosa, quem a achará? O seu valor muito excede o de rubis... Abre a sua boca com sabedoria, e a lei da beneficência está na sua língua.",
+            "contexto": "Mulher virtuosa"
+        }
+    ],
+    "papel_homem": [
+        {
+            "passagem": "1 Timóteo 5:8",
+            "texto": "Mas, se alguém não tem cuidado dos seus, e principalmente dos da sua família, negou a fé, e é pior do que o infiel.",
+            "contexto": "Responsabilidade familiar"
+        }
+    ]
+}
+
+class BiblicalCounselor:
+    def __init__(self):
+        logger.info("Inicializando conselheiro bíblico...")
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.model_name = "pierreguillou/bert-base-cased-squad-v1.1-portuguese"
+        
         try:
-            answer, metadata = qa_model.answer_question(theme, question)
-            return (
-                answer,
-                f"📖 Tema: {metadata['tema'].replace('_', ' ').title()}\n"
-                f"✝️ Passagem: {metadata['passagem']}\n"
-                f"🕊️ Contexto: {metadata['contexto_completo']}"
-            )
+            self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
+            self.model = AutoModelForQuestionAnswering.from_pretrained(self.model_name)
+            self.model.to(self.device)
+            logger.info(f"Modelo carregado com sucesso no dispositivo: {self.device}")
         except Exception as e:
-            logger.error(f"Erro ao processar pergunta: {str(e)}")
-            return "Perdoe-nos, houve uma dificuldade em processar sua pergunta. Vamos orar e tentar novamente.", "N/A"
+            logger.error(f"Erro ao carregar modelo: {str(e)}")
+            raise
 
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        gr.Markdown(
-            """
-            # 🕊️ Conselheiro Bíblico Virtual
-            ### Centro de Aconselhamento e Edificação
+    def get_answer(self, theme: str, question: str) -> tuple:
+        try:
+            # Obtém o contexto bíblico para o tema
+            if theme not in BIBLICAL_PASSAGES:
+                return "Tema não encontrado em nossa base.", "Sem referências disponíveis."
             
-            _"Toda a Escritura é inspirada por Deus e útil para o ensino, para a repreensão, 
-            para a correção e para a instrução na justiça." - 2 Timóteo 3:16_
-            """
-        )
+            passage = BIBLICAL_PASSAGES[theme][0]
+            context = passage["texto"]
+            
+            # Tokenização
+            inputs = self.tokenizer.encode_plus(
+                question,
+                context,
+                return_tensors="pt",
+                max_length=512,
+                truncation=True,
+                padding="max_length"
+            ).to(self.device)
 
-        with gr.Tabs():
-            # Aba de Aconselhamento
-            with gr.Tab("📚 Aconselhamento Bíblico"):
-                with gr.Row():
-                    with gr.Column():
-                        theme = gr.Dropdown(
-                            choices=[
-                                "casamento",
-                                "criacao_filhos",
-                                "papel_mulher",
-                                "papel_homem",
-                                "financas",
-                                "trabalho"
-                            ],
-                            label="Tema de Aconselhamento",
-                            value="casamento"
-                        )
-                        question = gr.Textbox(
-                            label="Sua Pergunta",
-                            placeholder="Qual sua dúvida sobre este tema?",
-                            lines=2
-                        )
-                        submit_btn = gr.Button("🙏 Buscar Orientação Bíblica", variant="primary")
-                    
-                    with gr.Column():
-                        theme_info = gr.Textbox(
-                            label="Passagens Relacionadas ao Tema",
-                            lines=5
-                        )
-                        theme.change(fn=get_theme_summary, inputs=[theme], outputs=[theme_info])
-
-                counseling_output = gr.Textbox(label="Conselho Bíblico", lines=5)
-                references = gr.Textbox(label="Referências Bíblicas", lines=3)
-                
-                submit_btn.click(
-                    fn=process_counseling,
-                    inputs=[theme, question],
-                    outputs=[counseling_output, references]
-                )
-
-            # Aba de Versículo do Dia
-            with gr.Tab("✨ Versículo do Dia"):
-                gr.Markdown("### Palavra para Reflexão")
-                verse_output = gr.Textbox(label="", lines=4)
-                refresh_btn = gr.Button("🔄 Novo Versículo")
-                refresh_btn.click(fn=get_verse_of_day, inputs=[], outputs=[verse_output])
-
-            # Aba de Testemunhos
-            with gr.Tab("💝 Compartilhe seu Testemunho"):
-                gr.Markdown("""
-                ### Compartilhe sua Experiência
-                Conte como a Palavra de Deus tem transformado sua vida
-                """)
-                testimony_input = gr.Textbox(
-                    label="Seu Testemunho",
-                    placeholder="Compartilhe aqui sua experiência...",
-                    lines=5
-                )
-                share_btn = gr.Button("❤️ Compartilhar Testemunho")
-                testimony_output = gr.Textbox(label="Status")
-                share_btn.click(
-                    fn=save_testimony,
-                    inputs=[testimony_input],
-                    outputs=[testimony_output]
-                )
-
-            # Aba de Recursos
-            with gr.Tab("📖 Recursos"):
-                gr.Markdown("""
-                ### Recursos para Estudo Bíblico
-                
-                #### 📚 Temas Disponíveis:
-                - Casamento e Família
-                - Criação de Filhos
-                - Papel da Mulher
-                - Papel do Homem
-                - Finanças
-                - Trabalho e Vocação
-                
-                #### 🙏 Como Usar:
-                1. Escolha um tema de seu interesse
-                2. Faça sua pergunta específica
-                3. Receba orientação baseada nas Escrituras
-                4. Medite nas passagens fornecidas
-                
-                #### ✝️ Lembre-se:
-                - Ore antes de buscar orientação
-                - Leia o contexto completo das passagens
-                - Busque aconselhamento pastoral presencial quando necessário
-                """)
-
-        # Rodapé
-        gr.Markdown("""
-        ---
-        💖 Desenvolvido para edificação do Corpo de Cristo | Versão 2.0
-        
-        _"Porque a palavra de Deus é viva e eficaz..." - Hebreus 4:12_
-        """)
+            # Obtenção da resposta
+            with torch.no_grad():
+                outputs = self.model(**inputs)
+            
+            answer_start = torch.argmax(outputs.start_logits)
+            answer_end = torch.argmax(outputs.end_logits)
+            
+            tokens = self.tokenizer.convert_ids_to_tokens(inputs["input_ids"][0])
+            answer = tokens[answer_start:answer_end + 1]
+            answer = self.tokenizer.convert_tokens_to_string(answer)
+
+            # Formatação da resposta
+            formatted_answer = f"Com base na palavra de Deus, {answer}"
+            reference = f"📖 {passage['passagem']}\n✝️ Contexto: {passage['contexto']}"
 
-    return demo
+            return formatted_answer, reference
+
+        except Exception as e:
+            logger.error(f"Erro ao processar pergunta: {str(e)}")
+            return "Perdoe-nos, ocorreu um erro ao processar sua pergunta.", "Erro no processamento"
+
+def create_interface():
+    try:
+        counselor = BiblicalCounselor()
+        logger.info("Interface sendo criada...")
+
+        with gr.Blocks(theme=gr.themes.Soft()) as demo:
+            gr.Markdown("""
+            # 🕊️ Conselheiro Bíblico
+            ### Orientação baseada nas Sagradas Escrituras
+            """)
+
+            with gr.Row():
+                with gr.Column():
+                    theme = gr.Dropdown(
+                        choices=list(BIBLICAL_PASSAGES.keys()),
+                        label="Tema de Aconselhamento",
+                        value="casamento"
+                    )
+                    question = gr.Textbox(
+                        label="Sua Pergunta",
+                        placeholder="O que gostaria de saber sobre este tema?",
+                        lines=2
+                    )
+                    submit_btn = gr.Button("🙏 Buscar Orientação")
+
+                with gr.Column():
+                    answer_output = gr.Textbox(
+                        label="Conselho Bíblico",
+                        lines=4
+                    )
+                    reference_output = gr.Textbox(
+                        label="Referência Bíblica",
+                        lines=2
+                    )
+
+            submit_btn.click(
+                fn=lambda t, q: counselor.get_answer(t, q),
+                inputs=[theme, question],
+                outputs=[answer_output, reference_output]
+            )
+
+            gr.Markdown("""
+            ---
+            💝 Desenvolvido para edificação | Baseado nas Sagradas Escrituras
+            """)
+
+        return demo
+
+    except Exception as e:
+        logger.error(f"Erro ao criar interface: {str(e)}")
+        raise
 
 if __name__ == "__main__":
-    logger.info("Iniciando ministério de aconselhamento bíblico aprimorado...")
-    demo = create_enhanced_interface()
-    demo.launch(
-        server_name="0.0.0.0",
-        share=True,
-        show_error=True
-    )
+    try:
+        logger.info("Iniciando aplicação...")
+        demo = create_interface()
+        demo.launch(
+            server_name="0.0.0.0",
+            share=True,
+            show_error=True
+        )
+    except Exception as e:
+        logger.error(f"Erro ao iniciar aplicação: {str(e)}")
+        raise