Update app.py

app.py

@@ -1,9 +1,13 @@
+# email_classifier_v2025.10.29-01.py
+
 import os
 import gradio as gr
 import requests
 import json
 from datetime import datetime
 
+__version__ = "v2025.10.29-01"
+
 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"
@@ -28,12 +32,13 @@ def groq_completion(prompt, sys_prompt=None):
         response = requests.post(GROQ_COMPLETION_URL, headers=headers, json=body)
         response.raise_for_status()
         return response.json()["choices"][0]["message"]["content"]
-    except Exception as e:
-        return "Error"
+    except Exception:
+        return "ERROR:model_request"
 
 def email_classifier_router(raw_email):
     prompt = (
-        "Return ONLY compact JSON for analysis and response. Given the business email below, extract:"
+        "Your reply MUST BE valid compact JSON. NO TEXT OR EXPLANATION before or after the JSON. "
+        "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"
@@ -44,16 +49,29 @@ def email_classifier_router(raw_email):
         f"{raw_email}"
     )
     raw_result = groq_completion(prompt)
+    if raw_result == "ERROR:model_request":
+        return {
+            "Category": "ERROR",
+            "Priority": "ERROR",
+            "Suggested Recipient": "ERROR",
+            "Draft Response": "Model request failed, please check API or retry.",
+            "Summary": "ERROR",
+            "Action Items": ["ERROR"]
+        }
     try:
         output = json.loads(raw_result)
-    except Exception as e:
+        # Ensure keys present, fallback if missing
+        for key in ["Category", "Priority", "Suggested Recipient", "Draft Response", "Summary", "Action Items"]:
+            if key not in output:
+                output[key] = ""
+    except Exception:
         output = {
-            "Category": "Error",
-            "Priority": "Error",
-            "Suggested Recipient": "Error",
-            "Draft Response": "",
-            "Summary": "Error",
-            "Action Items": ["Error"]
+            "Category": "PARSE ERROR",
+            "Priority": "PARSE ERROR",
+            "Suggested Recipient": "",
+            "Draft Response": f"Could not extract valid JSON. Raw LLM output:\n{raw_result}",
+            "Summary": "PARSE ERROR",
+            "Action Items": ["PARSE ERROR"]
         }
     return output
 
@@ -71,6 +89,8 @@ def add_to_history(email, cat, pri, summ, dra):
 css="""
 body { background: linear-gradient(120deg, #10193a 0%, #175ad7 120%) !important; font-family: 'Inter', 'Segoe UI', Arial, sans-serif;}
 .gradio-container { background: transparent !important; }
+.artifact-header { background:linear-gradient(90deg,#283e9d 70%,#36d1fa 150%); border-radius:28px; padding:38px 78px; box-shadow:0 10px 52px #184fa988; margin:0 auto 38px auto; display:flex;align-items:center;justify-content:center;}
+.artifact-header-text {font-size:3.1em;font-weight:900;color:#fff;text-shadow:0 4px 44px #09f6ef,0 2px 20px #2e96ff;}
 .card-block { background: rgba(27,36,68,0.96); border-radius: 30px; box-shadow: 0 6px 42px #0b2269bb, 0 1.5px 0 #186fc055 inset; padding: 38px 28px 29px 28px; margin: 32px 0; color: #fafeff; border: 2px solid #2847a46c; backdrop-filter: blur(8px);}
 input, textarea, .wrap.svelte-psy7la, .wrap.svelte-1ybjah6 { background: rgba(21,28,56,0.91)!important; border:1.5px solid #345ca3e3!important; color:#def1ff!important; border-radius: 14px !important; box-shadow: none !important;}
 label, .label.svelte-1ybjah6 { color: #70acf8!important; font-size: 1em !important; font-weight: 400;}
@@ -92,7 +112,13 @@ label, .label.svelte-1ybjah6 { color: #70acf8!important; font-size: 1em !importa
 """
 
 with gr.Blocks(css=css) as demo:
-    gr.Markdown("<div class='card-header' style='text-align:center;margin-bottom:-10px;'>AI Email Classifier & Router</div>")
+    gr.HTML("""
+    <div class="artifact-header">
+      <span class="artifact-header-text">📧 AI Email Classifier & Router</span>
+      <span style='margin-left:30px; font-size:1.5em; color:#fff;opacity:.50;'>Artifact <b>{ver}</b></span>
+    </div>
+    """.format(ver=__version__))
+
     with gr.Row():
         with gr.Column(scale=6, min_width=420):
             with gr.Group(elem_id="input-card", elem_classes=["card-block"]):
@@ -153,8 +179,9 @@ with gr.Blocks(css=css) as demo:
         acts = result.get("Action Items", [])
         acts = "\n".join(acts) if isinstance(acts, list) else acts
         add_to_history(email_text, cat, pri, summ, dra)
-        badge = f"<b>Category:</b> <span class='badge'>{cat}</span> <b>Priority:</b> <span class='badge priority-{pri.lower()}'>{pri}</span>"
-        result_labels_text = f"<div class='history-top'><span class='badge'>{cat}</span><span class='badge priority-{pri.lower()}'>{pri}</span></div>"
+        badge = (
+            f"<div class='history-top'><span class='badge'>{cat}</span><span class='badge priority-{pri.lower()}'>{pri}</span></div>"
+        )
         # Only show last 5 records
         recent_history = classification_history[-5:][::-1]
         history_html = "<div class='history-cardrow'>"
@@ -174,7 +201,7 @@ with gr.Blocks(css=css) as demo:
             </div>
             """
         history_html += "</div>"
-        return result_labels_text, rec, dra, summ, acts, history_html
+        return badge, rec, dra, summ, acts, history_html
 
     def clear_history():
         classification_history.clear()