import gradio as gr
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
import torch
import logging
import warnings
from typing import List, Tuple, Dict
import random
from datetime import datetime
from dataclasses import dataclass
from enum import Enum

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
warnings.filterwarnings('ignore')

class ThemeType(Enum):
    MARRIAGE = "casamento"
    FAMILY = "familia"
    SPIRITUAL = "vida_espiritual"
    WORK = "trabalho"
    RELATIONSHIPS = "relacionamentos"
    GENERAL = "geral"

@dataclass
class BiblicalExample:
    question: str
    passage: str
    text: str
    base_response: str
    application: str
    sentiment: str

class SapienciaBiblica:
    def __init__(self):
        logger.info("Inicializando SapiênciaBíblica...")
        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 = []
        self.biblical_examples = self.get_default_examples_dict()
        self.setup_model()

    def setup_model(self):
        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_default_examples_dict(self) -> Dict[str, List[BiblicalExample]]:
        return {
            "casamento": [
                BiblicalExample(
                    question="Como melhorar a comunicação no casamento?",
                    passage="Efésios 4:29",
                    text="Não saia da vossa boca nenhuma palavra torpe, mas só a que for boa para promover a edificação, para que dê graça aos que a ouvem.",
                    base_response="A comunicação efetiva no casamento requer sabedoria, paciência e amor. A Bíblia nos ensina a usar palavras que edificam e não destroem.",
                    application="Pratique escuta ativa, escolha momentos adequados para conversas importantes.",
                    sentiment="supportive"
                ),
                BiblicalExample(
                    question="Como resolver conflitos no casamento?",
                    passage="Efésios 4:26-32",
                    text="Não se ponha o sol sobre a vossa ira. Não deis lugar ao diabo.",
                    base_response="Resolva os conflitos rapidamente, não permita que a raiva se acumule.",
                    application="Resolução diária de conflitos e perdão mútuo.",
                    sentiment="instructive"
                )
            ],
            "familia": [
                BiblicalExample(
                    question="Como criar filhos segundo a Bíblia?",
                    passage="Provérbios 22:6",
                    text="Instrui o menino no caminho em que deve andar, e até quando envelhecer não se desviará dele.",
                    base_response="A criação dos filhos requer dedicação e instrução consistente.",
                    application="Estabeleça momentos diários de devocional em família.",
                    sentiment="instructive"
                )
            ],
            "vida_espiritual": [
                BiblicalExample(
                    question="Como desenvolver uma vida de oração?",
                    passage="1 Tessalonicenses 5:17",
                    text="Orai sem cessar.",
                    base_response="A vida de oração se desenvolve através da prática constante.",
                    application="Reserve um tempo diário para oração.",
                    sentiment="spiritual"
                )
            ],
            "trabalho": [
                BiblicalExample(
                    question="Como ter integridade no trabalho?",
                    passage="Colossenses 3:23",
                    text="E tudo quanto fizerdes, fazei-o de coração, como ao Senhor.",
                    base_response="Trabalhe com excelência e integridade como para o Senhor.",
                    application="Pratique ética e dedicação no trabalho.",
                    sentiment="professional"
                )
            ],
            "relacionamentos": [
                BiblicalExample(
                    question="Como construir amizades verdadeiras?",
                    passage="Provérbios 17:17",
                    text="Em todo tempo ama o amigo, e na angústia nasce o irmão.",
                    base_response="Amizades verdadeiras são construídas com amor e lealdade.",
                    application="Seja presente e fiel nas amizades.",
                    sentiment="friendly"
                )
            ],
            "geral": [
                BiblicalExample(
                    question="Como encontrar paz em tempos difíceis?",
                    passage="João 14:27",
                    text="Deixo-vos a paz, a minha paz vos dou.",
                    base_response="A verdadeira paz vem de Deus e permanece nas tribulações.",
                    application="Confie em Deus em todas as circunstâncias.",
                    sentiment="comforting"
                )
            ]
        }

    def get_examples_for_interface(self) -> List[List[str]]:
        examples = []
        for theme in self.biblical_examples:
            for example in self.biblical_examples[theme]:
                examples.append([theme, example.question])
        return examples

    def get_verse_of_day(self) -> str:
        verses = [
            ("João 3:16", "Porque Deus amou o mundo de tal maneira que deu o seu Filho unigênito..."),
            ("Salmos 23:1", "O Senhor é meu pastor e nada me faltará."),
            ("Filipenses 4:13", "Posso todas as coisas em Cristo que me fortalece."),
            ("Jeremias 29:11", "Porque eu bem sei os pensamentos que tenho a vosso respeito...")
        ]
        verse = random.choice(verses)
        return f"📖 Versículo do Dia:\n{verse[0]}\n\n{verse[1]}"

    def get_daily_prayer_focus(self) -> str:
        focuses = [
            "Gratidão e Louvor",
            "Família e Relacionamentos",
            "Sabedoria e Direção",
            "Paz e Serenidade",
            "Cura e Restauração"
        ]
        focus = random.choice(focuses)
        return f"🙏 Foco de Oração:\n{focus}"

    def analyze_sentiment(self, text: str) -> str:
        positive_words = {'alegria', 'esperança', 'paz', 'amor', 'gratidão', 'feliz', 'bem'}
        negative_words = {'tristeza', 'medo', 'ansiedade', 'preocupação', 'angústia', 'mal', 'dor'}
        
        text_words = set(text.lower().split())
        pos_count = len(text_words.intersection(positive_words))
        neg_count = len(text_words.intersection(negative_words))
        
        return 'positive' if pos_count > neg_count else 'negative' if neg_count > pos_count else 'neutral'

    def find_best_theme(self, question: str) -> str:
        question = question.lower()
        theme_keywords = {
            "casamento": ["casamento", "cônjuge", "esposa", "marido", "casal"],
            "familia": ["família", "filhos", "pais", "criação", "lar"],
            "vida_espiritual": ["oração", "jejum", "adoração", "espiritual", "fé"],
            "trabalho": ["trabalho", "emprego", "carreira", "profissão"],
            "relacionamentos": ["amizade", "relacionamento", "conflito", "perdão"],
        }
        
        for theme, keywords in theme_keywords.items():
            if any(keyword in question for keyword in keywords):
                return theme
        return "geral"

    def get_unique_response(self, question: str, theme: str = None) -> Tuple[str, Dict, str]:
        if not theme or theme not in self.biblical_examples:
            theme = self.find_best_theme(question)
            
        examples = self.biblical_examples[theme]
        example = random.choice(examples)
        sentiment = self.analyze_sentiment(question)
        
        response = self.format_response(example, sentiment)
        metadata = self.create_metadata(example, theme)
        history = self.save_to_history(question, theme, response, metadata)
        
        return response, metadata, history

    def format_response(self, example: BiblicalExample, sentiment: str) -> str:
        intro = {
            'positive': "Que bom que você está buscando orientação! ",
            'negative': "Entendo seu momento e estou aqui para ajudar. ",
            'neutral': "Agradeço sua busca por sabedoria. "
        }
        
        return f"""
🌟 Orientação Personalizada:
{intro[sentiment]}{example.base_response}

📖 Passagem Bíblica:
{example.passage}: {example.text}

✨ Aplicação Prática:
{example.application}

🙏 Observação: 
Esta orientação é baseada em princípios bíblicos. Para questões específicas, 
considere consultar sua liderança espiritual local.
"""

    def create_metadata(self, example: BiblicalExample, theme: str) -> Dict:
        return {
            "passagem": example.passage,
            "tema": theme,
            "aplicacao": example.application,
            "contexto": "Baseado em princípios bíblicos"
        }

    def save_to_history(self, question: str, theme: str, response: str, metadata: Dict) -> str:
        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:
        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 += "─" * 40 + "\n"
        return history_text

def create_interface():
    counselor = SapienciaBiblica()

    with gr.Blocks(theme=gr.themes.Soft()) as demo:
        gr.Markdown("""
        # 🕊️ SapiênciaBíblica
        ### Orientação Divina para Vida Moderna
        """)

        with gr.Row():
            with gr.Column():
                verse_of_day = gr.Textbox(
                    label="📖 Versículo do Dia",
                    value=counselor.get_verse_of_day(),
                    lines=4,
                    interactive=False
                )
                
            with gr.Column():
                prayer_focus = gr.Textbox(
                    label="🙏 Foco de Oração",
                    value=counselor.get_daily_prayer_focus(),
                    lines=4,
                    interactive=False
                )

        with gr.Row():
            with gr.Column():
                theme = gr.Dropdown(
                    choices=[t.value for t in ThemeType],
                    label="🎯 Tema (Opcional)",
                    value="geral",
                    info="Selecione um tema ou deixe em automático"
                )
                
                question = gr.Textbox(
                    label="❓ Sua Pergunta",
                    placeholder="Digite sua pergunta sobre qualquer tema bíblico...",
                    lines=3
                )
                
                submit_btn = gr.Button("🙏 Buscar Orientação", variant="primary")

            with gr.Column():
                answer_output = gr.Textbox(
                    label="✨ Orientação",
                    lines=12
                )
                
                with gr.Accordion("📚 Detalhes"):
                    metadata_output = gr.JSON(
                        label="📋 Informações Detalhadas"
                    )
                    
                    history_output = gr.Textbox(
                        label="📚 Histórico",
                        lines=10,
                        interactive=False
                    )

        # Exemplos para a interface
        gr.Examples(
            examples=counselor.get_examples_for_interface(),
            inputs=[theme, question],
            outputs=[answer_output, metadata_output, history_output],
            fn=counselor.get_unique_response,
            label="📝 Exemplos de Perguntas",
            examples_per_page=5
        )

        submit_btn.click(
            fn=counselor.get_unique_response,
            inputs=[question, theme],
            outputs=[answer_output, metadata_output, history_output]
        )

    return demo

if __name__ == "__main__":
    try:
        logger.info("Iniciando SapiênciaBíblica...")
        demo = create_interface()
        demo.launch(
            server_name="0.0.0.0",
            share=True,
            show_error=True,
            server_port=7860
        )
    except Exception as e:
        logger.error(f"Erro ao iniciar aplicação: {str(e)}")
        raise