Update routers/memoriam.py

routers/memoriam.py

@@ -84,22 +84,41 @@ def create_gradient_bar(width: int, height: int, radius: int = 50) -> Image.Imag
             b = int(0xFF * (1 - ratio2) + 0xE3 * ratio2)
         draw.line([(x, 0), (x, height)], fill=(r, g, b, 255))
 
-    # Adiciona borda arredondada
+    # Bordas arredondadas
     mask = Image.new("L", (width, height), 0)
     mask_draw = ImageDraw.Draw(mask)
     mask_draw.rounded_rectangle([(0, 0), (width, height)], radius=radius, fill=255)
     gradient.putalpha(mask)
     return gradient
 
-def draw_year_text(draw: ImageDraw.Draw, prefix: str, year: str, x: int, y: int, font_path: str):
+def draw_year_with_icon(draw: ImageDraw.Draw, base_image: Image.Image, text: str, icon_path: str, x: int, y: int, font_path: str, icon_width: int):
     try:
         font = ImageFont.truetype(font_path, 24)
     except Exception:
         font = ImageFont.load_default()
 
-    text = f"{prefix} {year}"
-    spacing = 0.01 * 24  # 1%
-    draw.text((x, y), text, font=font, fill=(255, 255, 255), spacing=spacing)
+    spacing_between_icon_and_text = 8
+    text_width = draw.textlength(text, font=font)
+
+    # Carregar e redimensionar o ícone mantendo proporção da altura igual a altura do texto (aprox 28px)
+    icon = Image.open(icon_path).convert("RGBA")
+    icon_ratio = icon.width / icon.height
+    icon_height = 28  # altura aproximada para alinhamento com o texto (altura da linha)
+    icon_resized = icon.resize((icon_width, int(icon_width / icon_ratio))) if icon_width < icon.height else icon.resize((int(icon_height * icon_ratio), icon_height))
+
+    # Ajuste: usar icon_height fixa para alinhamento vertical (ícones diferentes podem ter alturas diferentes)
+    icon_resized = icon.resize((icon_width, icon_height))
+
+    # Y para alinhar verticalmente ícone e texto
+    icon_y = y
+    text_y = y
+
+    # Colar ícone na base_image
+    base_image.paste(icon_resized, (x, icon_y), icon_resized)
+
+    # Desenhar texto após o ícone + espaçamento
+    text_x = x + icon_width + spacing_between_icon_and_text
+    draw.text((text_x, text_y), text, font=font, fill=(255, 255, 255), spacing=0)
 
 def create_canvas(image_url: Optional[str], name: Optional[str], birth: Optional[str], death: Optional[str]) -> BytesIO:
     width, height = 1080, 1350
@@ -133,7 +152,7 @@ def create_canvas(image_url: Optional[str], name: Optional[str], birth: Optional
         y = 1100
         draw_single_line_text_centered(draw, name_upper, max_width, center_x, y, font_path)
 
-    # Barra e anos
+    # Barra gradiente
     bar_x = 480
     bar_y = int(1213)
     bar_width = 110
@@ -141,23 +160,27 @@ def create_canvas(image_url: Optional[str], name: Optional[str], birth: Optional
     bar = create_gradient_bar(bar_width, bar_height, radius=50)
     canvas.paste(bar, (bar_x, bar_y), bar)
 
-    # Textos laterais (anos)
+    # Altura da linha do texto para alinhamento vertical dos ícones e textos
     text_y = bar_y + (bar_height - 28) // 2
-    font_path_semibold = "fonts/Montserrat-SemiBold.ttf"
+
+    # Desenhar anos com ícones
+    font_path_semi = "fonts/Montserrat-SemiBold.ttf"
 
     if birth:
-        prefix = "⭐"
-        birth_text = f"{prefix} {birth}"
-        try:
-            font = ImageFont.truetype(font_path_semibold, 24)
-        except Exception:
-            font = ImageFont.load_default()
-        text_width = draw.textlength(birth_text, font=font)
-        draw_year_text(draw, prefix, birth, bar_x - 12 - text_width, text_y, font_path_semibold)
+        # calcular largura do texto para posicionar o ícone e o texto lado a lado
+        # posicionar a estrela + texto, sendo o x do texto após o ícone + espaçamento
+        # A estrela deve ficar 12px à esquerda da barra
+        icon_width_star = 30
+        # Ajusta x do ícone para ficar a 12px à esquerda do início da barra menos largura do texto e espaço do ícone+texto
+        text_width = draw.textlength(birth, font=ImageFont.truetype(font_path_semi, 24))
+        icon_x = bar_x - 12 - icon_width_star - 8 - int(text_width)  # 8px espaço ícone-texto
+        draw_year_with_icon(draw, canvas, birth, "star.png", icon_x, text_y, font_path_semi, icon_width_star)
 
     if death:
-        prefix = "✝"
-        draw_year_text(draw, prefix, death, bar_x + bar_width + 12, text_y, font_path_semibold)
+        icon_width_cross = 18
+        # Ícone + 8px + texto começa a 12px depois da barra
+        icon_x = bar_x + bar_width + 12
+        draw_year_with_icon(draw, canvas, death, "cross.png", icon_x, text_y, font_path_semi, icon_width_cross)
 
     buffer = BytesIO()
     canvas.save(buffer, format="PNG")
@@ -168,8 +191,8 @@ def create_canvas(image_url: Optional[str], name: Optional[str], birth: Optional
 def get_memoriam_image(
     image_url: Optional[str] = Query(None, description="URL da imagem de fundo"),
     name: Optional[str] = Query(None, description="Nome (será exibido em maiúsculas)"),
-    birth: Optional[str] = Query(None, description="Ano de nascimento"),
-    death: Optional[str] = Query(None, description="Ano de falecimento")
+    birth: Optional[str] = Query(None, description="Ano de nascimento (ex: 1943)"),
+    death: Optional[str] = Query(None, description="Ano de falecimento (ex: 2023)")
 ):
     try:
         buffer = create_canvas(image_url, name, birth, death)