app.py

import gradio as gr
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
import torch
import logging
import warnings
from typing import List, Tuple

# Configuração do logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)
warnings.filterwarnings('ignore')

# Exemplos de perguntas por tema
THEME_EXAMPLES = {
    "casamento": [
        ["casamento", "Como manter um casamento feliz segundo a Bíblia?"],
        ["casamento", "Quais são os princípios bíblicos para resolver conflitos no casamento?"],
        ["casamento", "Como demonstrar amor ao cônjuge de acordo com a Bíblia?"],
        ["casamento", "O que a Bíblia diz sobre perdão no casamento?"],
        ["casamento", "Como orar pelo meu casamento?"]
    ],
    "criacao_filhos": [
        ["criacao_filhos", "Como disciplinar os filhos segundo a Bíblia?"],
        ["criacao_filhos", "Qual a importância de ensinar a Palavra de Deus aos filhos?"],
        ["criacao_filhos", "Como criar filhos tementes a Deus?"],
        ["criacao_filhos", "Como lidar com filhos rebeldes de forma bíblica?"],
        ["criacao_filhos", "Como ensinar valores cristãos aos filhos pequenos?"]
    ],
    "papel_mulher": [
        ["papel_mulher", "Qual o papel da mulher cristã no lar?"],
        ["papel_mulher", "Como ser uma mãe segundo o coração de Deus?"],
        ["papel_mulher", "O que significa ser uma mulher virtuosa?"],
        ["papel_mulher", "Como equilibrar trabalho e família como mulher cristã?"],
        ["papel_mulher", "Como desenvolver uma vida de oração como mulher?"]
    ],
    "papel_homem": [
        ["papel_homem", "Quais as responsabilidades do homem como líder do lar?"],
        ["papel_homem", "Como ser um pai presente segundo a Bíblia?"],
        ["papel_homem", "O que significa ser o sacerdote do lar?"],
        ["papel_homem", "Como liderar a família espiritualmente?"],
        ["papel_homem", "Como demonstrar amor à esposa segundo as escrituras?"]
    ]
}

# Banco de passagens bíblicas [mantido o mesmo de antes]
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:
            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 carregar modelo: {str(e)}")
            raise

    def get_answer(self, theme: str, question: str) -> tuple:
        try:
            if theme not in BIBLICAL_PASSAGES:
                return "Tema não encontrado em nossa base.", "Sem referências disponíveis."
            
            passage = BIBLICAL_PASSAGES[theme][0]
            context = passage["texto"]
            
            inputs = self.tokenizer.encode_plus(
                question,
                context,
                return_tensors="pt",
                max_length=512,
                truncation=True,
                padding="max_length"
            ).to(self.device)

            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)

            formatted_answer = f"Com base na palavra de Deus, {answer}"
            reference = f"📖 {passage['passagem']}\n✝️ Contexto: {passage['contexto']}"

            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 get_theme_examples(theme: str) -> List[List[str]]:
    """Retorna exemplos para um tema específico"""
    return THEME_EXAMPLES.get(theme, [])

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
                    )

            # Exemplos dinâmicos baseados no tema
            examples = gr.Examples(
                examples=THEME_EXAMPLES["casamento"],  # Exemplos iniciais
                inputs=[theme, question],
                outputs=[answer_output, reference_output],
                fn=lambda t, q: counselor.get_answer(t, q),
                label="Exemplos de Perguntas"
            )

            # Atualiza exemplos quando o tema muda
            def update_examples(new_theme):
                examples.update(examples=THEME_EXAMPLES[new_theme])

            theme.change(
                fn=update_examples,
                inputs=[theme]
            )

            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

            #### Como usar:
            1. Selecione um tema de seu interesse
            2. Digite sua pergunta ou use um dos exemplos abaixo
            3. Clique em "Buscar Orientação" para receber o conselho bíblico
            4. Medite na resposta e nas referências fornecidas
            """)

        return demo

    except Exception as e:
        logger.error(f"Erro ao criar interface: {str(e)}")
        raise

if __name__ == "__main__":
    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