import gradio as gr

def ia_juridica(pergunta, area):
    resposta = f"""
👩‍⚖️ Assistente Jurídico Virtual

📌 *Área selecionada:* **{area}**
📌 *Pergunta:* {pergunta}

⚠️ Esta é uma resposta inicial baseada em inteligência artificial e **não substitui a consulta com um advogado**.  
Aqui está um direcionamento geral para sua dúvida:

"""

    # Respostas simples por área
    if area == "Trabalhista":
        resposta += "Seu caso envolve direitos trabalhistas. Situações comuns incluem demissão, FGTS, jornada, assédio, salário atrasado e verbas rescisórias."
    elif area == "Consumidor":
        resposta += "Se