Spaces:
Sleeping
Sleeping
import os | |
import gradio as gr | |
from groq import Groq | |
# Initialize Groq client | |
client = Groq(api_key=os.environ.get("GROQ_API_KEY")) | |
# System prompt with recipient's profile | |
SYSTEM_PROMPT = """ | |
You are an AI assistant that generates personalized emails. The recipient is John Doe, a 45-year-old marketing executive at a tech company. He is interested in innovative marketing strategies and has a busy schedule. | |
""" | |
def generate_email(name, project, key_benefits): | |
prompt = f"Generate an email to the person described in the system prompt. The email should be from {name}, about the project '{project}', and highlight the following key benefits: {key_benefits}" | |
chat_completion = client.chat.completions.create( | |
messages=[ | |
{"role": "system", "content": SYSTEM_PROMPT}, | |
{"role": "user", "content": prompt} | |
], | |
model="mixtral-8x7b-32768", | |
temperature=0.5, | |
max_tokens=500, | |
) | |
return chat_completion.choices[0].message.content | |
# Create Gradio interface | |
iface = gr.Interface( | |
fn=generate_email, | |
inputs=[ | |
gr.Textbox(label="Your Name"), | |
gr.Textbox(label="Project Name"), | |
gr.Textbox(label="Key Benefits") | |
], | |
outputs=gr.Textbox(label="Generated Email"), | |
title="Email Generator", | |
description="Generate a personalized email based on the given inputs." | |
) | |
# Launch the app | |
iface.launch() |