Update routers/video.py

routers/video.py

@@ -131,7 +131,7 @@ def render_headline_text(draw: ImageDraw.Draw, text: str, x: int, y: int, max_wi
         draw.text((centered_x, adjusted_y), line, fill=(0, 0, 0, 89), font=final_font)  # Texto com opacidade para sombra
         
         # Atualizar posição para próxima linha (subindo)
-        current_y -= int(line_height * 1.4)  # Line height de 130%
+        current_y -= int(line_height * 1.4)  # Line height de 140%
 
 def render_headline_text_white(draw: ImageDraw.Draw, text: str, x: int, y: int, max_width: int, max_lines: int = 5) -> None:
     """
@@ -238,7 +238,37 @@ def render_headline_text_white(draw: ImageDraw.Draw, text: str, x: int, y: int,
         draw.text((centered_x, adjusted_y), line, fill=(255, 255, 255, 255), font=final_font)
         
         # Atualizar posição para próxima linha (subindo)
-        current_y -= int(line_height * 1.4)  # Line height de 130%
+        current_y -= int(line_height * 1.4)  # Line height de 140%
+
+def render_badge(image: Image.Image, x: int, y: int) -> None:
+    """
+    Renderiza o badge.png acima do título.
+    
+    Args:
+        image: Imagem principal onde o badge será desenhado
+        x: Posição X centralizada
+        y: Posição Y (topo do badge)
+    """
+    try:
+        # Carregar o badge
+        badge = Image.open("badge.png")
+        
+        # Redimensionar para as dimensões especificadas (L: 197, Altura: 37)
+        badge = badge.resize((197, 37), Image.Resampling.LANCZOS)
+        
+        # Converter para RGBA se necessário
+        if badge.mode != 'RGBA':
+            badge = badge.convert('RGBA')
+        
+        # Calcular posição centralizada
+        badge_x = x - (197 // 2)  # Centralizar horizontalmente
+        
+        # Colar o badge na imagem principal
+        image.paste(badge, (badge_x, y), badge)
+        
+    except (OSError, IOError) as e:
+        # Se não conseguir carregar o badge, apenas ignora
+        print(f"Erro ao carregar badge: {e}")
 
 @router.get("/cover/video")
 def get_video_cover(
@@ -297,6 +327,58 @@ def get_video_cover(
             # Converter para RGBA para suportar transparência na sombra
             background_image = background_image.convert('RGBA')
             
+            # Calcular altura total do texto para posicionar o badge
+            # Criar uma imagem temporária para calcular a altura
+            temp_img = Image.new('RGBA', background_image.size, (0, 0, 0, 0))
+            temp_draw = ImageDraw.Draw(temp_img)
+            
+            # Calcular altura do texto
+            try:
+                font_path = "fonts/Montserrat-ExtraBold.ttf"
+                test_font = ImageFont.truetype(font_path, 70)
+            except (OSError, IOError):
+                test_font = ImageFont.load_default()
+            
+            # Simular quebra de texto para calcular altura
+            words = headline.upper().split()
+            lines = []
+            current_line = []
+            max_width = 720
+            
+            for word in words:
+                test_line = " ".join(current_line + [word])
+                bbox = temp_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))
+            
+            # Limitar a 5 linhas
+            if len(lines) > 5:
+                lines = lines[:5]
+            
+            # Calcular altura total do texto
+            total_text_height = 0
+            for line in lines:
+                bbox = temp_draw.textbbox((0, 0), line, font=test_font)
+                line_height = bbox[3] - bbox[1]
+                total_text_height += int(line_height * 1.4)
+            
+            # Posição do badge: 16px acima do topo do texto
+            badge_y = 1270 - total_text_height - 16
+            
+            # Renderizar o badge
+            render_badge(background_image, 540, badge_y)
+            
             # 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)