Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import AutoTokenizer, AutoModelForQuestionAnswering | |
| import torch | |
| import logging | |
| import warnings | |
| from typing import List, Tuple, Dict | |
| import random | |
| import hashlib | |
| from datetime import datetime | |
| logging.basicConfig(level=logging.INFO) | |
| logger = logging.getLogger(__name__) | |
| warnings.filterwarnings('ignore') | |
| class DiverseBiblicalCounselor: | |
| 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" | |
| self.session_history = [] | |
| # Movendo os exemplos para dentro da classe | |
| self.biblical_examples = { | |
| "casamento": [ | |
| { | |
| "pergunta": "Como resolver conflitos no casamento?", | |
| "passagem": "Efésios 4:26-32", | |
| "texto": """Não se ponha o sol sobre a vossa ira. Não deis lugar ao diabo. Toda amargura, cólera, ira, gritaria e blasfêmia sejam tiradas dentre vós, bem como toda malícia. Antes, sede bondosos uns para com os outros, compassivos, perdoando uns aos outros, como também Deus vos perdoou em Cristo.""", | |
| "resposta_base": """O princípio bíblico para resolução de conflitos no casamento envolve três aspectos fundamentais: não deixar a ira se prolongar, comunicar-se com amor e praticar o perdão mútuo. A Palavra nos ensina que devemos resolver os conflitos rapidamente, não permitindo que o sol se ponha sobre nossa ira.""", | |
| "aplicacao": "Resolução diária de conflitos, perdão e reconciliação" | |
| }, | |
| { | |
| "pergunta": "Como manter um casamento forte espiritualmente?", | |
| "passagem": "Eclesiastes 4:9-12", | |
| "texto": """É melhor serem dois do que um, porque têm melhor paga do seu trabalho. Porque se um cair, o outro levanta o seu companheiro; mas ai do que estiver só; pois, caindo, não haverá outro que o levante. Também, se dois dormirem juntos, eles se aquentarão; mas um só, como se aquentará? E, se alguém prevalecer contra um, os dois lhe resistirão; e o cordão de três dobras não se quebra tão depressa.""", | |
| "resposta_base": """A força espiritual do casamento vem da união em três dimensões: o marido, a esposa e Deus. Como um cordão de três dobras, esta união proporciona apoio mútuo, fortalecimento espiritual e resistência nas adversidades.""", | |
| "aplicacao": "Desenvolvimento espiritual conjunto e oração em casal" | |
| } | |
| ], | |
| "familia": [ | |
| { | |
| "pergunta": "Como educar filhos segundo a Bíblia?", | |
| "passagem": "Provérbios 22:6", | |
| "texto": """Instrui o menino no caminho em que deve andar, e até quando envelhecer não se desviará dele.""", | |
| "resposta_base": """A educação dos filhos deve ser fundamentada nos princípios bíblicos, com amor, disciplina e instrução constante. Os pais são responsáveis por guiar seus filhos no caminho do Senhor desde cedo.""", | |
| "aplicacao": "Ensino diário dos princípios bíblicos aos filhos" | |
| } | |
| ] | |
| } | |
| 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_themes(self) -> List[str]: | |
| """Retorna lista de temas disponíveis""" | |
| return list(self.biblical_examples.keys()) | |
| def get_examples(self) -> List[List[str]]: | |
| """Retorna exemplos formatados para a interface""" | |
| examples = [] | |
| for theme in self.biblical_examples: | |
| for example in self.biblical_examples[theme]: | |
| examples.append([theme, example["pergunta"]]) | |
| return examples | |
| def get_verse_of_the_day(self) -> str: | |
| """Retorna um versículo diário baseado na data""" | |
| all_verses = [] | |
| for theme_examples in self.biblical_examples.values(): | |
| for example in theme_examples: | |
| all_verses.append({ | |
| 'passagem': example['passagem'], | |
| 'texto': example['texto'] | |
| }) | |
| today = datetime.now().strftime("%Y%m%d") | |
| random.seed(today) | |
| verse = random.choice(all_verses) | |
| return f"📖 Versículo do Dia:\n{verse['passagem']}\n\n{verse['texto']}" | |
| def save_to_history(self, question: str, theme: str, response: str, metadata: Dict): | |
| """Salva a consulta no histórico com timestamp""" | |
| self.session_history.append({ | |
| "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), | |
| "theme": theme, | |
| "question": question, | |
| "response": response, | |
| "metadata": metadata | |
| }) | |
| return self.format_history() | |
| def format_history(self) -> str: | |
| """Formata o histórico de consultas para exibição""" | |
| if not self.session_history: | |
| return "Nenhuma consulta realizada ainda." | |
| history_text = "📚 Histórico de Consultas:\n\n" | |
| for entry in reversed(self.session_history[-5:]): | |
| history_text += f"🕒 {entry['timestamp']}\n" | |
| history_text += f"📌 Tema: {entry['theme']}\n" | |
| history_text += f"❓ Pergunta: {entry['question']}\n" | |
| history_text += f"📖 Passagem: {entry['metadata']['passagem']}\n" | |
| history_text += "─" * 50 + "\n" | |
| return history_text | |
| def get_unique_response(self, question: str, theme: str) -> Tuple[str, Dict, str]: | |
| """Gera uma resposta única baseada na pergunta e tema""" | |
| if theme not in self.biblical_examples: | |
| return "Tema não encontrado", {}, self.format_history() | |
| question_hash = hashlib.md5(question.lower().encode()).hexdigest() | |
| examples = self.biblical_examples[theme] | |
| selected_index = int(question_hash, 16) % len(examples) | |
| example = examples[selected_index] | |
| inputs = self.tokenizer.encode_plus( | |
| question, | |
| example["texto"], | |
| 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]) | |
| model_answer = tokens[answer_start:answer_end + 1] | |
| model_answer = self.tokenizer.convert_tokens_to_string(model_answer) | |
| response = f""" | |
| 🕊️ Conselho Bíblico: | |
| {example['resposta_base']} | |
| 📖 Passagem Bíblica: | |
| {example['passagem']}: {example['texto']} | |
| ✝️ Aplicação Prática: | |
| {example['aplicacao']} | |
| 💭 Reflexão Adicional: | |
| {model_answer} | |
| """ | |
| metadata = { | |
| "passagem": example['passagem'], | |
| "contexto": "Baseado em princípios bíblicos para " + theme.replace('_', ' '), | |
| "aplicacao": example['aplicacao'] | |
| } | |
| history = self.save_to_history(question, theme, response, metadata) | |
| return response, metadata, history | |
| def create_interface(): | |
| counselor = DiverseBiblicalCounselor() | |
| with gr.Blocks(theme=gr.themes.Soft()) as demo: | |
| gr.Markdown(""" | |
| # 🕊️ Conselheiro Bíblico | |
| ### Orientação Bíblica Personalizada | |
| """) | |
| with gr.Row(): | |
| verse_of_day = gr.Textbox( | |
| label="Versículo do Dia", | |
| value=counselor.get_verse_of_the_day(), | |
| lines=4, | |
| interactive=False | |
| ) | |
| with gr.Tabs(): | |
| with gr.TabItem("📝 Consulta"): | |
| with gr.Row(): | |
| with gr.Column(): | |
| theme = gr.Dropdown( | |
| choices=counselor.get_themes(), | |
| label="Tema", | |
| value="casamento" | |
| ) | |
| question = gr.Textbox( | |
| label="Sua Pergunta", | |
| placeholder="Digite sua pergunta...", | |
| lines=3 | |
| ) | |
| submit_btn = gr.Button("🙏 Buscar Orientação", variant="primary") | |
| with gr.Column(): | |
| answer_output = gr.Textbox( | |
| label="Resposta", | |
| lines=10 | |
| ) | |
| metadata_output = gr.Textbox( | |
| label="Referências", | |
| lines=3 | |
| ) | |
| with gr.TabItem("📚 Histórico"): | |
| history_output = gr.Textbox( | |
| label="Histórico de Consultas", | |
| value="Nenhuma consulta realizada ainda.", | |
| lines=15, | |
| interactive=False | |
| ) | |
| with gr.TabItem("ℹ️ Ajuda"): | |
| gr.Markdown(""" | |
| ### Como usar o Conselheiro Bíblico: | |
| 1. Selecione um tema relacionado à sua dúvida | |
| 2. Digite sua pergunta de forma clara e objetiva | |
| 3. Clique em "Buscar Orientação" | |
| 4. Receba orientação baseada em princípios bíblicos | |
| ### Temas Disponíveis: | |
| - Casamento: Conselhos para relacionamento conjugal | |
| - Família: Orientação para vida familiar | |
| ### Notas: | |
| - As respostas são baseadas em princípios bíblicos | |
| - Cada consulta é salva no histórico para referência futura | |
| - O versículo do dia é atualizado diariamente | |
| """) | |
| gr.Examples( | |
| examples=counselor.get_examples(), | |
| inputs=[theme, question], | |
| outputs=[answer_output, metadata_output, history_output], | |
| fn=lambda t, q: counselor.get_unique_response(q, t), | |
| label="Exemplos de Perguntas" | |
| ) | |
| submit_btn.click( | |
| fn=counselor.get_unique_response, | |
| inputs=[theme, question], | |
| outputs=[answer_output, metadata_output, history_output] | |
| ) | |
| return demo | |
| 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 |