
    Pi"                        d dl Z d dlmZ d dlmZmZmZmZmZm	Z	m
Z
mZ d dlmZmZ d dlmZmZmZ d dlmZ d dlmZ  e j        e          Zded	ed
dfdZ	 ddddedededz  dedz  d
df
dZ	 ddddededededz  d
df
dZ	 ddddedededz  dededz  d
dfdZded	ed
dfdZ ded	ed
dfdZ!ded	ed
edz  fdZ"dS )    N)
HTTPStatus)ApiCreateModelInstanceRequest!ApiCreateModelInstanceRequestBody$ApiCreateModelInstanceVersionRequest(ApiCreateModelInstanceVersionRequestBodyApiCreateModelRequestApiDeleteModelRequestApiGetModelRequestCreateModelSigningTokenRequest)BackendErrorbuild_kaggle_client)KaggleApiHTTPErrorhandle_callhandle_mutate_call)UploadDirectoryInfo)ModelHandle
owner_slug
model_slugreturnc                    t                      5 t                      | _        |_        |_        d_        t          fd           t                              d| d           d d d            d S # 1 swxY w Y   d S )NTc                  B     j         j                                      S N)modelsmodel_api_clientcreate_model
api_clientrs   l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/kagglehub/models_helpers.py<lambda>z_create_model.<locals>.<lambda>   s    :#4#E#R#RST#U#U     Model 'z
' Created.)	r   r   r   slugtitle
is_privater   loggerinfo)r   r   r   r   s     @@r   _create_modelr(      s    			 6*!##!UUUUUVVV4j4445556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s   AA::A>A>Fsigstoremodel_handle
upload_dirlicense_namer*   c                L   |                                 }t                      5 t                      | j        _        | j        _        t                      _        | j	        j        _
        |                                 j        _        |j        j        _        |j        j        _        |j        _        ||j        _        t#          fd           t$                              d|                                             d d d            d S # 1 swxY w Y   d S )Nc                  B     j         j                                      S r   )r   r   create_model_instancer   s   r   r    z(_create_model_instance.<locals>.<lambda>7   s    :#4#E#[#[\]#^#^ r!   zKYour model instance has been created.
Files are being processed...
See at: )to_protor   r   ownerr   modelr   r   body	variationinstance_slugframework_enum	frameworkfilesdirectoriesr*   r-   r   r&   r'   to_url)r+   r,   r-   r*   upload_protor   r   s        @@r   _create_model_instancer=   "   sN    &&((L			 
*)++#)#)244+5'6688#))5"#".AF^^^^^___s\h\o\o\q\qss	
 	
 	

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
s   C'DD D version_notesc                4   |                                 }t                      5 t                      | j        _        | j        _        |                                 _        | j	        _
        t                      _        |j        _        |j        j        _        |j        j        _        |j        _        t#          fd           t$                              d|                                             d d d            d S # 1 swxY w Y   d S )Nc                  B     j         j                                      S r   )r   r   create_model_instance_versionr   s   r   r    z0_create_model_instance_version.<locals>.<lambda>R   s    :#4#E#c#cde#f#f r!   zSYour model instance version has been created.
Files are being processed...
See at: )r1   r   r   r2   r   r3   r   r7   r8   r5   r6   r   r4   r?   r9   r:   r*   r   r&   r'   r;   )r+   r,   r?   r*   r<   r   r   s        @@r   _create_model_instance_versionrC   >   sK    &&((L			 
*022#)#)"1133&09;;,#))5"fffffgggM5A5H5H5J5JM M	
 	
 	

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
s   CDDDr9   c                    	 t          | |||           d S # t          $ r6}|j        t          j        k    rt          | |||           n|Y d }~d S d }~ww xY w)Nr)   )r=   r   
error_coder   CONFLICTrC   )r+   r9   r-   r?   r*   es         r    create_model_instance_or_versionrH   Y   s    |UL8TTTTTT   <:...*<X`aaaaaI baaaaas    
A+AAc                    	 t                      5 t                      | _        |_        t	          fd           d d d            d S # 1 swxY w Y   d S # t
          $ rz}|j        f|j        j        t          j	        k    s|j        j        t          j
        k    r2t                              d| d|  d           t          | |           n|Y d }~d S d }~wt          $ r}|d }~ww xY w)Nc                  B     j         j                                      S r   )r   r   	get_modelr   s   r   r    z)create_model_if_missing.<locals>.<lambda>q   s    
 1 B L LQ O O r!   r"   z2' does not exist or access is forbidden for user 'z '. Creating or handling Model...)r   r
   r   r   r   r   responsestatus_coder   	NOT_FOUND	FORBIDDENr&   r'   r(   	Exceptionr   r   rG   r   r   s      @@r   create_model_if_missingrR   k   s    "" 	Qj"$$A%AL%ALOOOOOPPP		Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q 	Q
  
 
 
:!J"j&:::z%)===KK E*  E  EXb  E  E  E   *j1111I 211111    sG   A 0AA AA AA 
C/&A/CC/(C**C/c                 r   	 t                      5 t                      | _        |_        t	          fd           d d d            d S # 1 swxY w Y   d S # t
          $ rP}|j        <|j        j        t          j	        k    r"t                              d| d|  d           n|Y d }~d S d }~ww xY w)Nc                  B     j         j                                      S r   )r   r   delete_modelr   s   r   r    zdelete_model.<locals>.<lambda>   s    z'8'I'V'VWX'Y'Y r!   zCould not delete Model '' for user 'z'...)r   r	   r   r   r   r   rL   rM   r   rN   r&   r'   rQ   s      @@r   rU   rU      s;   
 "" 	[j%''A%AL%ALYYYYYZZZ		[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[ 	[
    :!aj&<
@T&T&TKK[:[[:[[[\\\\I ]\\\\\s;   A 0AA AA AA 
B6&AB11B6c                 x   	 t                      5 t                      | _        |_        t	          fd          }|j        cddd           S # 1 swxY w Y   dS # t          $ rM}|j        ;|j        j        t          j
        k    r!t                              d| d|  d           Y d}~dS d}~ww xY w)zFReturns a JWT for signing if authorized for /{owner_slug}/{model_slug}c                  B     j         j                                      S r   )r   r   create_model_signing_tokenr   s   r   r    zsigning_token.<locals>.<lambda>   s    :+<+M+h+hij+k+k r!   Nz'Could not get Signing token for Model 'rV   z'. Skipping signing...r>   )r   r   r   r   r   id_tokenr   rL   rM   r   rN   r&   r'   )r   r   rL   rG   r   r   s       @@r   signing_tokenr[      s&    "" 	%j.00A%AL%AL"#k#k#k#k#kllH$	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	%    :!aj&<
@T&T&TKKt*ttR\ttt   rrrrrs;   A" 6AA" AA" AA" "
B9,AB44B9r   )r>   )#logginghttpr   (kagglesdk.models.types.model_api_servicer   r   r   r   r   r	   r
   r   kagglehub.clientsr   r   kagglehub.exceptionsr   r   r   kagglehub.gcs_uploadr   kagglehub.handler   	getLogger__name__r&   strr(   boolr=   rC   rH   rR   rU   r[    r!   r   <module>rh      s         	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 @ ? ? ? ? ? ? ? T T T T T T T T T T 4 4 4 4 4 4 ( ( ( ( ( (		8	$	$6c 6s 6t 6 6 6 6  $

 "
 
 

#
 *

 Tk
 

 
 
 
> 

 "
 
 

#
 

 Tk
 

 
 
 
> 	 "   * 	 Tk 
   $      ,S c d    c s sTz      r!   