ajusta verificação de conteúdo

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import pipeline
+import re
 
 content_model = pipeline("image-classification", model="facebook/convnext-base-224")
 nsfw_model = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
@@ -12,15 +13,74 @@ def analyze_image(image):
     top_nsfw = max(nsfw_preds, key=lambda x: x["score"])
     
     label = top_content["label"].lower()
-    is_human = "human" in label or "person" in label or "people" in label
-    is_dog = "dog" in label or "retriever" in label or "puppy" in label
-
+    
+    human_keywords = [
+        "human", "person", "people", "man", "woman", "child", "baby", "boy", "girl",
+        "face", "portrait", "selfie", "crowd", "family", "couple", "teenager"
+    ]
+    is_human = any(keyword in label for keyword in human_keywords)
+    
+    dog_keywords = [
+        "dog", "puppy", "retriever", "labrador", "golden", "beagle", "bulldog",
+        "poodle", "german shepherd", "chihuahua", "terrier", "hound", "mastiff",
+        "canine", "pet", "animal"
+    ]
+    is_dog = any(keyword in label for keyword in dog_keywords)
+    
+    violence_keywords = [
+        "blood", "wound", "injury", "hurt", "pain", "fight", "violence", "weapon",
+        "knife", "gun", "attack", "aggression", "conflict", "battle", "war"
+    ]
+    
+    suffering_keywords = [
+        "sad", "crying", "tears", "depressed", "miserable", "suffering", "pain",
+        "distress", "anguish", "grief", "mourning", "funeral", "death", "dead",
+        "dying", "illness", "sick", "injured", "abandoned", "neglected"
+    ]
+    
+    abuse_keywords = [
+        "abuse", "mistreatment", "cruelty", "torture", "beaten", "chained",
+        "caged", "starving", "malnourished", "neglected", "abandoned"
+    ]
+    
     adult_content = top_nsfw["label"].lower() == "nsfw"
-    violence = "blood" in label or "wound" in label
-    sensitive = adult_content or violence
+    violence = any(keyword in label for keyword in violence_keywords)
+    suffering = any(keyword in label for keyword in suffering_keywords)
+    abuse = any(keyword in label for keyword in abuse_keywords)
+    
+    sensitive = adult_content or violence or suffering or abuse
+    
+    if is_human:
+        content_type = "human"
+    elif is_dog:
+        content_type = "dog"
+    else:
+        content_type = "unknown"
+    
+    description_parts = []
+    
+    if content_type == "human":
+        description_parts.append("Imagem contendo pessoa(s)")
+    elif content_type == "dog":
+        description_parts.append("Imagem contendo cão/cachorro")
+    else:
+        description_parts.append(f"Imagem contendo {label}")
+    
+    if adult_content:
+        description_parts.append("com conteúdo adulto")
+    if violence:
+        description_parts.append("com violência")
+    if suffering:
+        description_parts.append("mostrando sofrimento")
+    if abuse:
+        description_parts.append("com maus tratos")
+    
+    if sensitive:
+        description_parts.append("- CONTEÚDO SENSÍVEL")
+    else:
+        description_parts.append("- conteúdo seguro")
     
-    content_type = "human" if is_human else "dog" if is_dog else "unknown"
-    description = f"Imagem contendo {label}. Conteúdo {'seguro' if not sensitive else 'sensível'}."
+    description = " ".join(description_parts) + "."
 
     return {
         "content": content_type,