import os
import traceback
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI # Atualizado para importação mais recente
from langchain_core.messages import SystemMessage, HumanMessage, AIMessage # Importa tipos de mensagem
import gradio as gr

# Carrega a chave do .env (bom para desenvolvimento local)
load_dotenv()

# Obtém a chave da API via variável de ambiente (essencial para Hugging Face Spaces Secrets)
# Certifique-se de configurar a Secret 'OPENROUTER_API_KEY' no seu Space
api_key = os.getenv("OPENROUTER_API_KEY")

# Verifica se a chave está disponível
if not api_key:
    print("⚠️ Aviso: Variável OPENROUTER_API_KEY não encontrada. Verifique as Secrets do seu Space.")
    llm = None
else:
    # Define as variáveis que o LangChain espera
    os.environ["OPENAI_API_KEY"] = api_key # Langchain usa essa variável
    os.environ["OPENAI_API_BASE"] = "https://openrouter.ai/api/v1" # OpenRouter base URL

    # Instancia o modelo
    try:
        llm = ChatOpenAI(
            # Você pode experimentar outros modelos otimizados para código se disponíveis via OpenRouter
            model="deepseek/deepseek-r1:free",
            temperature=0.5 # Talvez uma temperatura mais baixa seja melhor para explicações técnicas
        )
    except Exception as e:
        print(f"❌ Erro ao instanciar o modelo LLM: {e}")
        llm = None

# --- Mensagem de Sistema (Instrução para o Tutor de Python) ---
system_prompt = (
    "Você é um Assistente de Tutor para um curso de introdução e intermediário de Python. "
    "Seu papel é auxiliar os alunos a:\n"
    "- Entender conceitos fundamentais e avançados de Python (variáveis, tipos de dados, loops, funções, classes, módulos, etc.).\n"
    "- Explicar o funcionamento de trechos de código Python.\n"
    "- Ajudar a identificar e depurar erros comuns em códigos Python (erros de sintaxe, lógica, exceções).\n"
    "- Sugerir boas práticas de programação em Python.\n"
    "- Indicar recursos úteis para estudo (documentação oficial, tutoriais específicos, etc.).\n\n"
    "Responda de forma clara, didática, passo a passo e paciente. Ao analisar código com erro, explique a causa provável do erro e sugira como corrigi-lo, em vez de apenas fornecer a solução pronta. Incentive o aluno a pensar. "
    "Mantenha o foco estritamente em Python e conceitos de programação relacionados. Se a pergunta for muito fora de escopo, informe educadamente que você só pode ajudar com Python."
)

# --- Função para o Chatbot (sem alterações na lógica interna) ---
def responder(message, history):
    """
    Processa a mensagem do usuário, mantém o histórico e retorna a resposta do LLM.
    """
    if not llm:
        return "❌ Erro: A configuração da API ou do modelo LLM falhou. Verifique as configurações e a chave API."

    try:
        messages = [SystemMessage(content=system_prompt)]
        for human_msg, ai_msg in history:
            messages.append(HumanMessage(content=human_msg))
            messages.append(AIMessage(content=ai_msg))
        messages.append(HumanMessage(content=message))

        resposta = llm.invoke(messages)
        return resposta.content

    except Exception as e:
        print(f"❌ Erro durante a chamada do LLM:\n{traceback.format_exc()}")
        return f"❌ Desculpe, ocorreu um erro ao processar sua solicitação. Verifique a entrada ou tente novamente."

# --- Interface Gradio para o Tutor de Python ---

# !!! Lembre-se de atualizar o arquivo 'bot_avatar.png' no seu repo se desejar !!!
bot_avatar_path = "bot_avatar.png"
user_avatar_path = None

app = gr.ChatInterface(
    fn=responder,
    chatbot=gr.Chatbot(
        label="Chat com o Tutor",
        avatar_images=(user_avatar_path, bot_avatar_path),
        height=600
    ),
    textbox=gr.Textbox(
        placeholder="Digite sua dúvida sobre Python ou cole um trecho de código...",
        label="Sua Mensagem",
        scale=7
    ),
    title="Assistente-Tutor de Python 🐍",
    description="Precisa de ajuda com Python? Pergunte sobre conceitos, peça para explicar código, auxílio com erros e mais!",
    examples=[
        "Qual a diferença entre lista e tupla em Python?",
        "Como funciona um loop `for` com `range`?",
        "Me explique o que é programação orientada a objetos em Python.",
        "Meu código `print('Olá' + 5)` está dando TypeError, por quê?",
        "Como posso ler o conteúdo de um arquivo de texto?",
        "O que significa o erro 'IndentationError'?"
    ]
    # Botões padrão (Retry, Undo, Clear) serão adicionados automaticamente
)

# --- Lançamento da Aplicação ---
if __name__ == "__main__":
    app.launch()