Update app.py

app.py

@@ -2,111 +2,63 @@ import sys
 import os
 import uvicorn
 import numpy as np
-from fastapi import FastAPI
 import gradio as gr
-
-# --- 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"]
+from fastapi import FastAPI
 
 app = FastAPI()
 
-# --- Full Hackathon Dataset ---
+# Shared 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)},
-    {"difficulty": "hard", "description": "IT password reset phish", "keywords": ["password", "urgent"], "sentiment": "negative", "context": "security", "correct_actions": (2, 1, 2)},
-    {"difficulty": "hard", "description": "Ransomware threat", "keywords": ["hacked", "legal", "threat"], "sentiment": "negative", "context": "security", "correct_actions": (2, 2, 2)},
-    {"difficulty": "hard", "description": "Fake GDPR notice", "keywords": ["breach", "legal"], "sentiment": "negative", "context": "security", "correct_actions": (2, 1, 2)},
-    {"difficulty": "hard", "description": "Law firm misuse letter", "keywords": ["unauthorized", "breach", "legal"], "sentiment": "negative", "context": "legal", "correct_actions": (2, 2, 2)},
+    {"difficulty": "easy", "description": "Spam promo", "correct_actions": (0, 0, 0)},
+    {"difficulty": "easy", "description": "Routine support", "correct_actions": (0, 1, 1)},
+    {"difficulty": "hard", "description": "IT password reset phish", "correct_actions": (2, 1, 2)},
+    {"difficulty": "hard", "description": "Ransomware threat", "correct_actions": (2, 2, 2)},
+    {"difficulty": "hard", "description": "Fake GDPR notice", "correct_actions": (2, 1, 2)},
 ]
 
-def _classify_with_llm(email: dict) -> np.ndarray:
-    """Agent Logic to secure 1.000 score"""
-    desc = email.get('description', '').lower()
-    kws = [k.lower() for k in email.get('keywords', [])]
-    
-    # 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):
-        # 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 or "attorney" in desc:
-        return np.array([2, 2, 2])
+# Labels from env
+URGENCY_LABELS = ["General", "Billing", "Security Breach"]
+ROUTING_LABELS = ["AI Auto-Reply", "Tech Support", "Legal"]
+RESOLUTION_LABELS = ["Archive", "Draft Reply", "Escalate to Human"]
 
-    # Check for 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])
+from env import EmailTriageEnv
 
-    # Default
-    return np.array([0, 0, 0])
+def classify_logic(desc):
+    desc = desc.lower()
+    if any(x in desc for x in ["password", "hacked", "breach", "phish"]):
+        if "threat" in desc or "ransom" in desc: return [2, 2, 2]
+        return [2, 1, 2]
+    return [0, 0, 0]
 
-def run_task_demo(task: str) -> str:
+def run_demo(task):
     try:
-        # Env initialization
-        env = EmailTriageEnv(task=task, shuffle=False)
+        env = EmailTriageEnv(task=task)
         env.reset()
+        results = []
+        total_reward = 0
         
-        # Accessing the queue fixed in env.py
-        email_queue = list(env._queue)
-        lines = []
-        cumulative_score = 0.0
-        
-        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"
+        for i, email in enumerate(env._queue):
+            action = classify_logic(email['description'])
+            _, reward, _, _, info = env.step(action)
+            total_reward += reward
             
-            # 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: {u_lab} | {ro_lab} | {re_lab}\n"
-                         f"   🏆 Status: {status}\n" + "-"*40)
+            status = "✅ MATCH" if reward >= 1.0 else "❌ MISMATCH"
+            results.append(f"#{i+1} {email['description'][:30]}...\n   Agent: {URGENCY_LABELS[action[0]]} | Status: {status}")
 
-        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)
+        score = max(0, total_reward / len(env._queue)) if env._queue else 0
+        return "\n\n".join(results) + f"\n\nFINAL SCORE: {score:.3f}"
     except Exception as e:
-        return f"System Error: {str(e)}"
+        return f"Error: {str(e)}"
 
-# Gradio Dashboard
-with gr.Blocks(title="Email Gatekeeper") as demo:
+with gr.Blocks() 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...")
+    diff = gr.Dropdown(["easy", "hard"], value="easy", label="Difficulty")
+    btn = gr.Button("Analyze Emails")
+    out = gr.Textbox(label="Logs", lines=15)
     
-    run_btn.click(fn=run_task_demo, inputs=task_dropdown, outputs=output_box)
+    btn.click(run_demo, inputs=diff, outputs=out)
 
 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)