Update routers/memoriam.py

routers/memoriam.py

@@ -91,15 +91,17 @@ def create_gradient_bar(width: int, height: int, radius: int = 50) -> Image.Imag
     gradient.putalpha(mask)
     return gradient
 
-def draw_year_text(draw: ImageDraw.Draw, text: str, x: int, y: int, font_path: str):
+def draw_year_text(draw: ImageDraw.Draw, prefix: str, year: str, x: int, y: int, font_path: str):
     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)
 
-def create_canvas(image_url: Optional[str], name: Optional[str], birth_year: Optional[str], death_year: Optional[str]) -> BytesIO:
+def create_canvas(image_url: Optional[str], name: Optional[str], birth: Optional[str], death: Optional[str]) -> BytesIO:
     width, height = 1080, 1350
     canvas = Image.new("RGBA", (width, height), (0, 0, 0, 0))  # Fundo transparente
 
@@ -140,13 +142,22 @@ def create_canvas(image_url: Optional[str], name: Optional[str], birth_year: Opt
     canvas.paste(bar, (bar_x, bar_y), bar)
 
     # Textos laterais (anos)
-    text_y = bar_y + (bar_height - 28) // 2  # centralizado verticalmente com base na altura da linha
+    text_y = bar_y + (bar_height - 28) // 2
+    font_path_semibold = "fonts/Montserrat-SemiBold.ttf"
 
-    if birth_year:
-        draw_year_text(draw, birth_year, bar_x - 12 - draw.textlength(birth_year, font=ImageFont.truetype("fonts/Montserrat-SemiBold.ttf", 24)), text_y, "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)
 
-    if death_year:
-        draw_year_text(draw, death_year, bar_x + bar_width + 12, text_y, "fonts/Montserrat-SemiBold.ttf")
+    if death:
+        prefix = "✝"
+        draw_year_text(draw, prefix, death, bar_x + bar_width + 12, text_y, font_path_semibold)
 
     buffer = BytesIO()
     canvas.save(buffer, format="PNG")
@@ -157,11 +168,11 @@ def create_canvas(image_url: Optional[str], name: Optional[str], birth_year: Opt
 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_year: Optional[str] = Query(None, description="Ano de nascimento (ex: 1943)"),
-    death_year: Optional[str] = Query(None, description="Ano de falecimento (ex: 2023)")
+    birth: Optional[str] = Query(None, description="Ano de nascimento"),
+    death: Optional[str] = Query(None, description="Ano de falecimento")
 ):
     try:
-        buffer = create_canvas(image_url, name, birth_year, death_year)
+        buffer = create_canvas(image_url, name, birth, death)
         return StreamingResponse(buffer, media_type="image/png")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erro ao gerar imagem: {str(e)}")