eliza_llm / app.py
giseldo's picture
Update app.py
8acbfc0 verified
raw
history blame
1.42 kB
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
with gr.Blocks() as demo:
gr.Markdown("# ELIZA Chatbot")
gr.Markdown("Converse com ELIZA, uma terapeuta virtual baseada no chatbot clássico de 1966.")
chatbot = 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()