Update routers/image.py

routers/image.py

@@ -427,9 +427,9 @@ def calculate_text_height(text: str, font, max_width: int) -> int:
     
     return total_height
 
-def render_slide_text(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width: int, max_lines: int = 5, center_vertical: bool = False) -> None:
+def render_slide_text(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width: int, max_lines: int = 5, center_vertical: bool = False, text_color: tuple = (0, 0, 0, 255)) -> None:
     """
-    Renderiza texto para slides com fonte padrão, cor #000000, tamanho inicial 32px.
+    Renderiza texto para slides com fonte padrão, tamanho inicial 32px.
     Suporta formatação com asteriscos para negrito.
     
     Args:
@@ -440,6 +440,7 @@ def render_slide_text(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width
         max_width: Largura máxima do texto
         max_lines: Número máximo de linhas (padrão: 5)
         center_vertical: Se True, centraliza verticalmente o texto na posição Y
+        text_color: Cor do texto como tuple (R, G, B, A)
     """
     import re
     
@@ -642,7 +643,7 @@ def render_slide_text(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width
         current_x = x
         for i, (segment_text, segment_font) in enumerate(line):
             font_to_use = final_regular_font if segment_font == regular_font else final_bold_font
-            draw.text((current_x, current_y), segment_text, fill=(0, 0, 0, 255), font=font_to_use)  # Cor #000000
+            draw.text((current_x, current_y), segment_text, fill=text_color, font=font_to_use)
             bbox = font_to_use.getbbox(segment_text)
             current_x += bbox[2] - bbox[0]
             
@@ -657,19 +658,64 @@ def render_slide_text(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width
         # Aplicar line height de 120% (adicionar 20% de espaçamento extra)
         current_y += int(max_line_height * 1.20)
 
-def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None) -> BytesIO:
+def get_slide_color(slide_color: int) -> tuple:
+    """
+    Retorna a cor do slide baseada no número fornecido.
+    
+    Args:
+        slide_color: Número da cor (1-8)
+    
+    Returns:
+        Tuple (R, G, B, A) da cor
+    """
+    color_map = {
+        1: (0, 0, 0, 255),        # #000000 - Preto
+        2: (191, 198, 147, 255),  # #BFC693
+        3: (246, 164, 29, 255),   # #f6a41d
+        4: (245, 244, 239, 255),  # #f5f4ef
+        5: (111, 180, 165, 255),  # #6fb4a5
+        6: (25, 83, 208, 255),    # #1953d0
+        7: (255, 80, 17, 255),    # #ff5011
+        8: (255, 217, 194, 255),  # #ffd9c2
+    }
+    
+    # Se o número for inválido, usar preto como padrão
+    return color_map.get(slide_color, (0, 0, 0, 255))
+
+def get_text_color(slide_color: int) -> tuple:
+    """
+    Retorna a cor do texto baseada na cor do slide.
+    
+    Args:
+        slide_color: Número da cor do slide (1-8)
+    
+    Returns:
+        Tuple (R, G, B, A) da cor do texto
+    """
+    # Se for cor 1 (preto), usar texto branco, senão usar texto preto
+    if slide_color == 1:
+        return (255, 255, 255, 255)  # Branco
+    else:
+        return (0, 0, 0, 255)  # Preto
+
+def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None, slide_color: int = 1) -> BytesIO:
     """
     Cria um canvas para slides com imagem cortada baseada no parâmetro slide.
     
     Args:
         image_url: URL da imagem de fundo
         slide: 1 para cortar início (880px), 2 para cortar final (400px)
+        text: Texto opcional para exibir no slide
+        slide_color: Cor do slide (1-8)
     """
     # Dimensões do poster
     canvas_width, canvas_height = 1080, 1350
     
-    # Criar canvas com fundo #70b4a7
-    canvas = Image.new("RGBA", (canvas_width, canvas_height), color=(112, 180, 167, 255))  # #70b4a7
+    # Obter cor do slide
+    slide_bg_color = get_slide_color(slide_color)
+    
+    # Criar canvas com cor do slide
+    canvas = Image.new("RGBA", (canvas_width, canvas_height), color=slide_bg_color)
     
     # Baixar imagem original
     img = download_image_from_url(image_url)
@@ -743,11 +789,14 @@ def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None)
     # Colar imagem no canvas
     canvas.paste(img_with_mask, (x_pos, y_pos), img_with_mask)
     
+    # Obter cor do texto baseada na cor do slide
+    text_color = get_text_color(slide_color)
+    
     # Adicionar texto se slide for 1 e text for fornecido
     if slide == 1 and text and text.strip():
         draw = ImageDraw.Draw(canvas)
         # Renderizar texto com as especificações: X: 200, Y: 210, L: 815, MAX 5 linhas
-        render_slide_text(draw, text, x=200, y=210, max_width=815, max_lines=5)
+        render_slide_text(draw, text, x=200, y=210, max_width=815, max_lines=5, text_color=text_color)
     
     # Adicionar texto se slide for 2 e text for fornecido
     if slide == 2 and text and text.strip():
@@ -755,7 +804,7 @@ def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None)
         # Renderizar texto com as especificações: X: 445, Y: 430, L: 585, A: 748, MAX 19 linhas
         # A imagem do slide 2 vai de Y: 425 a Y: 1205 (altura 780px)
         # Para centralizar o texto na área da imagem, o centro é Y: 425 + (780/2) = 815
-        render_slide_text(draw, text, x=445, y=815, max_width=585, max_lines=19, center_vertical=True)
+        render_slide_text(draw, text, x=445, y=815, max_width=585, max_lines=19, center_vertical=True, text_color=text_color)
     
     buffer = BytesIO()
     canvas.convert("RGB").save(buffer, format="PNG")
@@ -887,12 +936,13 @@ def get_image(
     citation: Optional[str] = Query(None, description="Citação a ser exibida entre aspas (fonte Cheltenham Normal 700)"),
     text_position: str = Query("bottom", description="Posição do texto: 'top' ou 'bottom'"),
     citation_position: str = Query("top", description="Posição da citação: 'top' ou 'bottom'"),
-    slide: Optional[int] = Query(None, description="Número do slide: 1 para início (880px), 2 para final (400px)")
+    slide: Optional[int] = Query(None, description="Número do slide: 1 para início (880px), 2 para final (400px)"),
+    slide_color: Optional[int] = Query(1, description="Cor do slide: 1=#000000, 2=#BFC693, 3=#f6a41d, 4=#f5f4ef, 5=#6fb4a5, 6=#1953d0, 7=#ff5011, 8=#ffd9c2")
 ):
     try:
         # Se o parâmetro slide for fornecido, usar a funcionalidade de slides
         if slide is not None:
-            buffer = create_slide_canvas(image_url, slide, text)
+            buffer = create_slide_canvas(image_url, slide, text, slide_color)
         else:
             # Usar funcionalidade normal de criação de imagem
             buffer = create_canvas(image_url, text, citation, text_position, citation_position)