Update routers/image.py

routers/image.py

@@ -4,7 +4,7 @@ from PIL import Image, ImageDraw, ImageFont
 from io import BytesIO
 import requests
 import re
-from typing import Optional, List, Tuple
+from typing import Optional, List, Tuple, Union
 
 router = APIRouter()
 
@@ -39,6 +39,112 @@ def resize_and_crop_to_fill(img: Image.Image, target_width: int, target_height:
     top = (scale_height - target_height) // 2
     return img_resized.crop((left, top, left + target_width, top + target_height))
 
+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_collage_background(image_urls: List[str], canvas_width: int, canvas_height: int) -> Image.Image:
+    """Cria uma colagem como fundo baseada na lista de URLs para Instagram"""
+    num_images = len(image_urls)
+    border_size = 4 if num_images > 1 else 0  # Linha mais fina e elegante
+
+    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:
+        # 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:
+        # Layout: 2 em cima, 1 embaixo
+        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:
+        # 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:
+        # Layout: 2 em cima, 3 embaixo
+        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:
+        # Layout 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 create_gradient_overlay(width: int, height: int, positions: List[str], expanded: bool = False) -> Image.Image:
     gradient = Image.new("RGBA", (width, height))
     draw = ImageDraw.Draw(gradient)
@@ -282,7 +388,7 @@ def add_logo(canvas: Image.Image):
     except Exception as e:
         print(f"Aviso: Erro ao carregar a logo: {e}")
 
-def create_canvas(image_url: Optional[str], text: Optional[str], text_position: str = "bottom",
+def create_canvas(image_url: Optional[Union[str, List[str]]], text: Optional[str], text_position: str = "bottom",
                  citation: Optional[str] = None, citation_direction: str = "bottom", 
                  text_color: str = "white") -> BytesIO:
     width, height = get_device_dimensions()
@@ -306,10 +412,21 @@ def create_canvas(image_url: Optional[str], text: Optional[str], text_position:
 
     canvas = Image.new("RGBA", (width, height), color=(255, 255, 255, 255))
 
+    # Processar imagem(s) - suporte para colagem
     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)
 
     # Determinar posições do gradiente e se precisa expandir
     # Só aplicar gradiente se o texto não for preto
@@ -492,7 +609,7 @@ def create_canvas(image_url: Optional[str], text: Optional[str], text_position:
     buffer.seek(0)
     return buffer
 
-def create_cover_canvas(image_url: Optional[str], title: Optional[str], title_position: str = "bottom",
+def create_cover_canvas(image_url: Optional[Union[str, List[str]]], title: Optional[str], title_position: str = "bottom",
                        text_color: str = "white") -> BytesIO:
     width, height = get_device_dimensions()
     padding_x, top_padding = 60, 60
@@ -501,10 +618,21 @@ def create_cover_canvas(image_url: Optional[str], title: Optional[str], title_po
 
     canvas = Image.new("RGBA", (width, height), color=(255, 255, 255, 255))
 
+    # Processar imagem(s) - suporte para colagem
     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)
 
     gradient_positions = []
     # Só aplicar gradiente se o texto não for preto
@@ -546,7 +674,7 @@ def create_cover_canvas(image_url: Optional[str], title: Optional[str], title_po
 
 @router.get("/create/image")
 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)"),
     text: Optional[str] = Query(None, description="Texto com suporte a tags <strong>"),
     text_position: str = Query("bottom", description="Posição do texto: 'top' para topo ou 'bottom' para parte inferior"),
     citation: Optional[str] = Query(None, description="Texto da citação"),
@@ -561,7 +689,7 @@ def get_news_image(
 
 @router.get("/create/cover/image")
 def get_cover_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)"),
     title: Optional[str] = Query(None, description="Título da capa"),
     title_position: str = Query("bottom", description="Posição do título: 'top' para topo ou 'bottom' para parte inferior"),
     text_color: str = Query("white", description="Cor do texto: 'white' (padrão) ou 'black'. Se 'black', remove o gradiente de fundo")