@danhachuel/thunderbolt 0.7.39 → 0.7.40

package/hermes_ui/media_generation.py

@@ -40,7 +40,7 @@ from .provider_routing import (
     route_json_request,
 )
 from .storage import STORAGE, ensure_storage, write_json
-from .thumbnail_generation import _compose_thumbnail_prompt, generate_thumbnail_image, normalize_thumbnail_bytes
+from .thumbnail_generation import _compose_thumbnail_prompt, generate_thumbnail_image, infer_thumbnail_aspect_ratio, normalize_thumbnail_bytes
 from .canva_client import CanvaClient
 from .canva_mcp_workflow import run_direct_canva_thumbnail
 
@@ -267,7 +267,14 @@ def _poll_image(card: Mapping[str, Any], request_id: str, *, attempts: int = 24,
     raise MediaGenerationError("O provider de imagem não concluiu dentro do limite de polling.")
 
 
-def _download_or_write(image_bytes: bytes | None, url: str, destination: Path, card: Mapping[str, Any]) -> Path:
+def _download_or_write(
+    image_bytes: bytes | None,
+    url: str,
+    destination: Path,
+    card: Mapping[str, Any],
+    *,
+    aspect_ratio: str = "16:9",
+) -> Path:
     if image_bytes is None and url:
         try:
             response = requests.get(url, headers={"Authorization": f"Bearer {_api_key(card)}"} if _api_key(card) else {}, timeout=180)
@@ -278,7 +285,7 @@ def _download_or_write(image_bytes: bytes | None, url: str, destination: Path, c
     if not image_bytes:
         raise MediaGenerationError("O provider concluiu a chamada mas não devolveu uma imagem utilizável.")
     try:
-        image_bytes = normalize_thumbnail_bytes(image_bytes)
+        image_bytes = normalize_thumbnail_bytes(image_bytes, aspect_ratio)
     except Exception as exc:
         raise MediaGenerationError(str(exc)) from exc
     destination.parent.mkdir(parents=True, exist_ok=True)
@@ -317,9 +324,8 @@ def _image_endpoint(card: Mapping[str, Any]) -> str:
     raise MediaGenerationError(f"O provider {card.get('provider')} não tem endpoint de imagem configurado.")
 
 
-def _thumbnail_aspect_ratio(thumbnail_blueprint: Mapping[str, Any] | None) -> str:
-    content = str((thumbnail_blueprint or {}).get("content") or "")
-    return "9:16" if re.search(r"\b9\s*:\s*16\b|VERTICAL", content, flags=re.IGNORECASE) else "16:9"
+def _thumbnail_aspect_ratio(thumbnail_blueprint: Mapping[str, Any] | None, prompt: str = "") -> str:
+    return infer_thumbnail_aspect_ratio(prompt, dict(thumbnail_blueprint or {}))
 
 
 def _image_request(card: dict[str, Any], prompt: str, *, topic: str = "", lettering_text: str = "", lettering_prompt: str = "", thumbnail_blueprint: Mapping[str, Any] | None = None) -> Any:
@@ -332,7 +338,7 @@ def _image_request(card: dict[str, Any], prompt: str, *, topic: str = "", letter
         lettering_text=lettering_text,
         lettering_prompt=lettering_prompt,
     )
-    aspect_ratio = _thumbnail_aspect_ratio(thumbnail_blueprint)
+    aspect_ratio = _thumbnail_aspect_ratio(thumbnail_blueprint, prompt)
     requested_size = ("1024x1792" if aspect_ratio == "9:16" else "1792x1024") if provider == "pollinations" else ("720x1280 minimum" if aspect_ratio == "9:16" else "1280x720 minimum")
     constrained_prompt = _append_generation_constraints(
         image_prompt,
@@ -433,7 +439,7 @@ def generate_image_for_card(
             )
         except Exception as exc:
             raise MediaGenerationError(str(exc)) from exc
-    aspect_ratio = _thumbnail_aspect_ratio(thumbnail_blueprint)
+    aspect_ratio = _thumbnail_aspect_ratio(thumbnail_blueprint, prompt)
     if provider == "nano_banana":
         merged = dict(settings)
         merged["gemini_image_api_key"] = _api_key(card)
@@ -454,6 +460,7 @@ def generate_image_for_card(
                 lettering_text=lettering_text,
                 lettering_prompt=lettering_prompt,
                 reference_image=reference_image,
+                aspect_ratio=aspect_ratio,
             )
         except Exception as exc:
             raise MediaGenerationError(str(exc)) from exc
@@ -486,7 +493,7 @@ def generate_image_for_card(
             )
             buffer = io.BytesIO()
             image.save(buffer, format="PNG")
-            return _download_or_write(buffer.getvalue(), "", destination, card)
+            return _download_or_write(buffer.getvalue(), "", destination, card, aspect_ratio=aspect_ratio)
         except Exception as exc:
             raise MediaGenerationError(f"Hugging Face text-to-image falhou: {str(exc)[:240]}") from exc
 
@@ -514,7 +521,7 @@ def generate_image_for_card(
         elif request_id and style == "kie":
             urls = _poll_kie_task(routed.card, request_id, endpoint="/jobs/recordInfo", veo=False)
             url = urls[0] if urls else ""
-    return _download_or_write(image_bytes, url, destination, routed.card)
+    return _download_or_write(image_bytes, url, destination, routed.card, aspect_ratio=aspect_ratio)
 
 
 def _is_retryable_media_error(exc: BaseException) -> bool:

package/hermes_ui/pipeline_worker.py

@@ -32,7 +32,7 @@ from hermes_ui.llm_providers import active_llm_card, provider_definition
 from hermes_ui.media_generation import MediaGenerationError, _append_generation_constraints, generate_image_from_pool, generate_video_from_pool
 from hermes_ui.media_providers import FULL_IA_VIDEO_PROVIDER_CODES, media_cards_for_pool, media_provider_definition
 from hermes_ui.material_sources import material_api_keys, material_source_cards, selected_material_source
-from hermes_ui.thumbnail_generation import ThumbnailGenerationError, generate_thumbnail_image
+from hermes_ui.thumbnail_generation import ThumbnailGenerationError, generate_thumbnail_image, infer_thumbnail_aspect_ratio
 from hermes_ui.thumbnail_blueprints import thumbnail_blueprint_for_channel
 from hermes_ui.voice_preview import synthesize_preview
 
@@ -97,6 +97,7 @@ def _generate_pipeline_thumbnail(
             variant_index=variant_index,
             lettering_text=lettering_text,
             lettering_prompt=lettering_prompt,
+            aspect_ratio=infer_thumbnail_aspect_ratio(prompt, thumbnail_blueprint),
         )
     return generate_image_from_pool(
         settings,

package/hermes_ui/thumbnail_generation.py

@@ -31,6 +31,22 @@ class ThumbnailGenerationError(RuntimeError):
     """Raised when Nano Banana cannot produce a thumbnail image."""
 
 
+def infer_thumbnail_aspect_ratio(
+    prompt: str = "",
+    blueprint: dict[str, Any] | None = None,
+    *,
+    fallback: str = DEFAULT_ASPECT_RATIO,
+) -> str:
+    """Resolve the requested thumbnail orientation without trusting provider defaults."""
+    blueprint_content = str((blueprint or {}).get("content") or "")
+    source = f"{blueprint_content}\n{str(prompt or '')}"
+    if re.search(r"\b9\s*:\s*16\b|\bvertical\b|\bportrait\b", source, flags=re.IGNORECASE):
+        return "9:16"
+    if re.search(r"\b16\s*:\s*9\b|\blandscape\b|\bwide\b", source, flags=re.IGNORECASE):
+        return "16:9"
+    return fallback if fallback in {"16:9", "9:16"} else DEFAULT_ASPECT_RATIO
+
+
 def normalize_thumbnail_bytes(image_bytes: bytes, aspect_ratio: str = DEFAULT_ASPECT_RATIO) -> bytes:
     """Return a non-distorted JPEG in the requested thumbnail orientation."""
     try:
@@ -188,6 +204,7 @@ def generate_thumbnail_image(
         lettering_text=lettering_text,
         lettering_prompt=lettering_prompt,
     )
+    effective_aspect_ratio = infer_thumbnail_aspect_ratio(clean_prompt, fallback=aspect_ratio)
 
     model = str(settings.get("gemini_image_model") or DEFAULT_GEMINI_IMAGE_MODEL).strip()
     image_size = DEFAULT_IMAGE_SIZE
@@ -203,7 +220,7 @@ def generate_thumbnail_image(
         "response_format": {
             "type": "image",
             "mime_type": DEFAULT_MIME_TYPE,
-            "aspect_ratio": aspect_ratio,
+            "aspect_ratio": effective_aspect_ratio,
             "image_size": image_size,
         },
         "store": False,
@@ -230,7 +247,7 @@ def generate_thumbnail_image(
         raise ThumbnailGenerationError("A API Nano Banana devolveu uma resposta que não é JSON.") from exc
     if str(payload.get("status") or "completed").lower() not in {"completed", "succeeded"}:
         raise ThumbnailGenerationError(f"A interação Nano Banana terminou com estado inesperado: {payload.get('status') or 'desconhecido'}.")
-    image_bytes = normalize_thumbnail_bytes(_extract_image_bytes(payload), aspect_ratio)
+    image_bytes = normalize_thumbnail_bytes(_extract_image_bytes(payload), effective_aspect_ratio)
 
     ensure_storage()
     output_dir = STORAGE / "thumbnails"
@@ -258,6 +275,7 @@ __all__ = [
     "DEFAULT_ASPECT_RATIO",
     "DEFAULT_GEMINI_IMAGE_MODEL",
     "DEFAULT_IMAGE_SIZE",
+    "infer_thumbnail_aspect_ratio",
     "THUMBNAIL_HEIGHT",
     "THUMBNAIL_WIDTH",
     "GEMINI_INTERACTIONS_ENDPOINT",

package/hermes_ui/thumbnails.py

@@ -11,7 +11,7 @@ from .creative_generation import _language_instruction, generate_thumbnail_promp
 from .media_generation import generate_image_from_pool
 from .media_providers import media_cards_for_pool
 from .storage import STORAGE, ensure_storage, now, read_json, update_json, write_json
-from .thumbnail_generation import ThumbnailGenerationError, generate_thumbnail_image
+from .thumbnail_generation import ThumbnailGenerationError, generate_thumbnail_image, infer_thumbnail_aspect_ratio
 from .thumbnail_blueprints import thumbnail_blueprint_for_task
 
 
@@ -41,7 +41,7 @@ def _generate_image_with_pool(
             reference_image=reference_image,
             lettering_text=lettering_text,
             lettering_prompt=lettering_prompt,
-            aspect_ratio="9:16" if "9:16" in str((thumbnail_blueprint or {}).get("content") or "") else "16:9",
+            aspect_ratio=infer_thumbnail_aspect_ratio(prompt, thumbnail_blueprint),
         )
     return generate_image_from_pool(
         settings,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@danhachuel/thunderbolt",
-  "version": "0.7.39",
+  "version": "0.7.40",
   "description": "Thunderbolt — interface local para operação de canais faceless e motor MoneyPrinterTurbo",
   "license": "MIT",
   "main": "scripts/cli.mjs",