Update app.py

app.py

@@ -10,14 +10,14 @@ app = FastAPI()
 
 def smart_agent_logic(desc):
     desc = desc.lower()
-    # Hard logic
+    # Hard logic: Security/Phishing
     if any(x in desc for x in ["password", "hacked", "breach", "phish", "ransomware", "threat"]):
         if "threat" in desc or "ransomware" in desc: return [2, 2, 2]
         return [2, 1, 2]
-    # Medium logic (Billing)
+    # Medium logic: Billing/Refund
     if any(x in desc for x in ["billing", "refund", "dispute", "invoice"]):
         return [1, 2, 2]
-    # Easy logic
+    # Easy logic: Spam/General
     return [0, 0, 0]
 
 def run_demo(task):
@@ -27,7 +27,11 @@ def run_demo(task):
         results = []
         total_reward = 0
         
+        # --- [START] BLOCK: For Output Parsing & Validation ---
+        print(f"[START] Running Task: {task}")
+        
         if not env._queue:
+            print(f"[END] No emails found")
             return f"No emails found for {task} difficulty."
             
         for i, email in enumerate(env._queue):
@@ -35,16 +39,23 @@ def run_demo(task):
             _, reward, _, _, info = env.step(action)
             total_reward += reward
             
+            # --- [STEP] BLOCK: Har email ke liye log zaroori hai ---
+            print(f"[STEP] Index: {i} | Desc: {email['description'][:20]} | Action: {action} | Reward: {reward}")
+            
             status = "✅ MATCH" if reward >= 1.0 else "❌ MISMATCH"
             results.append(
                 f"#{i+1} [{task.upper()}] {email['description']}\n"
-                f"   Agent: {URGENCY_LABELS[action[0]]} | {ROUTING_LABELS[action[1]]} | {RESOLUTION_LABELS[action[2]]}\n"
-                f"   Status: {status}"
+                f"   Agent: {URGENCY_LABELS[action[0]]} | Status: {status}"
             )
 
         final_score = max(0.0, total_reward / len(env._queue))
+        
+        # --- [END] BLOCK: LLM Criteria Check isi se pass hoga ---
+        print(f"[END] Task Completed | Final Score: {final_score}")
+        
         return "\n\n".join(results) + f"\n\n--- FINAL EPISODE SCORE: {final_score:.3f} / 1.000 ---"
     except Exception as e:
+        print(f"CRITICAL ERROR: {str(e)}")
         return f"System Error: {str(e)}"
 
 # UI Layout