Update routers/video.py

routers/video.py

@@ -1,13 +1,220 @@
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, Query
 from fastapi.responses import StreamingResponse
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont, ImageFilter
 from io import BytesIO
 import requests
+from typing import Optional
 
 router = APIRouter()
 
+def render_headline_text(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width: int, max_lines: int = 5) -> None:
+    """
+    Renderiza headline com fonte Montserrat-ExtraBold.
+    
+    Args:
+        draw: Objeto ImageDraw para desenhar
+        text: Texto do headline
+        x: Posição X
+        y: Posição Y
+        max_width: Largura máxima do texto
+        max_lines: Número máximo de linhas (padrão: 5)
+    """
+    if not text.strip():
+        return
+    
+    # Carregar fonte Montserrat-ExtraBold
+    try:
+        font_path = "fonts/Montserrat-ExtraBold.ttf"
+        base_font = ImageFont.truetype(font_path, 70)
+    except (OSError, IOError):
+        # Fallback para fonte padrão se não encontrar a fonte
+        base_font = ImageFont.load_default()
+    
+    # Dividir texto em palavras
+    words = text.split()
+    if not words:
+        return
+    
+    # Função para quebrar texto em linhas
+    def wrap_text(text, font_size, max_width):
+        try:
+            test_font = ImageFont.truetype(font_path, font_size)
+        except (OSError, IOError):
+            test_font = ImageFont.load_default()
+        
+        lines = []
+        current_line = []
+        
+        for word in words:
+            test_line = " ".join(current_line + [word])
+            bbox = draw.textbbox((0, 0), test_line, font=test_font)
+            line_width = bbox[2] - bbox[0]
+            
+            if line_width <= max_width:
+                current_line.append(word)
+            else:
+                if current_line:
+                    lines.append(" ".join(current_line))
+                    current_line = [word]
+                else:
+                    # Palavra muito longa, adiciona mesmo assim
+                    lines.append(word)
+        
+        if current_line:
+            lines.append(" ".join(current_line))
+        
+        return lines
+    
+    # Encontrar o tamanho de fonte ideal
+    font_size = 70  # Tamanho inicial
+    min_font_size = 20  # Tamanho mínimo
+    
+    # Tentar diferentes tamanhos de fonte até encontrar um que caiba em max_lines
+    while font_size >= min_font_size:
+        lines = wrap_text(text, font_size, max_width)
+        
+        # Se o texto cabe em max_lines ou menos, usar este tamanho
+        if len(lines) <= max_lines:
+            break
+        
+        # Reduzir o tamanho da fonte
+        font_size -= 2
+    
+    # Garantir que não seja menor que o tamanho mínimo
+    font_size = max(font_size, min_font_size)
+    
+    # Carregar fonte final
+    try:
+        final_font = ImageFont.truetype(font_path, font_size)
+    except (OSError, IOError):
+        final_font = ImageFont.load_default()
+    
+    # Quebrar texto final
+    lines = wrap_text(text, font_size, max_width)
+    
+    # Se ainda não couber em max_lines, forçar quebra nas primeiras max_lines
+    if len(lines) > max_lines:
+        combined_text = " ".join(lines[:max_lines-1] + [" ".join(lines[max_lines-1:])])
+        lines = wrap_text(combined_text, font_size, max_width)
+        lines = lines[:max_lines]
+    
+    # Calcular altura total do texto
+    line_heights = []
+    for line in lines:
+        bbox = draw.textbbox((0, 0), line, font=final_font)
+        line_height = bbox[3] - bbox[1]
+        line_heights.append(line_height)
+    
+    # Calcular posição inicial (Y é a posição da primeira linha)
+    current_y = y
+    
+    # Desenhar cada linha
+    for i, line in enumerate(lines):
+        line_height = line_heights[i]
+        
+        # Desenhar texto principal (branco)
+        draw.text((x, current_y), line, fill=(0, 0, 0, 89), font=final_font)  # Texto com opacidade para sombra
+        
+        # Atualizar posição para próxima linha
+        current_y += int(line_height * 1.2)  # Line height de 120%
+
+def render_headline_text_white(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width: int, max_lines: int = 5) -> None:
+    """
+    Renderiza headline branco com fonte Montserrat-ExtraBold.
+    """
+    if not text.strip():
+        return
+    
+    # Carregar fonte Montserrat-ExtraBold
+    try:
+        font_path = "fonts/Montserrat-ExtraBold.ttf"
+    except (OSError, IOError):
+        font_path = None
+    
+    # Dividir texto em palavras
+    words = text.split()
+    if not words:
+        return
+    
+    # Função para quebrar texto em linhas
+    def wrap_text(text, font_size, max_width):
+        try:
+            test_font = ImageFont.truetype(font_path, font_size)
+        except (OSError, IOError):
+            test_font = ImageFont.load_default()
+        
+        lines = []
+        current_line = []
+        
+        for word in words:
+            test_line = " ".join(current_line + [word])
+            bbox = draw.textbbox((0, 0), test_line, font=test_font)
+            line_width = bbox[2] - bbox[0]
+            
+            if line_width <= max_width:
+                current_line.append(word)
+            else:
+                if current_line:
+                    lines.append(" ".join(current_line))
+                    current_line = [word]
+                else:
+                    lines.append(word)
+        
+        if current_line:
+            lines.append(" ".join(current_line))
+        
+        return lines
+    
+    # Encontrar o tamanho de fonte ideal
+    font_size = 70
+    min_font_size = 20
+    
+    while font_size >= min_font_size:
+        lines = wrap_text(text, font_size, max_width)
+        if len(lines) <= max_lines:
+            break
+        font_size -= 2
+    
+    font_size = max(font_size, min_font_size)
+    
+    # Carregar fonte final
+    try:
+        final_font = ImageFont.truetype(font_path, font_size)
+    except (OSError, IOError):
+        final_font = ImageFont.load_default()
+    
+    # Quebrar texto final
+    lines = wrap_text(text, font_size, max_width)
+    
+    if len(lines) > max_lines:
+        combined_text = " ".join(lines[:max_lines-1] + [" ".join(lines[max_lines-1:])])
+        lines = wrap_text(combined_text, font_size, max_width)
+        lines = lines[:max_lines]
+    
+    # Calcular altura total do texto
+    line_heights = []
+    for line in lines:
+        bbox = draw.textbbox((0, 0), line, font=final_font)
+        line_height = bbox[3] - bbox[1]
+        line_heights.append(line_height)
+    
+    # Calcular posição inicial
+    current_y = y
+    
+    # Desenhar cada linha (branco)
+    for i, line in enumerate(lines):
+        line_height = line_heights[i]
+        
+        # Desenhar texto principal (branco)
+        draw.text((x, current_y), line, fill=(255, 255, 255, 255), font=final_font)
+        
+        # Atualizar posição para próxima linha
+        current_y += int(line_height * 1.2)
+
 @router.get("/cover/video")
-def get_video_cover():
+def get_video_cover(
+    headline: Optional[str] = Query(None, description="Headline a ser exibido")
+):
     """
     Endpoint que retorna uma imagem de fundo com dimensões 1080x1920.
     """
@@ -56,6 +263,49 @@ def get_video_cover():
         if background_image.mode != 'RGB':
             background_image = background_image.convert('RGB')
         
+        # Adicionar headline se fornecido
+        if headline and headline.strip():
+            # Converter para RGBA para suportar transparência na sombra
+            background_image = background_image.convert('RGBA')
+            
+            # Criar uma nova imagem para o texto com transparência
+            text_img = Image.new('RGBA', background_image.size, (0, 0, 0, 0))
+            text_draw = ImageDraw.Draw(text_img)
+            
+            # Renderizar o texto na imagem temporária (para sombra)
+            render_headline_text(
+                text_draw, 
+                headline, 
+                x=180, 
+                y=945, 
+                max_width=720, 
+                max_lines=5
+            )
+            
+            # Aplicar desfoque na sombra
+            shadow_img = text_img.copy()
+            shadow_img = shadow_img.filter(ImageFilter.GaussianBlur(radius=18))
+            
+            # Aplicar offset de 3,3 na sombra
+            shadow_offset = Image.new('RGBA', background_image.size, (0, 0, 0, 0))
+            shadow_offset.paste(shadow_img, (3, 3))
+            
+            # Combinar sombra com imagem de fundo
+            background_image = Image.alpha_composite(background_image, shadow_offset)
+            
+            # Criar imagem para texto principal (branco)
+            text_white_img = Image.new('RGBA', background_image.size, (0, 0, 0, 0))
+            text_white_draw = ImageDraw.Draw(text_white_img)
+            
+            # Renderizar texto branco
+            render_headline_text_white(text_white_draw, headline, 180, 945, 720, 5)
+            
+            # Combinar texto principal
+            background_image = Image.alpha_composite(background_image, text_white_img)
+            
+            # Converter de volta para RGB
+            background_image = background_image.convert('RGB')
+        
         # Converter para BytesIO
         buffer = BytesIO()
         background_image.save(buffer, format="JPEG", quality=95)