Update routers/cover.py

routers/cover.py

@@ -1,6 +1,6 @@
 from fastapi import APIRouter, Query, HTTPException
 from fastapi.responses import StreamingResponse
-from PIL import Image
+from PIL import Image, ImageDraw
 from io import BytesIO
 import requests
 
@@ -52,12 +52,42 @@ def resize_and_crop_to_fill(img: Image.Image, target_width: int, target_height:
     
     return img_resized.crop((left, top, right, bottom))
 
+def add_rounded_corners(img: Image.Image, radius: int) -> Image.Image:
+    """
+    Adiciona cantos arredondados à imagem.
+    
+    Args:
+        img: Imagem PIL
+        radius: Raio do arredondamento em pixels
+    
+    Returns:
+        Imagem com cantos arredondados
+    """
+    # Criar máscara para os cantos arredondados
+    mask = Image.new("L", img.size, 0)
+    draw = ImageDraw.Draw(mask)
+    
+    # Desenhar retângulo com cantos arredondados
+    draw.rounded_rectangle(
+        [(0, 0), (img.width, img.height)],
+        radius=radius,
+        fill=255
+    )
+    
+    # Aplicar máscara à imagem
+    output = Image.new("RGBA", img.size, (0, 0, 0, 0))
+    output.paste(img, (0, 0))
+    output.putalpha(mask)
+    
+    return output
+
 def create_cover_image(image_url: str) -> BytesIO:
     """
     Cria uma capa para reels do Instagram.
     - Fundo preto
     - Largura: 1080, Altura: 1920
     - Imagem renderizada em Largura: 1080, Altura: 1440, X: 0, Y: 240
+    - Cantos arredondados com raio de 145px
     """
     # Dimensões do canvas
     canvas_width = 1080
@@ -75,8 +105,11 @@ def create_cover_image(image_url: str) -> BytesIO:
     img_height = 1440
     filled_img = resize_and_crop_to_fill(img, img_width, img_height)
     
+    # Aplicar arredondamento de 145px nos cantos
+    rounded_img = add_rounded_corners(filled_img, 145)
+    
     # Colar imagem na posição X: 0, Y: 240
-    canvas.paste(filled_img, (0, 240), filled_img)
+    canvas.paste(rounded_img, (0, 240), rounded_img)
     
     # Converter para bytes
     buffer = BytesIO()