Ë
    o…pjk  ã            	      óÜ   — U d Z ddlmZ ddlmZmZmZ 	 ddlmZ dddd	œZ
d
ddd	œZh d£ZdZdZddddœddddœddddœddddœddddœdd d!dœgZd"ed#<   	 	 	 	 	 	 d'd$„Zy%# e	$ r d&d„ZY ŒOw xY w)(z­Model catalog and aspect ratio resolution for Abacus AI image generation.

Defines the supported model catalog, model-specific aspect ratio formats,
and resolution helpers.
é    )Úannotations)ÚAnyÚDictÚList)Úresolve_aspect_ratioc                ó   — | S )N© )Úrs    ú3/root/.hermes/plugins/image_gen/abacus_ai/models.pyr   r      s   € °A¨Xó    Ú	square_hdÚlandscape_16_9Úportrait_16_9)ÚsquareÚ	landscapeÚportraitú1:1z16:9z9:16>   ÚdalleÚseedreamÚ
midjourneyÚflux_pro_ultrag     Àr@Ú	flux2_proÚnano_banana_prozNano Banana Proz.Highest resolution (up to 4K), Google DeepMind)ÚidÚdisplayÚ	strengthsr   zFLUX Pro UltrazHighest quality, photorealisticz
FLUX 2 Proz&High quality, photorealistic (default)Úflux2zFLUX 2zFast, good qualityr   Ú
MidJourneyzArtistic, stylistic generationsr   zDALL-Ez!Creative, strong prompt adherencezList[Dict[str, Any]]ÚMODEL_CATALOGc                ó  — t        | «      }|j                  «       j                  dd«      j                  dd«      }t        D ].  }|j                  dd«      |v sŒt        j                  |d«      c S  t        j                  |d«      S )a  Resolve the aspect ratio string for a specific model.

    Different models use different aspect ratio formats:

    * FLUX models: ``square_hd``, ``landscape_16_9``, ``portrait_16_9``
    * MidJourney / DALL-E / FLUX Pro Ultra: ``1:1``, ``16:9``, ``9:16``
    Ú_Ú Ú-r   r   )r   ÚlowerÚreplaceÚ_STANDARD_ASPECT_MODELSÚ_STANDARD_ASPECT_RATIOSÚgetÚ_FLUX_ASPECT_RATIOS)Úaspect_ratioÚmodel_idÚaspectÚ	model_keyÚ	std_models        r   Úresolve_aspect_ratio_for_modelr/   O   s‚   € ô " ,Ó/€Fð —‘Ó ×(Ñ(¨¨bÓ1×9Ñ9¸#¸rÓB€IÜ,ò >ˆ	Ø×Ñ˜S "Ó%¨Ò2Ü*×.Ñ.¨v°uÓ=Ò=ð>ô
 ×"Ñ" 6¨;Ó7Ð7r   N)r
   ÚstrÚreturnr0   )r*   r0   r+   r0   r1   r0   )Ú__doc__Ú
__future__r   Útypingr   r   r   Úagent.image_gen_providerr   ÚImportErrorr)   r'   r&   Ú_REQUEST_TIMEOUTÚDEFAULT_MODELr   Ú__annotations__r/   r	   r   r   ú<module>r:      sù   ðòõ #ç "Ñ "ð6Ý=ð Ø!ØñÐ ð ØØñÐ ò PÐ àÐ à€ð  Ø$ØEñð Ø#Ø6ñð ØØ=ñð ØØ)ñð ØØ6ñð ØØ8ñð5'€Ð#ó ðD8Øð8Ø!$ð8àô8øðE ò 6Ý5ð6ús   •A Á	A+Á*A+