Update app.py

app.py

@@ -12,8 +12,6 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 warnings.filterwarnings('ignore')
 
-# Existing BIBLICAL_EXAMPLES dictionary remains the same...
-
 class DiverseBiblicalCounselor:
     def __init__(self):
         logger.info("Inicializando conselheiro bíblico...")
@@ -21,6 +19,35 @@ class DiverseBiblicalCounselor:
         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)
@@ -30,6 +57,34 @@ class DiverseBiblicalCounselor:
             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({
@@ -47,7 +102,7 @@ class DiverseBiblicalCounselor:
             return "Nenhuma consulta realizada ainda."
             
         history_text = "📚 Histórico de Consultas:\n\n"
-        for entry in reversed(self.session_history[-5:]):  # Shows last 5 consultations
+        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"
@@ -55,32 +110,53 @@ class DiverseBiblicalCounselor:
             history_text += "─" * 50 + "\n"
         return history_text
 
-    def get_verse_of_the_day(self) -> str:
-        """Retorna um versículo diário baseado na data"""
-        all_verses = []
-        for theme_examples in BIBLICAL_EXAMPLES.values():
-            for example in theme_examples:
-                all_verses.append({
-                    'passagem': example['passagem'],
-                    'texto': example['texto']
-                })
-        
-        # Use a data atual como seed para selecionar o versículo
-        today = datetime.now().strftime("%Y%m%d")
-        random.seed(today)
-        verse = random.choice(all_verses)
+    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]
         
-        return f"📖 Versículo do Dia:\n{verse['passagem']}\n\n{verse['texto']}"
+        inputs = self.tokenizer.encode_plus(
+            question,
+            example["texto"],
+            return_tensors="pt",
+            max_length=512,
+            truncation=True,
+            padding="max_length"
+        ).to(self.device)
 
-    def get_unique_response(self, question: str, theme: str) -> Tuple[str, Dict]:
-        """Gera uma resposta única baseada na pergunta e tema"""
-        # Previous implementation remains the same...
-        # Get response and metadata as before
-        response, metadata = super().get_unique_response(question, theme)  # Assuming this exists
+        with torch.no_grad():
+            outputs = self.model(**inputs)
         
-        # Save to history
-        history = self.save_to_history(question, theme, response, metadata)
+        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():
@@ -92,7 +168,6 @@ def create_interface():
         ### Orientação Bíblica Personalizada
         """)
 
-        # Verse of the day section
         with gr.Row():
             verse_of_day = gr.Textbox(
                 label="Versículo do Dia",
@@ -106,7 +181,7 @@ def create_interface():
                 with gr.Row():
                     with gr.Column():
                         theme = gr.Dropdown(
-                            choices=list(BIBLICAL_EXAMPLES.keys()),
+                            choices=counselor.get_themes(),
                             label="Tema",
                             value="casamento"
                         )
@@ -145,8 +220,7 @@ def create_interface():
                 
                 ### Temas Disponíveis:
                 - Casamento: Conselhos para relacionamento conjugal
-                - Criação de Filhos: Orientação para pais
-                - [outros temas conforme disponível no sistema]
+                - Família: Orientação para vida familiar
                 
                 ### Notas:
                 - As respostas são baseadas em princípios bíblicos
@@ -154,27 +228,16 @@ def create_interface():
                 - O versículo do dia é atualizado diariamente
                 """)
 
-        # Examples section
         gr.Examples(
-            examples=[
-                ["casamento", "Como resolver conflitos no casamento?"],
-                ["casamento", "Como manter um casamento forte espiritualmente?"],
-                ["casamento", "Qual o papel do perdão no casamento?"],
-                ["casamento", "Como manter a fidelidade no casamento?"],
-                ["casamento", "Como lidar com diferenças no casamento?"]
-            ],
+            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"
         )
 
-        def update_outputs(theme, question):
-            response, metadata, history = counselor.get_unique_response(question, theme)
-            return response, metadata, history
-
         submit_btn.click(
-            fn=update_outputs,
+            fn=counselor.get_unique_response,
             inputs=[theme, question],
             outputs=[answer_output, metadata_output, history_output]
         )