File: //usr/local/lib/python3.10/dist-packages/openai/cli/_api/__pycache__/image.cpython-310.pyc
o
���g� � @ s� d dl mZ d dlmZmZmZ d dlmZ ddlm Z m
Z
ddl m Z m
Z
mZ ddlmZ dd lmZ er<d d
lmZ ddd�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� d�ZdS )� )� annotations)�
TYPE_CHECKING�Any�cast)�ArgumentParser� )�
get_client� print_model� )� NOT_GIVEN�NotGiven�
NotGivenOr)� BaseModel)� BufferReader)�_SubParsersAction� subparser�!_SubParsersAction[ArgumentParser]�return�Nonec C s� | � d�}|jddtd� |jddtdd� |jd d
td d � |jd
dtddd� |jdtdd � |jtjtd� | � d�}|jddtd� |jddtdd� |jd d
td d � |jddtddd� |jd
dtddd� |jdtdd � |jddtd d d� |jtjt d� | � d �}|jddtd� |jd d
td d � |jddtddd� |jd
dtddd� |jdtdd � |jtj
t d� d S ) Nzimages.generatez-mz--model)�typez-pz--promptT)r �requiredz-nz --num-images� )r �defaultz-sz--size� 1024x1024zSize of the output image)r r �helpz--response-format�url)�func�
args_modelz images.editz-Iz--imagez@Image to modify. Should be a local path and a PNG encoded image.)r r r z-Mz--maskFz�Path to a mask image. It should be the same size as the image you're editing and a RGBA PNG image. The Alpha channel acts as the mask.zimages.create_variation) �
add_parser� add_argument�str�int� set_defaults�CLIImage�create�CLIImageCreateArgs�edit�CLIImageEditArgs�create_variation�CLIImageCreateVariationArgs)r �sub� r+ �@/usr/local/lib/python3.10/dist-packages/openai/cli/_api/image.py�register sR
��
�r- c @ �: e Zd ZU ded< ded< ded< ded< eZded< d S )
r% r �promptr! �
num_images�size�response_format�NotGivenOr[str]�modelN��__name__�
__module__� __qualname__�__annotations__r r4 r+ r+ r+ r, r% = �
r% c @ r. )
r) r �imager! r0 r1 r2 r3 r4 Nr5 r+ r+ r+ r, r) E r: r) c @ sN e Zd ZU ded< ded< ded< ded< ded< eZded < eZded
< d S ) r' r r; r! r0 r1 r2 r/ r3 �maskr4 N)r6 r7 r8 r9 r r< r4 r+ r+ r+ r, r' M s
r' c @ s6 e Zd Zeddd��Zeddd ��Zedd d ��Zd
S )r# �argsr% r r c C s: t � jj| j| j| jtt| j�tt| j �d�}t
|� d S )N)r4 r/ �nr1 r2 ) r �images�generater4 r/ r0 r r r1 r2 r )r= r; r+ r+ r, r$ X s
� zCLIImage.creater) c C sx t | jd��}t|�� dd�}W d � n1 sw Y t� jj| jd|f| jt t
| j �t t
| j �d�}t
|� d S )N�rbzUpload progress��descr; )r4 r; r> r1 r2 )�openr; r �readr r? r( r4 r0 r r r1 r2 r )r= � file_reader�
buffer_readerr; r+ r+ r, r( e s �
� zCLIImage.create_variationr' c
C s� t | jd��}t|�� dd�}W d � n1 sw Y t| jt�r't}n t | jd��}t|�� dd�}W d � n1 s@w Y t� j j
| j | j d|f| j
t|t�s[d|fn|tt| j�tt| j�d�}t|� d S )NrA zImage upload progressrB z
Mask progressr; r< )r4 r/ r; r> r< r1 r2 )rD r; r rE �
isinstancer< r r r r? r& r4 r/ r0 r r r1 r2 r )r= rF rG r< r; r+ r+ r, r& u s$ � �
� z
CLIImage.editN)r= r% r r )r= r) r r )r= r' r r )r6 r7 r8 � staticmethodr$ r( r& r+ r+ r+ r, r# W s r# N)r r r r )�
__future__r �typingr r r �argparser �_utilsr r �_typesr r r
�_modelsr � _progressr r r- r% r) r' r# r+ r+ r+ r, �<module> s
.