Update routers/inference_createposter.py

routers/inference_createposter.py

@@ -1,6 +1,6 @@
 import os
 import logging
-from urllib.parse import urlencode, quote
+from urllib.parse import urlencode, quote, parse_qs, urlparse, urlunparse
 from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
 from google import genai
@@ -18,6 +18,71 @@ class PosterResponse(BaseModel):
     result: dict
     urls: list
 
+def fix_citation_quotes(citation_text: str) -> str:
+    """
+    Corrige as aspas no texto de citação:
+    - Se não tiver aspas no início e fim, adiciona ❝ ❞
+    - Se tiver aspas comuns " ", substitui por ❝ ❞
+    - Se tiver outras aspas, substitui por ❝ ❞
+    """
+    if not citation_text or citation_text.strip() == "":
+        return citation_text
+    
+    text = citation_text.strip()
+    
+    # Verificar se já tem as aspas corretas
+    if text.startswith('❝') and text.endswith('❞'):
+        return text
+    
+    # Remover aspas existentes do início e fim
+    quote_chars = ['"', "'", '"', '"', ''', ''']
+    
+    # Remover aspas do início
+    while text and text[0] in quote_chars:
+        text = text[1:]
+    
+    # Remover aspas do fim
+    while text and text[-1] in quote_chars:
+        text = text[:-1]
+    
+    # Adicionar as aspas corretas
+    return f"❝{text.strip()}❞"
+
+def fix_url_citation(url: str) -> str:
+    """
+    Analisa uma URL e corrige o parâmetro citation se presente
+    """
+    try:
+        # Parse da URL
+        parsed_url = urlparse(url)
+        query_params = parse_qs(parsed_url.query)
+        
+        # Verificar se tem parâmetro citation
+        if 'citation' in query_params and query_params['citation']:
+            original_citation = query_params['citation'][0]
+            
+            # Corrigir as aspas
+            fixed_citation = fix_citation_quotes(original_citation)
+            
+            # Substituir o valor na query
+            query_params['citation'] = [fixed_citation]
+            
+            # Reconstruir a query string
+            new_query = urlencode(
+                {k: v[0] if isinstance(v, list) and len(v) == 1 else v for k, v in query_params.items()}, 
+                quote_via=quote
+            )
+            
+            # Reconstruir a URL
+            new_parsed_url = parsed_url._replace(query=new_query)
+            return urlunparse(new_parsed_url)
+        
+        return url
+        
+    except Exception as e:
+        logger.warning(f"Erro ao processar URL para correção de citation: {e}")
+        return url
+
 def format_url(base_url: str, endpoint: str, params: dict) -> str:
     """
     Formata uma URL completa com os parâmetros dados
@@ -47,6 +112,8 @@ def generate_urls_from_result(result: dict, base_url: str = "https://habulaj-new
     # Se for notícia simples
     if "endpoint" in result and "params" in result:
         url = format_url(base_url, result["endpoint"], result["params"])
+        # Corrigir citation na URL se presente
+        url = fix_url_citation(url)
         urls.append(url)
     
     # Se for carrossel com capa e slides
@@ -58,6 +125,8 @@ def generate_urls_from_result(result: dict, base_url: str = "https://habulaj-new
                 result["cover"]["endpoint"], 
                 result["cover"]["params"]
             )
+            # Corrigir citation na URL se presente
+            cover_url = fix_url_citation(cover_url)
             urls.append(cover_url)
         
         # Adicionar URLs dos slides
@@ -68,6 +137,8 @@ def generate_urls_from_result(result: dict, base_url: str = "https://habulaj-new
                     slide["endpoint"],
                     slide["params"]
                 )
+                # Corrigir citation na URL se presente
+                slide_url = fix_url_citation(slide_url)
                 urls.append(slide_url)
     
     return urls