Update routers/image.py

routers/image.py

@@ -664,7 +664,7 @@ def get_logo_path(slide_color: int) -> str:
     else:
         return "recurveblack.png"  # Logo preta
 
-def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None, color: int = 1) -> BytesIO:
+def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None, color: int = 1, text_color: str = "white") -> BytesIO:
     """
     Cria um canvas para slides com imagem cortada baseada no parâmetro slide.
     
@@ -755,14 +755,17 @@ 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(color)
+    # Obter cor do texto baseada no parâmetro text_color
+    if text_color.lower() == "black":
+        text_color_tuple = (0, 0, 0, 255)  # #000000
+    else:  # white (padrão)
+        text_color_tuple = (255, 255, 255, 255)  # branco
     
     # 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: 155, L: 815, MAX 5 linhas
-        render_slide_text(draw, text, x=200, y=155, max_width=815, max_lines=5, text_color=text_color)
+        render_slide_text(draw, text, x=200, y=155, max_width=815, max_lines=5, text_color=text_color_tuple)
     
     # Adicionar texto se slide for 2 e text for fornecido
     if slide == 2 and text and text.strip():
@@ -770,7 +773,7 @@ def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None,
         # Renderizar texto com as especificações: X: 445, Y: 370, L: 585, MAX 19 linhas
         # A imagem do slide 2 agora vai de Y: 370 a Y: 1150 (altura 780px)
         # Para centralizar o texto na área da imagem, o centro é Y: 370 + (780/2) = 760
-        render_slide_text(draw, text, x=445, y=760, max_width=585, max_lines=19, center_vertical=True, text_color=text_color)
+        render_slide_text(draw, text, x=445, y=760, max_width=585, max_lines=19, center_vertical=True, text_color=text_color_tuple)
     
     # Adicionar logo no canto inferior direito
     try:
@@ -798,7 +801,7 @@ def create_slide_canvas(image_url: str, slide: int, text: Optional[str] = None,
     buffer.seek(0)
     return buffer
     
-def create_quote_canvas(quote_text: str, role: Optional[str] = None, name: Optional[str] = None, color: int = 1, quote_background: bool = False, image_url: Optional[str] = None) -> BytesIO:
+def create_quote_canvas(quote_text: str, role: Optional[str] = None, name: Optional[str] = None, color: int = 1, quote_background: bool = False, image_url: Optional[str] = None, text_color: str = "white") -> BytesIO:
     """
     Cria um canvas para quotes com fundo colorido, imagem quote.png e texto centralizado.
     
@@ -896,28 +899,26 @@ def create_quote_canvas(quote_text: str, role: Optional[str] = None, name: Optio
         text_x = 180
         text_y = canvas_height // 2  # Centralizar verticalmente
     
-    # Obter cor do texto baseada no quote_background
-    if quote_background:
-        # Quando quote_background=True, sempre usar branco
-        text_color = (255, 255, 255, 255)
-    else:
-        # Quando quote_background=False, usar cor baseada na cor do fundo
-        text_color = get_text_color(color)
+    # Obter cor do texto baseada no parâmetro text_color
+    if text_color.lower() == "black":
+        text_color_tuple = (0, 0, 0, 255)  # #000000
+    else:  # white (padrão)
+        text_color_tuple = (255, 255, 255, 255)  # branco
     
     # Adicionar imagem quote.png ou quoteblank.png acima do texto
     try:
-        # Escolher imagem baseada no quote_background e cor
+        # Escolher imagem baseada no quote_background e text_color
         if quote_background:
             # Quando quote_background=True, sempre usar quotered.png
             quote_image_name = "quotered.png"
         else:
-            # Quando quote_background=False, escolher baseado na cor
-            if color == 6:
-                # Cor 6 sempre usa quoteblank.png
-                quote_image_name = "quoteblank.png"
+            # Quando quote_background=False, escolher baseado no text_color
+            if text_color.lower() == "black":
+                # Se text_color=black, usar quote.png (preta)
+                quote_image_name = "quote.png"
             else:
-                # Outras cores: quotered.png se texto branco, quote.png se texto preto
-                quote_image_name = "quotered.png" if text_color == (255, 255, 255, 255) else "quote.png"
+                # Se text_color=white, usar quotered.png (vermelha)
+                quote_image_name = "quotered.png"
         quote_img = Image.open(quote_image_name).convert("RGBA")
         quote_width, quote_height = 87, 67.46
         
@@ -1043,11 +1044,11 @@ def create_quote_canvas(quote_text: str, role: Optional[str] = None, name: Optio
     quote_text_with_symbol = quote_text + '”'
     
     # Renderizar texto com as especificações solicitadas
-    render_quote_text(draw, quote_text_with_symbol, text_x, text_y, text_width, text_color, quote_background)
+    render_quote_text(draw, quote_text_with_symbol, text_x, text_y, text_width, text_color_tuple, quote_background)
     
     # Renderizar cargo e nome se fornecidos
     if role or name:
-        render_author_info(draw, canvas, role, name, text_x, text_y, text_width, quote_text_with_symbol, text_color, quote_background)
+        render_author_info(draw, canvas, role, name, text_x, text_y, text_width, quote_text_with_symbol, text_color_tuple, quote_background)
     
     buffer = BytesIO()
     canvas.convert("RGB").save(buffer, format="PNG")
@@ -1284,23 +1285,17 @@ def render_author_info(draw: ImageDraw.Draw, canvas: Image.Image, role: Optional
     except (OSError, IOError):
         font = ImageFont.load_default()
     
-    # Determinar cores baseadas no quote_background
-    if quote_background:
-        # Quando quote_background=True, sempre usar branco
+    # Determinar cores baseadas no text_color
+    if text_color == (255, 255, 255, 255):  # Se texto principal for branco
+        # Cor do cargo: branco
         cargo_color = (255, 255, 255, 255)
+        # Cor do nome: branco com 65% de opacidade
         nome_color = (255, 255, 255, int(255 * 0.65))
-    else:
-        # Quando quote_background=False, usar cores baseadas na cor do texto principal
-        if text_color == (255, 255, 255, 255):  # Se texto principal for branco
-            # Cor do cargo: branco
-            cargo_color = (255, 255, 255, 255)
-            # Cor do nome: branco com 65% de opacidade
-            nome_color = (255, 255, 255, int(255 * 0.65))
-        else:  # Se texto principal for preto
-            # Cor do cargo: 140F09 (RGB: 20, 15, 9)
-            cargo_color = (20, 15, 9, 255)
-            # Cor do nome: 140F09 com 65% de opacidade
-            nome_color = (20, 15, 9, int(255 * 0.65))
+    else:  # Se texto principal for preto
+        # Cor do cargo: 140F09 (RGB: 20, 15, 9)
+        cargo_color = (20, 15, 9, 255)
+        # Cor do nome: 140F09 com 65% de opacidade
+        nome_color = (20, 15, 9, int(255 * 0.65))
     
     # Calcular altura do texto principal para posicionar o autor
     # Usar a mesma lógica do render_quote_text para calcular a altura
@@ -1513,7 +1508,7 @@ def render_author_info(draw: ImageDraw.Draw, canvas: Image.Image, role: Optional
             current_y += name_height + 5
 
 def create_canvas(image_url: str, text: Optional[str] = None, 
-                 text_position: str = "bottom", background: bool = False, color: int = 1, background_position: str = "bottom") -> BytesIO:
+                 text_position: str = "bottom", background: bool = False, color: int = 1, background_position: str = "bottom", text_color: str = "white") -> BytesIO:
     # Dimensões fixas do Instagram
     width, height = 1080, 1350
     
@@ -1598,19 +1593,21 @@ def create_canvas(image_url: str, text: Optional[str] = None,
         background_rect = Image.new("RGBA", (width, background_height), color=bg_color)
         canvas.paste(background_rect, (0, background_start_y))
     else:
-        # Adicionar gradiente apenas se background=False
+        # Adicionar gradiente apenas se background=False e text_color não for black
         # Lógica inteligente para posicionamento do gradiente
         
-        # Se text_position=bottom, sempre gradiente embaixo (obrigatório)
-        if text_position == "bottom":
-            gradient_overlay = create_gradient_overlay(width, height, "bottom")
-            canvas = Image.alpha_composite(canvas, gradient_overlay)
-        
-        # Se text_position=top
-        elif text_position == "top":
-            # Sempre gradiente em cima quando texto está no topo
-            gradient_overlay = create_gradient_overlay(width, height, "top")
-            canvas = Image.alpha_composite(canvas, gradient_overlay)
+        # Se text_color=black, não adicionar gradientes pretos
+        if text_color.lower() != "black":
+            # Se text_position=bottom, sempre gradiente embaixo (obrigatório)
+            if text_position == "bottom":
+                gradient_overlay = create_gradient_overlay(width, height, "bottom")
+                canvas = Image.alpha_composite(canvas, gradient_overlay)
+            
+            # Se text_position=top
+            elif text_position == "top":
+                # Sempre gradiente em cima quando texto está no topo
+                gradient_overlay = create_gradient_overlay(width, height, "top")
+                canvas = Image.alpha_composite(canvas, gradient_overlay)
     
     # Adicionar logo na posição (X: 880, Y: 1260)
     try:
@@ -1662,14 +1659,17 @@ def create_canvas(image_url: str, text: Optional[str] = None,
         else:
             text_y = None
     
-    # Obter cor do texto baseada na cor do fundo
-    text_color = get_text_color(color)
+    # Obter cor do texto baseada no parâmetro text_color
+    if text_color.lower() == "black":
+        text_color_tuple = (0, 0, 0, 255)  # #000000
+    else:  # white (padrão)
+        text_color_tuple = (255, 255, 255, 255)  # branco
     
     # Renderizar texto se fornecido
     if text and text.strip() and text_y is not None:
         # Alinhar pela base quando no bottom; pelo topo quando no top
         align_text_top = (text_position == "top")
-        render_text(draw, text, x=78, y=text_y, max_width=924, max_lines=5, align_top=align_text_top, text_color=text_color)
+        render_text(draw, text, x=78, y=text_y, max_width=924, max_lines=5, align_top=align_text_top, text_color=text_color_tuple)
     
     buffer = BytesIO()
     canvas.convert("RGB").save(buffer, format="PNG")
@@ -1688,18 +1688,19 @@ def get_image(
     quote: Optional[str] = Query(None, description="Texto da quote para gerar poster com fundo colorido"),
     role: Optional[str] = Query(None, description="Position/role of the person who said the quote"),
     name: Optional[str] = Query(None, description="Name of the person who said the quote"),
-    quote_background: bool = Query(False, description="Se True, centraliza texto, role/name e símbolo da quote")
+    quote_background: bool = Query(False, description="Se True, centraliza texto, role/name e símbolo da quote"),
+    text_color: str = Query("white", description="Cor do texto: 'white' ou 'black'")
 ):
     try:
         # Se o parâmetro quote for fornecido, usar a funcionalidade de quote
         if quote is not None and quote.strip():
-            buffer = create_quote_canvas(quote, role, name, color, quote_background, image_url)
+            buffer = create_quote_canvas(quote, role, name, color, quote_background, image_url, text_color)
         # Se o parâmetro slide for fornecido, usar a funcionalidade de slides
         elif slide is not None:
-            buffer = create_slide_canvas(image_url, slide, text, color)
+            buffer = create_slide_canvas(image_url, slide, text, color, text_color)
         else:
             # Usar funcionalidade normal de criação de imagem
-            buffer = create_canvas(image_url, text, text_position, background, color, background_position)
+            buffer = create_canvas(image_url, text, text_position, background, color, background_position, text_color)
         return StreamingResponse(buffer, media_type="image/png")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erro ao gerar imagem: {str(e)}")