Update app.py

app.py

@@ -10,13 +10,6 @@ import sys
 import logging
 import urllib3  # Import urllib3 to handle warnings
 
-# --- Patch asyncio for HuggingFace Spaces / Gradio conflicts ---
-try:
-    import nest_asyncio
-    nest_asyncio.apply()
-except ImportError:
-    print("⚠️ nest_asyncio not installed. It is recommended for running Asyncio apps in Spaces.")
-
 # --- Suppress specific noisy asyncio errors on shutdown ---
 if sys.version_info >= (3, 10):
     logging.getLogger("asyncio").setLevel(logging.WARNING)
@@ -414,11 +407,19 @@ def rag_predict_fn(text: str):
             else:
                 explanation = "" # Just label found
             
-            # Normalize label
-            if "PHISHING" in label_part:
-                label = "PHISHING"
-            elif "BENIGN" in label_part:
+            # Normalize label part
+            label_part = parts[0].replace("LABEL:", "").strip().upper()
+            
+            # Robust parsing logic
+            if "BENIGN" in label_part or "SAFE" in label_part or "LEGITIMATE" in label_part:
+                label = "BENIGN"
+            elif "NOT PHISHING" in label_part:
                 label = "BENIGN"
+            elif "PHISHING" in label_part or "MALICIOUS" in label_part:
+                label = "PHISHING"
+                
+            # Clean up explanation (remove artifacts)
+            explanation = explanation.replace("> EOF by user", "").strip()
         except:
             pass # Fallback to showing full response