Update server/app.py

server/app.py

@@ -2,17 +2,26 @@ import sys
 import os
 import uvicorn
 import numpy as np
-import re
 from fastapi import FastAPI
 import gradio as gr
 
-# Setup path and imports
-sys.path.append(os.path.dirname(os.path.abspath(__file__)))
-from env import EmailTriageEnv, URGENCY_LABELS, ROUTING_LABELS, RESOLUTION_LABELS
+# --- PATH FIX: Sabse pehle current directory ko path mein add karte hain ---
+current_dir = os.path.dirname(os.path.abspath(__file__))
+if current_dir not in sys.path:
+    sys.path.append(current_dir)
+
+# Ab import kaam karega
+try:
+    from env import EmailTriageEnv, URGENCY_LABELS, ROUTING_LABELS, RESOLUTION_LABELS
+except ImportError:
+    # Fallback labels agar env.py na mile (Error se bachne ke liye)
+    URGENCY_LABELS = ["General", "Billing", "Security Breach"]
+    ROUTING_LABELS = ["AI Auto-Reply", "Tech Support", "Legal"]
+    RESOLUTION_LABELS = ["Archive", "Draft Reply", "Escalate to Human"]
 
 app = FastAPI()
 
-# --- Shared Dataset ---
+# --- Full Hackathon Dataset ---
 EMAIL_DATASET = [
     {"difficulty": "easy", "description": "Spam promo", "keywords": ["free", "offer"], "sentiment": "positive", "context": "spam", "correct_actions": (0, 0, 0)},
     {"difficulty": "easy", "description": "Routine support", "keywords": ["slow", "error"], "sentiment": "neutral", "context": "tech", "correct_actions": (0, 1, 1)},
@@ -27,30 +36,34 @@ def _classify_with_llm(email: dict) -> np.ndarray:
     desc = email.get('description', '').lower()
     kws = [k.lower() for k in email.get('keywords', [])]
     
-    # Check for Security Threats (Highest Reward/Risk)
-    sec_triggers = ["password", "hacked", "breach", "unauthorized", "urgent", "security", "credential"]
+    # Check for Security Threats
+    sec_triggers = ["password", "hacked", "breach", "unauthorized", "urgent", "security", "credential", "phish"]
     if any(t in desc for t in sec_triggers) or any(k in sec_triggers for k in kws):
-        # Legal + Security = Legal Routing
-        if any(l in desc for l in ["legal", "lawsuit", "attorney", "threat", "audit"]):
+        # Security + Legal/Threat
+        if any(l in desc for l in ["legal", "lawsuit", "attorney", "threat", "audit", "court"]):
             return np.array([2, 2, 2])
         return np.array([2, 1, 2]) # Security + Tech
 
     # Check for Legal
-    if "legal" in desc or "lawsuit" in desc:
+    if "legal" in desc or "lawsuit" in desc or "attorney" in desc:
         return np.array([2, 2, 2])
 
     # Check for Billing
-    if any(b in desc for b in ["invoice", "payment", "refund", "billing"]):
+    if any(b in desc for b in ["invoice", "payment", "refund", "billing", "overdue"]):
         if "dispute" in desc or "refund" in desc:
             return np.array([1, 2, 2])
         return np.array([1, 0, 1])
 
+    # Default
     return np.array([0, 0, 0])
 
 def run_task_demo(task: str) -> str:
     try:
+        # Env initialization
         env = EmailTriageEnv(task=task, shuffle=False)
         env.reset()
+        
+        # Accessing the queue fixed in env.py
         email_queue = list(env._queue)
         lines = []
         cumulative_score = 0.0
@@ -58,29 +71,42 @@ def run_task_demo(task: str) -> str:
         for i, email in enumerate(email_queue):
             action = _classify_with_llm(email)
             _, _, _, _, info = env.step(action)
+            
             reward = info.get("raw_reward", 0)
             cumulative_score += reward
             
             status = "✅ EXACT MATCH (+1.0)" if reward >= 0.9 else "❌ MISMATCH"
+            
+            # Action labels safety check
+            u_lab = URGENCY_LABELS[action[0]]
+            ro_lab = ROUTING_LABELS[action[1]]
+            re_lab = RESOLUTION_LABELS[action[2]]
+            
             lines.append(f"#{i+1:02d} [{task.upper()}] {email['description'][:40]}...\n"
-                         f"   ▶ Agent: {URGENCY_LABELS[action[0]]} | {ROUTING_LABELS[action[1]]} | {RESOLUTION_LABELS[action[2]]}\n"
+                         f"   ▶ Agent: {u_lab} | {ro_lab} | {re_lab}\n"
                          f"   🏆 Status: {status}\n" + "-"*40)
 
-        final = max(0.0, min(1.0, cumulative_score / len(email_queue)))
+        final = max(0.0, min(1.0, cumulative_score / len(email_queue))) if email_queue else 0.0
         lines.append(f"\nTOTAL EPISODE SCORE: {final:.3f} / 1.000")
         return "\n".join(lines)
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"System Error: {str(e)}"
 
-# UI Layout
-with gr.Blocks() as demo:
-    gr.Markdown("# 📧 Email Gatekeeper")
-    task_dropdown = gr.Dropdown(choices=["easy", "medium", "hard"], value="easy", label="Select Difficulty")
-    run_btn = gr.Button("Analyze Emails")
-    output_box = gr.Textbox(lines=20, label="Reward Breakdown")
+# Gradio Dashboard
+with gr.Blocks(title="Email Gatekeeper") as demo:
+    gr.Markdown("# 📧 Email Gatekeeper AI")
+    gr.Markdown("Select difficulty and click Analyze to evaluate the Agent.")
+    
+    with gr.Row():
+        task_dropdown = gr.Dropdown(choices=["easy", "medium", "hard"], value="easy", label="Select Difficulty")
+        run_btn = gr.Button("Analyze Emails", variant="primary")
+        
+    output_box = gr.Textbox(lines=20, label="Evaluation Logs", placeholder="Results will appear here...")
+    
     run_btn.click(fn=run_task_demo, inputs=task_dropdown, outputs=output_box)
 
 app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":
+    # Port 7860 is mandatory for Hugging Face Spaces
     uvicorn.run(app, host="0.0.0.0", port=7860)