my-email-bot / app.py
siddhartharya's picture
Update app.py
16aec6d verified
raw
history blame
1.41 kB
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()