Update routers/profanity.py

routers/profanity.py

@@ -1,6 +1,6 @@
 from fastapi import APIRouter, Query, HTTPException
 from fastapi.responses import StreamingResponse
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFont
 from io import BytesIO
 import requests
 from typing import Optional
@@ -31,33 +31,49 @@ def resize_and_crop_to_fill(img: Image.Image, target_width: int, target_height:
     right = left + target_width
     bottom = top + target_height
 
-    img_cropped = img_resized.crop((left, top, right, bottom))
-    return img_cropped
+    return img_resized.crop((left, top, right, bottom))
 
 def create_canvas(image_url: Optional[str] = None) -> BytesIO:
     width, height = 1080, 1350
     canvas = Image.new("RGB", (width, height), color=(255, 255, 255))
 
-    # Preencher com imagem se fornecida
+    # Preenchimento com imagem (caso enviada)
     if image_url:
         img = download_image_from_url(image_url)
-        img_filled = resize_and_crop_to_fill(img, width, height)
-        canvas.paste(img_filled, (0, 0))
+        filled_img = resize_and_crop_to_fill(img, width, height)
+        canvas.paste(filled_img, (0, 0))
 
-    # Desenhar header_bar
     draw = ImageDraw.Draw(canvas)
-    header_color = (255, 254, 255)
-    draw.rectangle([(0, 40), (1080, 40 + 70)], fill=header_color)
 
-    # Inserir logo CNN
+    # Header rosa claro
+    draw.rectangle([(0, 40), (1080, 110)], fill=(255, 254, 255))
+
+    # Logo CNN
     try:
         logo_url = "https://huggingface.co/spaces/habulaj/newapi/resolve/main/cnn.png"
         logo = download_image_from_url(logo_url)
         logo_resized = logo.resize((104, 104), Image.LANCZOS)
-        canvas.paste(logo_resized, (12, 21), logo_resized)  # Usa canal alpha como máscara
+        canvas.paste(logo_resized, (12, 21), logo_resized)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erro ao carregar logo: {e}")
 
+    # Texto "Pense bem, pense CNN."
+    try:
+        font_path = "fonts/Inter-Bold.otf"
+        font_size = 38
+        font = ImageFont.truetype(font_path, font_size)
+        text = "PENSE BEM, PENSE CNN."
+
+        x = 138
+        y = 52
+        letter_spacing = 1  # 1% tracking aproximado
+
+        for char in text:
+            draw.text((x, y), char, font=font, fill=(0, 0, 0))
+            x += draw.textlength(char, font=font) + letter_spacing
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Erro ao carregar fonte: {e}")
+
     # Exportar imagem final
     buffer = BytesIO()
     canvas.save(buffer, format="PNG")