Update routers/news.py

routers/news.py

@@ -3,7 +3,7 @@ from fastapi.responses import StreamingResponse
 from PIL import Image, ImageDraw, ImageFont
 from io import BytesIO
 import requests
-from typing import Optional
+from typing import Optional, List, Union
 
 router = APIRouter()
 
@@ -42,6 +42,184 @@ def resize_and_crop_to_fill(img: Image.Image, target_width: int, target_height:
 
     return img_resized.crop((left, top, right, bottom))
 
+def create_collage_background(image_urls: List[str], canvas_width: int, canvas_height: int, device: str) -> Image.Image:
+    """Cria uma colagem como fundo baseada na lista de URLs"""
+    num_images = len(image_urls)
+    border_size = 12 if num_images > 1 else 0
+    is_web = device.lower() == "web"
+
+    images = [download_image_from_url(url) for url in image_urls]
+    canvas = Image.new("RGBA", (canvas_width, canvas_height), (255, 255, 255, 255))
+
+    if num_images == 1:
+        img = resize_and_crop_to_fill(images[0], canvas_width, canvas_height)
+        canvas.paste(img, (0, 0))
+
+    elif num_images == 2:
+        # Ambos dispositivos: lado a lado
+        slot_width = (canvas_width - border_size) // 2
+        img1 = resize_and_crop_to_fill(images[0], slot_width, canvas_height)
+        img2 = resize_and_crop_to_fill(images[1], slot_width, canvas_height)
+        canvas.paste(img1, (0, 0))
+        canvas.paste(img2, (slot_width + border_size, 0))
+
+    elif num_images == 3:
+        if is_web:
+            # Web: 1 grande à esquerda, 2 pequenas empilhadas à direita
+            left_width = (canvas_width - border_size) * 2 // 3
+            right_width = canvas_width - left_width - border_size
+            half_height = (canvas_height - border_size) // 2
+            
+            img1 = resize_and_crop_to_fill(images[0], left_width, canvas_height)
+            img2 = resize_and_crop_to_fill(images[1], right_width, half_height)
+            img3 = resize_and_crop_to_fill(images[2], right_width, half_height)
+            
+            canvas.paste(img1, (0, 0))
+            canvas.paste(img2, (left_width + border_size, 0))
+            canvas.paste(img3, (left_width + border_size, half_height + border_size))
+        else:
+            # IG: layout original
+            half_height = (canvas_height - border_size) // 2
+            half_width = (canvas_width - border_size) // 2
+            img1 = resize_and_crop_to_fill(images[0], half_width, half_height)
+            img2 = resize_and_crop_to_fill(images[1], half_width, half_height)
+            img3 = resize_and_crop_to_fill(images[2], canvas_width, half_height)
+            canvas.paste(img1, (0, 0))
+            canvas.paste(img2, (half_width + border_size, 0))
+            canvas.paste(img3, (0, half_height + border_size))
+
+    elif num_images == 4:
+        if is_web:
+            # Web: 4 imagens lado a lado horizontalmente
+            slot_width = (canvas_width - 3 * border_size) // 4
+            for i in range(4):
+                img = resize_and_crop_to_fill(images[i], slot_width, canvas_height)
+                x_pos = i * (slot_width + border_size)
+                canvas.paste(img, (x_pos, 0))
+        else:
+            # IG: Layout 2x2
+            half_height = (canvas_height - border_size) // 2
+            half_width = (canvas_width - border_size) // 2
+            img1 = resize_and_crop_to_fill(images[0], half_width, half_height)
+            img2 = resize_and_crop_to_fill(images[1], half_width, half_height)
+            img3 = resize_and_crop_to_fill(images[2], half_width, half_height)
+            img4 = resize_and_crop_to_fill(images[3], half_width, half_height)
+            canvas.paste(img1, (0, 0))
+            canvas.paste(img2, (half_width + border_size, 0))
+            canvas.paste(img3, (0, half_height + border_size))
+            canvas.paste(img4, (half_width + border_size, half_height + border_size))
+
+    elif num_images == 5:
+        if is_web:
+            # Web: 3 em cima, 2 embaixo
+            top_height = (canvas_height - border_size) * 3 // 5
+            bottom_height = canvas_height - top_height - border_size
+            
+            available_width = canvas_width - 2 * border_size
+            third_width = available_width // 3
+            third_width_last = canvas_width - (third_width * 2 + border_size * 2)
+            half_width = (canvas_width - border_size) // 2
+            
+            # 3 imagens em cima
+            img1 = resize_and_crop_to_fill(images[0], third_width, top_height)
+            img2 = resize_and_crop_to_fill(images[1], third_width, top_height)
+            img3 = resize_and_crop_to_fill(images[2], third_width_last, top_height)
+            canvas.paste(img1, (0, 0))
+            canvas.paste(img2, (third_width + border_size, 0))
+            canvas.paste(img3, (third_width * 2 + border_size * 2, 0))
+            
+            # 2 imagens embaixo
+            y_offset = top_height + border_size
+            img4 = resize_and_crop_to_fill(images[3], half_width, bottom_height)
+            img5 = resize_and_crop_to_fill(images[4], half_width, bottom_height)
+            canvas.paste(img4, (0, y_offset))
+            canvas.paste(img5, (half_width + border_size, y_offset))
+        else:
+            # IG: layout original
+            top_height = (canvas_height - border_size) * 2 // 5
+            bottom_height = canvas_height - top_height - border_size
+            half_width = (canvas_width - border_size) // 2
+
+            img1 = resize_and_crop_to_fill(images[0], half_width, top_height)
+            img2 = resize_and_crop_to_fill(images[1], half_width, top_height)
+            canvas.paste(img1, (0, 0))
+            canvas.paste(img2, (half_width + border_size, 0))
+
+            y_offset = top_height + border_size
+            third_width = (canvas_width - 2 * border_size) // 3
+            third_width_last = canvas_width - (third_width * 2 + border_size * 2)
+            
+            img3 = resize_and_crop_to_fill(images[2], third_width, bottom_height)
+            img4 = resize_and_crop_to_fill(images[3], third_width, bottom_height)
+            img5 = resize_and_crop_to_fill(images[4], third_width_last, bottom_height)
+            canvas.paste(img3, (0, y_offset))
+            canvas.paste(img4, (third_width + border_size, y_offset))
+            canvas.paste(img5, (third_width * 2 + border_size * 2, y_offset))
+
+    elif num_images == 6:
+        if is_web:
+            # Web: 3x2 (3 colunas, 2 linhas)
+            half_height = (canvas_height - border_size) // 2
+            available_width = canvas_width - 2 * border_size
+            third_width = available_width // 3
+            third_width_last = canvas_width - (third_width * 2 + border_size * 2)
+
+            # Primeira linha
+            img1 = resize_and_crop_to_fill(images[0], third_width, half_height)
+            img2 = resize_and_crop_to_fill(images[1], third_width, half_height)
+            img3 = resize_and_crop_to_fill(images[2], third_width_last, half_height)
+            canvas.paste(img1, (0, 0))
+            canvas.paste(img2, (third_width + border_size, 0))
+            canvas.paste(img3, (third_width * 2 + border_size * 2, 0))
+
+            # Segunda linha
+            y_offset = half_height + border_size
+            img4 = resize_and_crop_to_fill(images[3], third_width, half_height)
+            img5 = resize_and_crop_to_fill(images[4], third_width, half_height)
+            img6 = resize_and_crop_to_fill(images[5], third_width_last, half_height)
+            canvas.paste(img4, (0, y_offset))
+            canvas.paste(img5, (third_width + border_size, y_offset))
+            canvas.paste(img6, (third_width * 2 + border_size * 2, y_offset))
+        else:
+            # IG: layout original (3x2)
+            half_height = (canvas_height - border_size) // 2
+            third_width = (canvas_width - 2 * border_size) // 3
+            third_width_last = canvas_width - (third_width * 2 + border_size * 2)
+
+            # Primeira linha
+            img1 = resize_and_crop_to_fill(images[0], third_width, half_height)
+            img2 = resize_and_crop_to_fill(images[1], third_width, half_height)
+            img3 = resize_and_crop_to_fill(images[2], third_width, half_height)
+            canvas.paste(img1, (0, 0))
+            canvas.paste(img2, (third_width + border_size, 0))
+            canvas.paste(img3, (third_width * 2 + border_size * 2, 0))
+
+            # Segunda linha
+            y_offset = half_height + border_size
+            img4 = resize_and_crop_to_fill(images[3], third_width, half_height)
+            img5 = resize_and_crop_to_fill(images[4], third_width, half_height)
+            img6 = resize_and_crop_to_fill(images[5], third_width_last, half_height)
+            canvas.paste(img4, (0, y_offset))
+            canvas.paste(img5, (third_width + border_size, y_offset))
+            canvas.paste(img6, (third_width * 2 + border_size * 2, y_offset))
+
+    else:
+        raise HTTPException(status_code=400, detail="Apenas até 6 imagens são suportadas.")
+
+    return canvas
+
+def parse_image_urls(image_url_param: Union[str, List[str]]) -> List[str]:
+    """Converte o parâmetro de URL(s) em lista de URLs"""
+    if isinstance(image_url_param, list):
+        return image_url_param
+    elif isinstance(image_url_param, str):
+        # Se contém vírgulas, divide em múltiplas URLs
+        if ',' in image_url_param:
+            return [url.strip() for url in image_url_param.split(',') if url.strip()]
+        else:
+            return [image_url_param]
+    return []
+
 def create_gradient_overlay(width: int, height: int, text_position: str = "bottom") -> Image.Image:
     """
     Cria gradiente overlay baseado na posição do texto
@@ -175,7 +353,7 @@ def get_device_dimensions(device: str) -> tuple[int, int]:
     else:  # Instagram por padrão
         return (1080, 1350)
 
-def create_canvas(image_url: Optional[str], headline: Optional[str], device: str = "ig", 
+def create_canvas(image_url: Optional[Union[str, List[str]]], headline: Optional[str], device: str = "ig", 
                  text_position: str = "bottom", text_color: str = "white") -> BytesIO:
     width, height = get_device_dimensions(device)
     is_web = device.lower() == "web"
@@ -196,11 +374,21 @@ def create_canvas(image_url: Optional[str], headline: Optional[str], device: str
 
     canvas = Image.new("RGBA", (width, height), color=(255, 255, 255, 255))
 
-    # Adicionar imagem de fundo se fornecida
+    # Adicionar imagem(s) de fundo se fornecida(s)
     if image_url:
-        img = download_image_from_url(image_url)
-        filled_img = resize_and_crop_to_fill(img, width, height)
-        canvas.paste(filled_img, (0, 0))
+        parsed_urls = parse_image_urls(image_url)
+        if parsed_urls:
+            if len(parsed_urls) > 6:
+                raise HTTPException(status_code=400, detail="Máximo de 6 imagens permitidas")
+            
+            if len(parsed_urls) == 1:
+                # Uma única imagem - comportamento original
+                img = download_image_from_url(parsed_urls[0])
+                filled_img = resize_and_crop_to_fill(img, width, height)
+                canvas.paste(filled_img, (0, 0))
+            else:
+                # Múltiplas imagens - criar colagem
+                canvas = create_collage_background(parsed_urls, width, height, device)
 
     # Para Instagram: adicionar gradiente e texto
     if not is_web:
@@ -278,7 +466,7 @@ def create_canvas(image_url: Optional[str], headline: Optional[str], device: str
 
 @router.get("/cover/news")
 def get_news_image(
-    image_url: Optional[str] = Query(None, description="URL da imagem de fundo"),
+    image_url: Optional[Union[str, List[str]]] = Query(None, description="URL da imagem ou lista de URLs separadas por vírgula para colagem (máximo 6)"),
     headline: Optional[str] = Query(None, description="Texto do título (opcional para IG, ignorado para web)"),
     device: str = Query("ig", description="Dispositivo: 'ig' para Instagram (1080x1350) ou 'web' para Web (1280x720)"),
     text_position: str = Query("bottom", description="Posição do texto: 'top' para topo ou 'bottom' para parte inferior"),