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}"
    
    try:
        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
    except Exception as e:
        return f"An error occurred: {str(e)}"

# Create Gradio interface
with gr.Blocks() as demo:
    gr.Markdown("# Email Generator")
    gr.Markdown("Generate a personalized email based on the given inputs.")
    
    with gr.Row():
        with gr.Column():
            name = gr.Textbox(label="Your Name", placeholder="Enter your name")
            project = gr.Textbox(label="Project Name", placeholder="Enter the project name")
            benefits = gr.Textbox(label="Key Benefits", placeholder="List key benefits, separated by commas")
            submit_btn = gr.Button("Generate Email")
        
        with gr.Column():
            output = gr.Textbox(label="Generated Email", lines=10)
    
    submit_btn.click(generate_email, inputs=[name, project, benefits], outputs=output)

# Launch the app
demo.launch()