Update app.py

app.py

@@ -118,10 +118,19 @@ print("✅ All models loaded!")
 # Safety Filter Function
 # ----------------------
 def is_caption_safe(caption):
-    votes = moderation_model(caption)[0]
-    for item in votes:
-        if item["label"] in ["V", "V2"] and item["score"] > 0.5:  # Violence / Graphic Violence
-            return False
+    try:
+        votes = moderation_model(caption)
+        # If return_all_scores=True, it's [[{label, score}, ...]]
+        if isinstance(votes, list) and isinstance(votes[0], list):
+            votes = votes[0]
+        # Now safe to loop
+        for item in votes:
+            if isinstance(item, dict) and item.get("label") in ["V", "V2"] and item.get("score", 0) > 0.5:
+                return False
+    except Exception as e:
+        print("⚠️ Moderation failed:", e)
+
+    # Fallback keywords
     unsafe_keywords = ["gun", "blood", "skull", "kill", "corpse", "gore", "knife", "weapon"]
     if any(word in caption.lower() for word in unsafe_keywords):
         return False
@@ -129,6 +138,7 @@ def is_caption_safe(caption):
 
 
 
+
 # ----------------------
 # Caption + Translate + Speak
 # ----------------------