Update app.py

app.py

@@ -1,30 +1,25 @@
-import gradio as gr
 import os
+import gradio as gr
 from groq import Groq
 
 # Initialize Groq client
-client = Groq(
-    api_key=os.environ["GROQ_API_KEY"],
-)
+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"""
-    Write a professional email to {name} about the {project} project. 
-    Highlight the following key benefits:
-    {key_benefits}
-    
-    The email should be concise, engaging, and persuasive.
-    """
+    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": "user",
-                "content": prompt,
-            }
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user", "content": prompt}
         ],
-        model="mixtral-8x7b-32768",  # or another appropriate Groq model
-        temperature=0.7,
+        model="mixtral-8x7b-32768",
+        temperature=0.5,
         max_tokens=500,
     )
     
@@ -34,14 +29,14 @@ def generate_email(name, project, key_benefits):
 iface = gr.Interface(
     fn=generate_email,
     inputs=[
-        gr.Textbox(label="Recipient Name"),
+        gr.Textbox(label="Your Name"),
         gr.Textbox(label="Project Name"),
-        gr.Textbox(label="Key Benefits (comma-separated)")
+        gr.Textbox(label="Key Benefits")
     ],
     outputs=gr.Textbox(label="Generated Email"),
-    title="AI Email Generator",
-    description="Generate personalized emails using Groq LLM"
+    title="Email Generator",
+    description="Generate a personalized email based on the given inputs."
 )
 
-# Launch the interface
+# Launch the app
 iface.launch()