Update app.py

app.py

@@ -2,11 +2,38 @@ 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}",
@@ -15,7 +42,7 @@ def groq_completion(prompt, sys_prompt=None):
     body = {
         "model": GROQ_MODEL,
         "messages": [
-            {"role": "system", "content": sys_prompt or "You are a fast and reliable business email assistant."},
+            {"role": "system", "content": sys_prompt or "You are a helpful business email assistant."},
             {"role": "user", "content": prompt}
         ],
         "temperature": 0.3,
@@ -31,19 +58,21 @@ def groq_completion(prompt, sys_prompt=None):
             print("Groq API response:", response.text)
         return "Error"
 
-def email_classifier_router(raw_email):
+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])
     prompt = (
-        "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}"
+        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}"
     )
     raw_result = groq_completion(prompt)
     try:
@@ -56,69 +85,117 @@ def email_classifier_router(raw_email):
             "Suggested Recipient": "Error",
             "Draft Response": raw_result,
             "Summary": "Error",
-            "Action Items": ["Error"]
+            "Action Items": ["Error"],
+            "Entities": {"Invoice":"", "Date":"", "Customer":""}
         }
-    return output
+    # 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)
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-    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")
+    # 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")
     
-    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")
-
-    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')
-
-    support_button.click(lambda: ex1, None, email_box)
-    sales_button.click(lambda: ex2, None, email_box)
-    finance_button.click(lambda: ex3, None, email_box)
-    urgent_button.click(lambda: ex4, None, email_box)
-    spam_button.click(lambda: ex5, None, email_box)
-
-    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", [])
-        acts = "\n".join(acts) if isinstance(acts, list) else acts
-        return cat, pri, rec, dra, summ, acts
+    results = gr.JSON(label="Classification + Extraction Results")
+    sender_risk = gr.Textbox(label="Sender Risk Score", interactive=False)
+    entities = gr.JSON(label="Extracted Entities")
+    
+    # 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_and_render,
-        inputs=[email_box],
-        outputs=[category, priority, recipient, draft_response, summary, action_items]
+        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)
 
 if __name__ == "__main__":
     demo.launch()