Update app.py

app.py

@@ -2,38 +2,11 @@ import os
 import gradio as gr
 import requests
 import json
-from datetime import datetime
 
 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"
 
-# Dummy sender risk DB
-SENDER_REPUTATION = {
-    "it@company.com": "Trusted",
-    "accounts@supplier.com": "Trusted",
-    "spammer@promo.com": "Suspicious",
-    "unknown@phish.com": "Flagged"
-}
-
-# In-memory structures
-classification_history = []
-analytics_counter = {}
-
-default_categories = ["Support Request", "Sales Inquiry", "Finance/Billing", "Urgent Incident", "Spam/Marketing"]
-
-default_templates = [
-    "Thank you for your email. We are looking into your request and will reply shortly.",
-    "We have received your billing inquiry and will review your account information.",
-    "Your support request has been escalated to our technical team."
-]
-
-def get_sender(email_txt):
-    for line in email_txt.splitlines():
-        if line.lower().startswith("from:"):
-            return line.split(":",1)[-1].strip()
-    return ""
-
 def groq_completion(prompt, sys_prompt=None):
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
@@ -42,7 +15,7 @@ def groq_completion(prompt, sys_prompt=None):
     body = {
         "model": GROQ_MODEL,
         "messages": [
-            {"role": "system", "content": sys_prompt or "You are a helpful business email assistant."},
+            {"role": "system", "content": sys_prompt or "You are a fast and reliable business email assistant."},
             {"role": "user", "content": prompt}
         ],
         "temperature": 0.3,
@@ -58,21 +31,19 @@ def groq_completion(prompt, sys_prompt=None):
             print("Groq API response:", response.text)
         return "Error"
 
-def email_classifier_router(email_txt, categories, templates):
-    sender = get_sender(email_txt)
-    sender_score = SENDER_REPUTATION.get(sender,"Unknown")
-    category_prompt = f"Categories (choose one or suggest new): {', '.join(categories)}\n"
-    template_list = "\n".join([f"- {t}" for t in templates])
+def email_classifier_router(raw_email):
     prompt = (
-        f"{category_prompt}"
-        "Extract:\n"
-        "- Category\n- Priority\n- Suggested Recipient\n- Professional draft response\n"
-        "- Summary (one sentence)\n- Action Items (numbered list)\n- Entities: invoice numbers, dates, customer names if present.\n"
-        "Format as single compact JSON dictionary.\n"
-        '{"Category":"...", "Priority":"...", "Suggested Recipient":"...", "Draft Response":"...", "Summary":"...", "Action Items":["..."], "Entities":{"Invoice":"","Date":"","Customer":""}}\n\n'
-        f"{email_txt}\n"
-        "If reply, you may use these reply templates as inspiration:\n"
-        f"{template_list}"
+        "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:
@@ -85,117 +56,29 @@ def email_classifier_router(email_txt, categories, templates):
             "Suggested Recipient": "Error",
             "Draft Response": raw_result,
             "Summary": "Error",
-            "Action Items": ["Error"],
-            "Entities": {"Invoice":"", "Date":"", "Customer":""}
+            "Action Items": ["Error"]
         }
-    # Store in history & update analytics
-    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    entry = {
-        "datetime": now,
-        "email": email_txt,
-        "category": output.get("Category",""),
-        "priority": output.get("Priority",""),
-        "summary": output.get("Summary",""),
-        "draft": output.get("Draft Response",""),
-        "recipient": output.get("Suggested Recipient",""),
-        "entities": output.get("Entities",{}),
-        "actions": output.get("Action Items",[]),
-        "sender": sender,
-        "risk": sender_score
-    }
-    classification_history.append(entry)
-    analytics_counter[output.get("Category","Other")] = analytics_counter.get(output.get("Category","Other"),0)+1
-    return entry
-
-def show_history():
-    res = []
-    for entry in reversed(classification_history):
-        block = gr.Group()
-        summary = entry['summary']
-        txt = f"**Category:** {entry['category']}  \n**Priority:** {entry['priority']}  \n{summary}"
-        btns = gr.Row()
-        block.append(gr.Markdown(txt))
-        block.append(btns)
-        res.append(block)
-    return res
-
-def copy_email(entry_idx):
-    if 0 <= entry_idx < len(classification_history):
-        return classification_history[entry_idx]['email']
-    return ""
-
-def load_email(entry_idx):
-    if 0 <= entry_idx < len(classification_history):
-        return classification_history[entry_idx]['email']
-    return ""
-
-def clear_history():
-    classification_history.clear()
-    analytics_counter.clear()
-    return gr.update(visible=True)
+    return output
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-    # Editable categories and template management
-    with gr.Row():
-        gr.Markdown("📝 **Manage Categories and Reply Templates**")
-    categories_box = gr.Textbox(label="Categories (comma separated)", value=", ".join(default_categories))
-    templates_box = gr.Textbox(label="Reply Templates (one per line)", value="\n".join(default_templates), lines=3)
-
-    gr.Markdown("## 📨 Classify & Reply to Emails")
-    email_box = gr.Textbox(lines=8, label="Paste the entire raw email")
-    classify_btn = gr.Button("Classify Email", variant="primary")
-    
-    results = gr.JSON(label="Classification + Extraction Results")
-    sender_risk = gr.Textbox(label="Sender Risk Score", interactive=False)
-    entities = gr.JSON(label="Extracted Entities")
+    gr.Markdown("## 📨 AI Email Classifier & Router\nEasily classify, route, and summarize business emails.")
+
+    with gr.Row(equal_height=True):
+        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!"
+        support_button = gr.Button("Support Request")
+        sales_button = gr.Button("Sales Inquiry")
+        finance_button = gr.Button("Finance / Billing")
+        urgent_button = gr.Button("Urgent Incident")
+        spam_button = gr.Button("Spam / Marketing")
     
-    # Template suggester
-    template_select = gr.Dropdown(choices=default_templates, label="Insert Reply Template")
-    insert_btn = gr.Button("Insert Template into Draft")
-
-    # History display
-    gr.Markdown("### 🕒 Classification History")
-    history_output = gr.Markdown()
-    clear_btn = gr.Button("Clear History")
-
-    # Basic Analytics
-    analytics = gr.Markdown(label="Simple Analytics")
-
-    def classify(email, cats, tmps):
-        cat_list = [c.strip() for c in cats.split(",")]
-        tmpl_list = [t.strip() for t in tmps.split("\n") if t.strip()]
-        entry = email_classifier_router(email, cat_list, tmpl_list)
-        return (entry, entry["risk"], entry["entities"], render_history(classification_history), render_analytics(analytics_counter))
-
-    def render_history(history):
-        # Render HTML for cards
-        res = ""
-        for idx, h in enumerate(reversed(history)):
-            summary = h['summary']
-            res += f"""<div style="border:1px solid #333;margin:5px;padding:8px;">
-              <b>{h['datetime']}</b> <br>
-              <b>Category:</b> <span style='background:#448; color:white; border-radius:4px; padding:2px 8px;'>{h['category']}</span>
-              <b>Priority:</b> <span style='background:#aa4; color:black; border-radius:4px; padding:2px 8px;'>{h['priority']}</span>
-              <br>{summary[:120]}
-              <br>
-              <button onclick="navigator.clipboard.writeText({json.dumps(h['draft'])})">Copy Email</button>
-              <button onclick="navigator.clipboard.writeText({json.dumps(h['email'])})">Load</button>
-            </div>"""
-        return res
-
-    def render_analytics(counter):
-        res = "**Category Counts:**<br>"
-        for k, v in counter.items():
-            res += f"{k}: <b>{v}</b><br>"
-        return res
-
-    classify_btn.click(
-        classify,
-        inputs=[email_box, categories_box, templates_box],
-        outputs=[results, sender_risk, entities, history_output, analytics]
-    )
-    clear_btn.click(clear_history, None, history_output)
-    insert_btn.click(lambda t, d: d + "\n" + t, [template_select, email_box], email_box)
+    email_box = gr.Textbox(lines=8, label="Paste the entire raw email (including lines like 'From:', 'Subject:', etc.)", elem_id='email_box')
+    classify_btn = gr.Button("Classify Email", elem_id='classify_btn', variant="primary")
 
-if __name__ == "__main__":
-    demo.launch()
+    with gr.Column():
+        gr.Markdown("### ✅ Classification Results")
+        category = gr.Textbox(label="Category", elem_id='category