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()