eliza_llm / app.py
giseldo's picture
Update app.py
bf2b287 verified
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()