Update app.py

app.py

@@ -7,7 +7,7 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 GROQ_COMPLETION_URL = "https://api.groq.com/openai/v1/chat/completions"
 GROQ_MODEL = "llama-3.1-8b-instant"
 
-def groq_completion(prompt, system_prompt=None):
+def groq_completion(prompt, sys_prompt=None):
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
@@ -15,7 +15,7 @@ def groq_completion(prompt, system_prompt=None):
     body = {
         "model": GROQ_MODEL,
         "messages": [
-            {"role": "system", "content": system_prompt or "You are a fast and reliable business email classification assistant."},
+            {"role": "system", "content": sys_prompt or "You are a fast and reliable business email assistant."},
             {"role": "user", "content": prompt}
         ],
         "temperature": 0.3,
@@ -28,63 +28,98 @@ def groq_completion(prompt, system_prompt=None):
     except Exception as e:
         print("Groq API error:", e)
         if 'response' in locals():
-            print("Groq API response text:", response.text)
+            print("Groq API response:", response.text)
         return "Error"
 
-def email_assistant(email_text):
-    # Clear, strict prompt for JSON extraction
+def email_classifier_router(raw_email):
     prompt = (
-        "Analyze the following business email and extract ONLY the following elements as a compact JSON dictionary:\n"
-        "Category (Sales, Support, Spam, Finance, General),\n"
-        "Confidence (percentage),\n"
-        "Priority (High/Normal/Low),\n"
-        "Suggested Action or Forwarding Address,\n"
-        "Draft professional response,\n"
-        "Action items (bulleted list).\n\n"
-        "Return ONLY a compact JSON dictionary, nothing else:\n"
-        '{"Category":"...", "Confidence":"...", "Priority":"...", "Suggested Action":"...", "Draft Response":"...", "Action Items":"..."}\n\n'
-        f"EMAIL:\n{email_text}\n"
+        "Return ONLY compact JSON for analysis and response.\n"
+        "Given the business email below, extract:\n"
+        "- Category (Support Request, Sales Inquiry, Finance/Billing, Urgent Incident, Spam/Marketing)\n"
+        "- Priority (Low, Medium, High)\n"
+        "- Suggested Recipient (department or address)\n"
+        "- Professional draft reply\n"
+        "- Summary (one sentence)\n"
+        "- Action Items (numbered list)\n\n"
+        "Format:\n"
+        '{"Category":"...", "Priority":"...", "Suggested Recipient":"...", "Draft Response":"...", "Summary":"...", "Action Items":["...","..."]}\n\n'
+        f"{raw_email}"
     )
     raw_result = groq_completion(prompt)
     try:
         output = json.loads(raw_result)
-        return [
-            output.get("Category", "Error"),
-            output.get("Confidence", "Error"),
-            output.get("Priority", "Error"),
-            output.get("Suggested Action", "Error"),
-            output.get("Draft Response", "Error"),
-            output.get("Action Items", "Error"),
-        ]
     except Exception as e:
-        print(f"Parsing or LLM output error: {e}\nRaw output: {raw_result}")
-        # Fail gracefully: show result in all fields for debugging
-        return [raw_result]*6
+        print(f"JSON parse error: {e}\nRAW: {raw_result}")
+        output = {
+            "Category": "Error",
+            "Priority": "Error",
+            "Suggested Recipient": "Error",
+            "Draft Response": raw_result,
+            "Summary": "Error",
+            "Action Items": ["Error"]
+        }
+    return output
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Email Classification & Response Assistant (Groq powered)")
-    email_input = gr.TextArea(
-        label="Paste your email here (include Subject, From, Body if possible)",
-        placeholder="Subject: ...\nFrom: ...\nBody: ..."
+# Gradio UI
+with gr.Blocks(theme=gr.themes.Base(color_mode="dark")) as demo:
+    gr.Markdown(
+        "## 📨 AI Email Classifier & Router\n"
+        "Easily classify, route, and summarize business emails."
     )
-    submit = gr.Button("Analyze Email")
-    output_category = gr.Textbox(label="Category")
-    output_confidence = gr.Textbox(label="Confidence")
-    output_priority = gr.Textbox(label="Priority")
-    output_action = gr.Textbox(label="Suggested Action")
-    output_response = gr.Textbox(label="Draft Response")
-    output_items = gr.Textbox(label="Extracted Action Items")
-    submit.click(
-        email_assistant,
-        inputs=email_input,
-        outputs=[
-            output_category,
-            output_confidence,
-            output_priority,
-            output_action,
-            output_response,
-            output_items,
-        ]
+    
+    with gr.Row():
+        gr.Markdown("**Load an example:**")
+        ex1 = "From: help@company.com\nSubject: Unable to Login\nHi, I can't access my dashboard. Can you help?"
+        ex2 = "From: sales@prospect.com\nSubject: Inquiry about pricing\nHi, Can you send over your latest pricing for enterprise?"
+        ex3 = "From: accounts@supplier.com\nSubject: Invoice #2024-00123 Due Date Reminder\nDear Valued Client, This is a reminder invoice #2024-00123 for $1,500.00 is due..."
+        ex4 = "From: it@company.com\nSubject: Server Down Alert\nURGENT: Database server unreachable since 3AM. Needs escalation."
+        ex5 = "From: spammer@promo.com\nSubject: BIG SALE!!!\nDon't miss our special offers!"
+        gr.Button("Support Request", elem_id="exSupport").click(lambda: ex1, None, 'email_box')
+        gr.Button("Sales Inquiry", elem_id="exSales").click(lambda: ex2, None, 'email_box')
+        gr.Button("Finance / Billing", elem_id="exFinance").click(lambda: ex3, None, 'email_box')
+        gr.Button("Urgent Incident", elem_id="exUrgent").click(lambda: ex4, None, 'email_box')
+        gr.Button("Spam / Marketing", elem_id="exSpam").click(lambda: ex5, None, 'email_box')
+    
+    gr.Markdown("#### 📋 Paste or Load a Full Email:")
+    email_box = gr.Textbox(lines=8, label="Paste the entire raw email (including lines like 'From:', 'Subject:', etc.)", elem_id='email_box')
+    
+    with gr.Row():
+        classify_btn = gr.Button("Classify Email", elem_id='classify_btn', variant="primary")
+    
+    with gr.Column():
+        gr.Markdown("### ✅ Classification Results")
+        category = gr.Textbox(label="Category", elem_id='category')
+        priority = gr.Textbox(label="Priority", elem_id='priority')
+        recipient = gr.Textbox(label="Suggested Recipient", elem_id='recipient')
+
+    with gr.Column():
+        gr.Markdown("### ✨ Generated Response Draft")
+        draft_response = gr.Textbox(lines=8, label="Generated Draft", elem_id='draft_response')
+
+    with gr.Column():
+        gr.Markdown("#### Summary")
+        summary = gr.Textbox(label="Summary", elem_id='summary')
+
+    with gr.Column():
+        gr.Markdown("#### 🟣 Extracted Action Items")
+        action_items = gr.Textbox(lines=4, label="Extracted Action Items", elem_id='actions')
+
+    def classify_and_render(email_text):
+        result = email_classifier_router(email_text)
+        cat = result.get("Category", "")
+        pri = result.get("Priority", "")
+        rec = result.get("Suggested Recipient", "")
+        dra = result.get("Draft Response", "")
+        summ = result.get("Summary", "")
+        acts = result.get("Action Items", [])
+        # Render action items as multi-line string if list
+        acts = "\n".join(acts) if isinstance(acts, list) else acts
+        return cat, pri, rec, dra, summ, acts
+
+    classify_btn.click(
+        classify_and_render,
+        inputs=[email_box],
+        outputs=[category, priority, recipient, draft_response, summary, action_items]
     )
 
 if __name__ == "__main__":