app.py

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 Siddhant Goswami, the Co-Founder 100x Engineers and a Generative AI Instructor
He live in Bengaluru, Karnataka, India
He is Currently building 100x Engineers.
He has Built and scaled three products to seven figures.
Scenes (acquired by Unacademy),
Tapchief (acquired by Unacademy),
Relevel
"""

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