Update app/main.py

app/main.py

@@ -101,12 +101,25 @@ async def moderate_text(request: TextRequest):
 
     profanity_result = profanity_filter.check_profanity(text)
     # tox_result = toxicity_pipe(text)[0]
-    raw_result = toxicity_pipe(text)
-    tox_result = raw_result[0][0] if isinstance(raw_result[0], list) else raw_result[0]
+    # raw_result = toxicity_pipe(text)
+    # tox_result = raw_result[0][0] if isinstance(raw_result[0], list) else raw_result[0]
 
-    is_toxic_by_model = (
-        tox_result["label"].lower() in ["toxic", "label_1"] and tox_result["score"] > 0.5
+    # is_toxic_by_model = (
+    #     tox_result["label"].lower() in ["toxic", "label_1"] and tox_result["score"] > 0.5
+    # )
+    raw_result = toxicity_pipe(text)
+    # Handle both nested list [[{...}]] and flat list [{...}] outputs safely    
+    if isinstance(raw_result, list) and len(raw_result) > 0:
+        tox_result = raw_result[0][0] if isinstance(raw_result[0], list) else raw_result[0]    
+    else:       
+        tox_result = {"label": "neutral", "score": 0.0}
+        
+    is_toxic_by_model = (        
+        isinstance(tox_result, dict)        
+        and tox_result.get("label", "").lower() in ["toxic", "label_1"]
+        and tox_result.get("score", 0.0) > 0.5    
     )
+    
     is_threat_by_keyword = bool(threat_pattern.search(text))
 
     is_flagged = bool(