ajusta moderador para aceitar base64

app.py

@@ -5,6 +5,8 @@ import torch
 import timm
 from PIL import Image
 import numpy as np
+import base64
+from io import BytesIO
 
 try:
     eva02_model = timm.create_model('hf_hub:SmilingWolf/wd-eva02-large-tagger-v3', pretrained=True)
@@ -24,6 +26,18 @@ except Exception as e:
 content_model = pipeline("image-classification", model="facebook/convnext-base-224")
 nsfw_model = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
 
+def decode_base64_image(base64_string):
+    if ',' in base64_string:
+        base64_string = base64_string.split(',')[1]
+    
+    try:
+        image_data = base64.b64decode(base64_string)
+        image = Image.open(BytesIO(image_data))
+        return image
+    except Exception as e:
+        print(f"Erro ao decodificar base64: {e}")
+        return None
+
 def analyze_with_eva02(image):
     if eva02_model is None:
         return [], []
@@ -50,7 +64,33 @@ def analyze_with_eva02(image):
     
     return detected_tags, tag_scores
 
-def analyze_image(image):
+def analyze_image(image_input, base64_input=None):
+    image = None
+    
+    if base64_input and isinstance(base64_input, str) and base64_input.strip():
+        image = decode_base64_image(base64_input)
+        if image is None:
+            return {
+                "content": "unknown",
+                "adult_content": False,
+                "violence": False,
+                "sensitive_content": False,
+                "content_description": "Erro ao processar imagem base64."
+            }
+    elif image_input is not None:
+        image = image_input
+    else:
+        return {
+            "content": "unknown",
+            "adult_content": False,
+            "violence": False,
+            "sensitive_content": False,
+            "content_description": "Nenhuma imagem fornecida."
+        }
+    
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    
     if eva02_model is not None:
         eva02_tags_detected, eva02_scores = analyze_with_eva02(image)
         combined_labels = " ".join(eva02_tags_detected).lower()
@@ -131,14 +171,12 @@ def analyze_image(image):
         "grave", "tombstone", "funeral", "coffin", "burial"
     ]
     
-    # Palavras-chave para comportamento SUSPEITO (não normal)
     suspicious_keywords = [
         "unconscious", "motionless", "lifeless", "rigid", "cold", "pale",
         "injured", "wounded", "bleeding", "hurt", "pain", "distress",
         "abandoned", "neglected", "starving", "malnourished", "chained", "caged"
     ]
     
-    # Palavras-chave para comportamento NORMAL de cachorros (não suspeito)
     normal_dog_behavior = [
         "sleeping", "resting", "lying", "sitting", "playing", "running", "walking",
         "happy", "excited", "alert", "awake", "active", "energetic", "playful"
@@ -155,14 +193,10 @@ def analyze_image(image):
     abuse = any(keyword in combined_labels for keyword in abuse_keywords)
     death = any(keyword in combined_labels for keyword in death_keywords)
     
-    # Detecção mais inteligente para cachorros
     has_suspicious_behavior = any(keyword in combined_labels for keyword in suspicious_keywords)
     has_normal_behavior = any(keyword in combined_labels for keyword in normal_dog_behavior)
     has_death_indicators = any(keyword in combined_labels for keyword in death_keywords)
-    
-    # Se é um cachorro, só marcar como suspeito se:
-    # 1. Tem indicadores de morte OU
-    # 2. Tem comportamento suspeito E NÃO tem comportamento normal
+
     if is_dog:
         suspicious_animal = has_death_indicators or (has_suspicious_behavior and not has_normal_behavior)
         dead_dog = has_death_indicators
@@ -228,9 +262,13 @@ def analyze_image(image):
 
 demo = gr.Interface(
     fn=analyze_image,
-    inputs=gr.Image(type="pil"),
+    inputs=[
+        gr.Image(type="pil", label="Upload de Imagem ou Cole Base64"),
+        gr.Textbox(label="Ou Cole String Base64 Aqui", lines=3, placeholder="data:image/jpeg;base64,/9j/4AAQ...")
+    ],
     outputs="json",
-    title="Dog / Human Safety Detector"
+    title="Dog / Human Safety Detector",
+    description="Envie uma imagem ou cole uma string base64 para análise de moderação"
 )
 
 if __name__ == "__main__":