import gradio as gr
from groq import Groq

# Inicializa o cliente Groq - você precisará de uma API key
client = Groq()

def generate_response(message, history):
    # Cria um prompt no estilo ELIZA
    prompt = f"""Você é ELIZA, uma terapeuta que responde com empatia e faz perguntas 
    para entender melhor o paciente. Responda à seguinte mensagem do paciente: {message}"""
    
    # Faz a chamada para a API do Groq
    chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "system",
                "content": "Você é ELIZA, uma terapeuta empática que faz perguntas reflexivas."
            },
            {
                "role": "user",
                "content": message
            }
        ],
        model="mixtral-8x7b-32768",  # ou outro modelo disponível no Groq
        temperature=0.7,
        max_tokens=150
    )
    
    # Retorna a resposta gerada
    return chat_completion.choices[0].message.content

# Cria a interface Gradio

    
demo = gr.ChatInterface(
    generate_response,
    title="ELIZA Therapy Session",
    description="Compartilhe seus pensamentos e ELIZA irá ajudar você a refletir sobre eles."
)

# Inicia a interface
demo.launch()