Update app.py

app.py

@@ -1,98 +1,148 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModel, pipeline
-import torch
 import numpy as np
 from typing import List, Dict
 import logging
-from functools import lru_cache
+import json
+import re
+from pathlib import Path
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 class AssistenteBiblico:
     def __init__(self):
-        self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        logger.info(f"Usando dispositivo: {self.device}")
-        
-        model_name = "neuralmind/bert-base-portuguese-cased"
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-        self.model = AutoModel.from_pretrained(model_name).to(self.device)
-        self.analisador_sentimento = pipeline("sentiment-analysis", 
-                                            model="pierreguillou/bert-base-multilingual-cased-sentiment-brazilian-news")
-        
         self.dados_biblia = self.carregar_dados_biblia()
-        self.embeddings_versiculos = self.computar_embeddings()
+        self.prompts_tematicos = self.carregar_prompts()
         
-        # Dicionário de prompts temáticos
-        self.prompts_tematicos = {
-            "Família": [
-                "Como criar filhos segundo a Bíblia?",
-                "Como ter um casamento abençoado?",
-                "Como honrar pai e mãe?",
-                "Como lidar com conflitos familiares?",
-            ],
-            "Trabalho": [
-                "Como ser um bom profissional cristão?",
-                "Como lidar com problemas no trabalho?",
-                "Como encontrar propósito no trabalho?",
-                "Como administrar as finanças biblicamente?",
-            ],
-            "Saúde": [
-                "Como encontrar força na doença?",
-                "Como confiar em Deus nos momentos difíceis?",
-                "Como lidar com ansiedade e depressão?",
-                "Como encontrar paz interior?",
-            ],
-            "Relacionamentos": [
-                "Como perdoar quem nos magoou?",
-                "Como escolher boas amizades?",
-                "Como lidar com a solidão?",
-                "Como amar o próximo?",
-            ],
-            "Vida Espiritual": [
-                "Como crescer na fé?",
-                "Como orar melhor?",
-                "Como ler a Bíblia diariamente?",
-                "Como desenvolver intimidade com Deus?",
-            ]
-        }
-    
-    def carregar_dados_biblia(self):
+    def carregar_dados_biblia(self) -> List[Dict]:
+        """Carrega base de dados de versículos bíblicos"""
         return [
             {
                 "referencia": "João 3:16",
                 "texto": "Porque Deus amou o mundo de tal maneira que deu o seu Filho unigênito, para que todo aquele que nele crê não pereça, mas tenha a vida eterna.",
-                "topico": "Salvação"
+                "palavras_chave": ["amor", "salvação", "vida eterna", "deus", "jesus", "fé"],
+                "topicos": ["Salvação", "Amor de Deus"]
             },
-            # ... [mantém todos os versículos anteriores] ...
             {
-                "referencia": "Efésios 6:4",
-                "texto": "E vós, pais, não provoqueis vossos filhos à ira, mas criai-os na disciplina e na admoestação do Senhor.",
-                "topico": "Família"
+                "referencia": "Salmos 23:1-3",
+                "texto": "O Senhor é o meu pastor, nada me faltará. Ele me faz repousar em pastos verdejantes. Leva-me para junto das águas de descanso; refrigera-me a alma.",
+                "palavras_chave": ["paz", "descanso", "confiança", "proteção", "cuidado", "deus"],
+                "topicos": ["Paz", "Confiança"]
             },
             {
-                "referencia": "Colossenses 3:23",
-                "texto": "E tudo quanto fizerdes, fazei-o de coração, como ao Senhor e não aos homens.",
-                "topico": "Trabalho"
+                "referencia": "Filipenses 4:13",
+                "texto": "Posso todas as coisas naquele que me fortalece.",
+                "palavras_chave": ["força", "superação", "fé", "conquista", "jesus"],
+                "topicos": ["Força", "Superação"]
             },
             {
-                "referencia": "Filipenses 4:6-7",
-                "texto": "Não andeis ansiosos por coisa alguma; antes, em tudo, sejam os vossos pedidos conhecidos diante de Deus pela oração e súplica, com ação de graças. E a paz de Deus, que excede todo o entendimento, guardará os vossos corações e os vossos pensamentos em Cristo Jesus.",
-                "topico": "Saúde Mental"
+                "referencia": "Provérbios 22:6",
+                "texto": "Instrua a criança no caminho em que deve andar, e até quando envelhecer não se desviará dele.",
+                "palavras_chave": ["família", "educação", "filhos", "criação", "ensino"],
+                "topicos": ["Família", "Educação"]
+            },
+            {
+                "referencia": "1 Pedro 5:7",
+                "texto": "Lancem sobre ele toda a sua ansiedade, porque ele tem cuidado de vocês.",
+                "palavras_chave": ["ansiedade", "preocupação", "cuidado", "confiança", "paz"],
+                "topicos": ["Ansiedade", "Confiança"]
             }
         ]
-    
-    # [mantém os métodos anteriores de embeddings e busca...]
-    
-    def get_prompt_tematico(self, categoria: str, indice: int) -> str:
-        """Retorna um prompt específico de uma categoria"""
-        return self.prompts_tematicos.get(categoria, [])[indice]
-    
-    def get_categorias(self) -> List[str]:
-        """Retorna todas as categorias disponíveis"""
-        return list(self.prompts_tematicos.keys())
 
-# Interface Gradio atualizada
+    def carregar_prompts(self) -> Dict:
+        """Carrega prompts temáticos organizados por categoria"""
+        return {
+            "Família": {
+                "título": "👨‍👩‍👦 Família",
+                "prompts": [
+                    "Como criar filhos segundo a Bíblia?",
+                    "Como ter um casamento abençoado?",
+                    "Como honrar pai e mãe?",
+                    "Como lidar com conflitos familiares?"
+                ]
+            },
+            "Ansiedade": {
+                "título": "🧘‍♂️ Ansiedade e Paz",
+                "prompts": [
+                    "Como vencer a ansiedade?",
+                    "Como encontrar paz interior?",
+                    "Como confiar em Deus nos problemas?",
+                    "Como lidar com preocupações?"
+                ]
+            },
+            "Trabalho": {
+                "título": "💼 Trabalho",
+                "prompts": [
+                    "Como ser um bom profissional cristão?",
+                    "Como lidar com dificuldades no trabalho?",
+                    "Como servir a Deus no trabalho?",
+                    "Como administrar melhor o dinheiro?"
+                ]
+            },
+            "Saúde": {
+                "título": "🏥 Saúde",
+                "prompts": [
+                    "Como encontrar força na doença?",
+                    "Como orar por cura?",
+                    "Como ajudar alguém doente?",
+                    "Como ter saúde segundo a Bíblia?"
+                ]
+            },
+            "Relacionamentos": {
+                "título": "❤️ Relacionamentos",
+                "prompts": [
+                    "Como perdoar quem me magoou?",
+                    "Como escolher boas amizades?",
+                    "Como lidar com a solidão?",
+                    "Como amar o próximo?"
+                ]
+            }
+        }
+
+    def buscar_versiculos(self, texto: str, max_versiculos: int = 2) -> List[Dict]:
+        """Busca versículos relevantes baseado em palavras-chave"""
+        # Normaliza o texto para busca
+        texto = texto.lower()
+        palavras = set(re.findall(r'\w+', texto))
+        
+        # Calcula relevância dos versículos
+        versiculos_pontuados = []
+        for versiculo in self.dados_biblia:
+            pontos = sum(1 for palavra in palavras 
+                        if palavra in versiculo['palavras_chave'] or
+                        palavra in [p.lower() for p in versiculo['topicos']])
+            if pontos > 0:
+                versiculos_pontuados.append((pontos, versiculo))
+        
+        # Retorna os versículos mais relevantes
+        versiculos_ordenados = sorted(versiculos_pontuados, key=lambda x: x[0], reverse=True)
+        return [v[1] for v in versiculos_ordenados[:max_versiculos]]
+
+    def gerar_resposta(self, pergunta: str) -> str:
+        """Gera resposta baseada na pergunta e versículos encontrados"""
+        try:
+            versiculos = self.buscar_versiculos(pergunta)
+            
+            if not versiculos:
+                return "Desculpe, não encontrei versículos específicos para sua pergunta. Tente reformular ou escolher outro tema."
+            
+            resposta = "📖 Com base na sua pergunta, separei estes versículos:\n\n"
+            
+            for versiculo in versiculos:
+                resposta += f"➤ {versiculo['referencia']}\n"
+                resposta += f"{versiculo['texto']}\n\n"
+            
+            # Adiciona tópicos relacionados
+            topicos = set(sum([v['topicos'] for v in versiculos], []))
+            if topicos:
+                resposta += f"🏷️ Tópicos relacionados: {', '.join(topicos)}\n"
+            
+            return resposta
+            
+        except Exception as e:
+            logger.error(f"Erro gerando resposta: {str(e)}")
+            return "Desculpe, ocorreu um erro ao processar sua pergunta. Por favor, tente novamente."
+
 def criar_interface():
     assistente = AssistenteBiblico()
     
@@ -104,54 +154,57 @@ def criar_interface():
         except Exception as e:
             logger.error(f"Erro: {str(e)}")
             return historico + [(prompt, "Desculpe, ocorreu um erro.")]
-    
-    def usar_prompt_tematico(categoria: str, indice: int, historico: List) -> List:
-        prompt = assistente.get_prompt_tematico(categoria, indice)
-        return processar_prompt(prompt, historico)
-    
+
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.HTML("""
-            <div style="text-align: center; padding: 20px;">
-                <h1>🙏 Assistente Bíblico - Sua Luz na Palavra</h1>
-                <p>Encontre orientação bíblica para sua vida</p>
+            <div style="text-align: center; padding: 20px; background-color: #f0f8ff; border-radius: 10px;">
+                <h1 style="color: #2c3e50;">🙏 Assistente Bíblico</h1>
+                <p style="color: #34495e;">Encontre orientação e conforto na Palavra de Deus</p>
             </div>
         """)
         
         with gr.Row():
             with gr.Column(scale=2):
                 chatbot = gr.Chatbot(
-                    height=400,
+                    height=500,
                     show_label=False,
-                    layout="bubble",
+                    bubble=True,
+                    avatar_images=["🙋", "🤖"]
                 )
                 
                 with gr.Row():
                     msg = gr.Textbox(
                         show_label=False,
                         placeholder="Digite sua pergunta ou use os botões temáticos...",
-                        scale=4
+                        scale=8
                     )
                     limpar = gr.Button("🔄 Limpar", scale=1)
             
             with gr.Column(scale=1):
-                gr.Markdown("### 📚 Temas")
+                gr.Markdown("### 📚 Temas para Explorar")
                 
-                for categoria in assistente.get_categorias():
-                    with gr.Accordion(f"🎯 {categoria}", open=False):
-                        for i, prompt in enumerate(assistente.prompts_tematicos[categoria]):
-                            btn = gr.Button(prompt)
+                for categoria, dados in assistente.prompts_tematicos.items():
+                    with gr.Accordion(dados["título"], open=False):
+                        for prompt in dados["prompts"]:
+                            btn = gr.Button(prompt, size="sm")
                             btn.click(
-                                fn=lambda cat=categoria, idx=i, hist=chatbot: usar_prompt_tematico(cat, idx, hist),
-                                inputs=[chatbot],
+                                fn=processar_prompt,
+                                inputs=[gr.Textbox(value=prompt, visible=False), chatbot],
                                 outputs=[chatbot]
                             )
                 
                 gr.Markdown("""
-                ### 💡 Dicas
-                - Use os botões temáticos para perguntas comuns
-                - Digite suas próprias perguntas
-                - Explore diferentes temas da vida cristã
-                - Receba versículos específicos para cada situação
+                ### 💡 Como usar
+                1. Escolha um tema nos botões
+                2. Ou faça sua própria pergunta
+                3. Receba versículos e orientação
+                4. Explore diferentes temas
+                
+                ### ✨ Sugestões
+                - Seja específico nas perguntas
+                - Explore diferentes temas
+                - Use palavras-chave relevantes
+                - Reflita sobre os versículos
                 """)
         
         msg.submit(processar_prompt, [msg, chatbot], [chatbot])