ajusta verificação de conteúdo

app.py

@@ -12,20 +12,27 @@ def analyze_image(image):
     nsfw_preds = nsfw_model(image)
     top_nsfw = max(nsfw_preds, key=lambda x: x["score"])
     
-    label = top_content["label"].lower()
+    all_labels = []
+    for pred in content_preds:
+        all_labels.append(pred["label"].lower())
+    
+    for pred in nsfw_preds:
+        all_labels.append(pred["label"].lower())
+    
+    combined_labels = " ".join(all_labels)
     
     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)
+    is_human = any(keyword in combined_labels for keyword in human_keywords)
     
     dog_keywords = [
         "dog", "puppy", "retriever", "labrador", "golden", "beagle", "bulldog",
         "poodle", "german shepherd", "chihuahua", "terrier", "hound", "mastiff",
-        "canine", "pet", "animal"
+        "canine", "pet", "animal", "malamute", "malemute", "alaskan"
     ]
-    is_dog = any(keyword in label for keyword in dog_keywords)
+    is_dog = any(keyword in combined_labels for keyword in dog_keywords)
     
     violence_keywords = [
         "blood", "wound", "injury", "hurt", "pain", "fight", "violence", "weapon",
@@ -43,12 +50,25 @@ def analyze_image(image):
         "caged", "starving", "malnourished", "neglected", "abandoned"
     ]
     
+    death_keywords = [
+        "death", "dead", "dying", "corpse", "carcass", "deceased", "lifeless",
+        "motionless", "still", "rigid", "pale", "cold"
+    ]
+    
+    suspicious_keywords = [
+        "lying", "laying", "ground", "floor", "side", "horizontal", "flat",
+        "unconscious", "sleeping", "resting", "still", "motionless", "quiet"
+    ]
+    
     adult_content = top_nsfw["label"].lower() == "nsfw"
-    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)
+    violence = any(keyword in combined_labels for keyword in violence_keywords)
+    suffering = any(keyword in combined_labels for keyword in suffering_keywords)
+    abuse = any(keyword in combined_labels for keyword in abuse_keywords)
+    death = any(keyword in combined_labels for keyword in death_keywords)
     
-    sensitive = adult_content or violence or suffering or abuse
+    suspicious_animal = is_dog and any(keyword in combined_labels for keyword in suspicious_keywords)
+    
+    sensitive = adult_content or violence or suffering or abuse or death or suspicious_animal
     
     if is_human:
         content_type = "human"
@@ -74,6 +94,10 @@ def analyze_image(image):
         description_parts.append("mostrando sofrimento")
     if abuse:
         description_parts.append("com maus tratos")
+    if death:
+        description_parts.append("mostrando morte")
+    if suspicious_animal:
+        description_parts.append("com características suspeitas")
     
     if sensitive:
         description_parts.append("- CONTEÚDO SENSÍVEL")
@@ -81,13 +105,15 @@ def analyze_image(image):
         description_parts.append("- conteúdo seguro")
     
     description = " ".join(description_parts) + "."
+    
+    debug_info = f" [Debug: Labels detectadas: {combined_labels}]"
 
     return {
         "content": content_type,
         "adult_content": adult_content,
         "violence": violence,
         "sensitive_content": sensitive,
-        "content_description": description
+        "content_description": description + debug_info
     }
 
 demo = gr.Interface(