%9j"dZddlZddlZddlZddlZddlmZddlZddlZddl Z ddl m Z ddl m Z ddl m Z  ddl m Z dZ ddlZdZd Zd Zej*d Zej*d Zej*d Zej*dZej*dZej*dZej*dZdZd:dZd:dZd:dZ d:dZ!d:dZ"d;dZ#d:dZ$GddZ%GddZ&Gdde&Z'Gd d!Z(Gd"d#Z)Gd$d%e(Z*Gd&d'e(Z+Gd(d)Z,Gd*d+Z-d,Z.d:d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9Gd8d9Z:y#e$rd ZY?wxYw#e$rd ZYFwxYw)[a-z0-9_.-]{3,63})/(?P.+)$zm^projects/(?P[a-z0-9-]{6,30})/locations/(?P[^/]+)/models/(?P[A-Za-z0-9]+)$zS^projects/(?P[a-z0-9-]{6,30})/models/(?P[A-Za-z0-9_-]{1,60})$z;^projects/(?P[a-z0-9-]{6,30})/operations/[^/]+$c@tj|ttS)a  Returns an _MLService instance for an App. Args: app: A Firebase App instance (or None to use the default App). Returns: _MLService: An _MLService for the specified App instance. Raises: ValueError: If the app argument is invalid. )rget_app_service _ML_ATTRIBUTE _MLServiceapps M/var/www/html/truck-me/venv/lib/python3.12/site-packages/firebase_admin/ml.py_get_ml_servicerBs  ! !#}j AAcdt|}tj|j||S)zCreates a model in the current Firebase project. Args: model: An ml.Model to create. app: A Firebase app instance (or None to use the default app). Returns: Model: The model that was created in Firebase ML. r)rModel from_dict create_modelmodelr ml_services rrrQ,!%J ??:2259s? CCrcdt|}tj|j||S)zUpdates a model's metadata or model file. Args: model: The ml.Model to update. app: A Firebase app instance (or None to use the default app). Returns: Model: The updated model. r)rrr update_modelrs rrr_rrcht|}tj|j|d|S)aPublishes a Firebase ML model. A published model can be downloaded to client apps. Args: model_id: The id of the model to publish. app: A Firebase app instance (or None to use the default app). Returns: Model: The published model. Tpublishrrrr set_publishedmodel_idrrs r publish_modelr$ms2!%J ??:33Hd3KQT? UUrcht|}tj|j|d|S)zUnpublishes a Firebase ML model. Args: model_id: The id of the model to unpublish. app: A Firebase app instance (or None to use the default app). Returns: Model: The unpublished model. Frrr r"s runpublish_modelr&}s2!%J ??:33He3LRU? VVrcdt|}tj|j||S)zGets the model specified by the given ID. Args: model_id: The id of the model to get. app: A Firebase app instance (or None to use the default app). Returns: Model: The requested model. r)rrr get_modelr"s rr(r(s,!%J ??://9s? CCrcLt|}t|j||||S)a0Lists the current project's models. Args: list_filter: a list filter string such as ``tags:'tag_1'``. None will return all models. page_size: A number between 1 and 100 inclusive that specifies the maximum number of models to return per page. None for default. page_token: A next page token returned from a previous page of results. None for first page of results. app: A Firebase app instance (or None to use the default app). Returns: ListModelsPage: A (filtered) list of models. r)rListModelsPage list_models) list_filter page_size page_tokenrrs rr+r+s.!%J  Y  MMrc<t|}|j|y)zDeletes a model from the current project. Args: model_id: The id of the model you wish to delete. app: A Firebase app instance (or None to use the default app). N)r delete_modelr"s rr0r0s!%JH%rcxeZdZdZddZeddZdZdZdZ e dZ e d Z e jd Z ed Ze d Ze d Ze dZe dZe dZe dZe dZejdZe dZddZe dZejdZddZy)raIA Firebase ML Model object. Args: display_name: The display name of your model - used to identify your model in code. tags: Optional list of strings associated with your model. Can be used in list queries. model_format: A subclass of ModelFormat. (e.g. TFLiteFormat) Specifies the model details. Ncfd|_i|_d|_|||_|||_|||_yyN)_app_data _model_format display_nametags model_format)selfr7r8r9s r__init__zModel.__init__sF  !  # ,D   DI  # ,D  $rct|}d}|jdd}|jdd|rtj|}t |}||_||_|S)-Create an instance of the object from a dict.N tfliteModel@typer9)dictpop TFLiteFormatrrr5r4)clsdatar data_copy tflite_formattflite_format_datars rrzModel.from_dictsbJ  &]]=$? gt$ (223EFM=1   rcrtj|}|j|_|j|_yr3)rrr9r5)r:rEcopys r_update_from_dictzModel._update_from_dicts)t$ --ZZ rct||jr4|j|jk(xr|j|jk(SyNF) isinstance __class__r5r6r:others r__eq__z Model.__eq__< eT^^ ,::,Z1C1CuGZGZ1Z Zrc&|j| Sr3rRrPs r__ne__z Model.__ne__;;u%%%rc|jjdsyt|jjd\}}|S)z&The model's ID, unique to the project.nameN)r5get_validate_and_parse_name)r:_r#s rr#zModel.model_ids7zz~~f%.tzz~~f/EF 8rc8|jjdS)zaThe model's display name, used to refer to the model in code and in the Firebase console. displayNamer5rZr:s rr7zModel.display_nameszz~~m,,rc6t||jd<|S)Nr^)_validate_display_namer5)r:r7s rr7zModel.display_names$:<$H =! rc|syd}tjjd}tjj||}t||z j dz}|S)Nz%Y-%m-%dT%H:%M:%S.%fZri)datetimeutcfromtimestampstrptimeint total_seconds) date_string format_strepochdatetime_objectmilliss r_convert_to_milliszModel._convert_to_milliss], !!2215"++44[*Mo-<<>EF rc`tj|jjddS)zThe time the model was created. createTimeNrrnr5rZr`s r create_timezModel.create_time$'' |T(JKKrc`tj|jjddS)z$The time the model was last updated. updateTimeNrqr`s r update_timezModel.update_time rsrcx|jjdijdijdS)zValidation error message.statevalidationErrormessager_r`s rvalidation_errorzModel.validation_errors2zz~~gr*../@"EII)TTrcjt|jjdijdS)zMTrue if the model is published and available for clients to download.rx published)boolr5rZr`s rr}zModel.publisheds)DJJNN7B/33K@AArc8|jjdS)z,The entity tag (ETag) of the model resource.etagr_r`s rrz Model.etagzz~~f%%rc8|jjdS)z SHA256 hash of the model binary. modelHashr_r`s r model_hashzModel.model_hashzz~~k**rc8|jjdS)z.Tag strings, used for filtering query results.r8r_r`s rr8z Model.tags#rrc6t||jd<|S)Nr8)_validate_tagsr5)r:r8s rr8z Model.tags(s+D1 6 rct|jjdxr't|jjddkDS)z:True if the Model object is locked by an active operation.activeOperationsr)r~r5rZlenr`s rlockedz Model.locked-sDDJJNN#56@ '9:;a?A Arc |jsyt|j}|jj ddj d}|j |j |d|}|j|y)aQWaits for the model to be unlocked. (All active operations complete) Args: max_time_seconds: The maximum number of seconds to wait for the model to unlock. (None for no limit) Raises: exceptions.DeadlineExceeded: If max_time_seconds passed and the model is still locked. NrrrYT)wait_for_operationmax_time_seconds)rrr4r5rZhandle_operation get_operationrK)r:rrop_name model_dicts rwait_for_unlockedzModel.wait_for_unlocked3sx{{ $TYY/ **..!34Q7;;FC00  $ $W -#-1/  z*rc|jS)zeThe model's ``ModelFormat`` object, which represents the model's format and storage location.)r6r`s rr9zModel.model_formatGs!!!rc.| t|||_|Sr3)_validate_model_formatr6)r:r9s rr9zModel.model_formatMs  # "< 0) rct|j}|jr+|j|jj ||S)4Returns a serializable representation of the object. for_upload)rAr5r6updateas_dictr:rrJs rrz Model.as_dictTs>DJJ    KK**22j2I J r)NNNr3F)__name__ __module__ __qualname____doc__r; classmethodrrKrRrVpropertyr#r7setter staticmethodrnrrrvr{r}rrr8rrr9rrrrrs -    &-- LLLLUUBB &&++&& [[AA +(""  rrceZdZdZddZy) ModelFormatz?Abstract base class representing a Model Format such as TFLite.ctrNotImplementedErrorr:rs rrzModelFormat.as_dict^!!rNrrrrrrrrrrr\s I"rrceZdZdZd dZedZdZdZe dZ e dZ e jd Z e d Zd d Zy)rCzModel format representing a TFLite model. Args: model_source: A TFLiteModelSource sub class. Specifies the details of the model source. Nc4i|_d|_|||_yyr3)r5 _model_source model_sourcer:rs rr;zTFLiteFormat.__init__is$ !  # ,D  $rc`t|}t|j|}||_|S)r=)r)rArC_init_model_sourcer5)rDrErFrGs rrzTFLiteFormat.from_dictps0J $#2H2H2ST ' rct||jr4|j|jk(xr|j|jk(SyrM)rNrOr5rrPs rrRzTFLiteFormat.__eq__xrSrc&|j| Sr3rUrPs rrVzTFLiteFormat.__ne__~rWrc|jdd}|r t|S|jdd}|r&tjdtt |Sy)zInitialize the ML model source. gcsTfliteUriN)gcs_tflite_uri automlModelQAutoML model support is deprecated and will be removed in the next major version.) auto_ml_model)rBTFLiteGCSModelSourcewarningswarnDeprecationWarningTFLiteAutoMlSource)rErrs rrzTFLiteFormat._init_model_sourcesX.$7 '~F F5  MM+,> @%MB Brc|jS)zThe TF Lite model's location.)rr`s rrzTFLiteFormat.model_sources!!!rcL|t|ts td||_y)Nz0Model source must be a TFLiteModelSource object.)rNTFLiteModelSource TypeErrorrrs rrzTFLiteFormat.model_sources'  #l,=> RSS)rc8|jjdS)z'The size in bytes of the TF Lite model. sizeBytesr_r`s r size_byteszTFLiteFormat.size_bytesrrct|j}|jr+|j|jj |d|iS)rrr>)rAr5rrrrs rrzTFLiteFormat.as_dictsDDJJ    KK**22j2I Jt$$rr3r)rrrrr;rrrRrVrrrrrrrrrrrCrCcs - &  ""** ++%rrCceZdZdZddZy)rzIAbstract base class representing a model source for TFLite format models.ctrrrs rrzTFLiteModelSource.as_dictrrNrrrrrrrs S"rrcXeZdZdZdZdZedZedZedZ edZ y) _CloudStorageClientzCloud Storage helper classz gs://{0}/{1}zFirebase/ML/Models/{0}c&ts tdy)NznFailed to import the Cloud Storage library for Python. Make sure to install the "google-cloud-storage" module.) _GCS_ENABLED ImportErrorrrr_assert_gcs_enabledz'_CloudStorageClient._assert_gcs_enabledsNO Orctj|}|s td|jd|jdfS)N!GCS TFLite URI format is invalid. bucket_name blob_name)_GCS_TFLITE_URI_PATTERNmatch ValueErrorgroup)urimatchers r_parse_gcs_tflite_uriz)_CloudStorageClient._parse_gcs_tflite_uris@*//4@A A}}]+W]];-GGGrcltjtjj |}t j ||}tjj|}|j|}|j|tjj|j|S)z4Upload a model file to the specified Storage bucket.r) rrospathbasenamerbucket BLOB_NAMEformatblobupload_from_filenameGCS_URIrY)rmodel_file_namer file_namerrrs ruploadz_CloudStorageClient.uploads //1GG$$_5  5'1188C {{9% !!/2"**11&++yIIrctjtj|\}}tj||}|j |}|j dtjddS)zHMakes the gcs_tflite_uri readable for GET for 10 minutes via signed_uri.rv4 )minutesGET)version expirationmethod) rrrrrrgenerate_signed_urlrd timedelta)rrrrrrs rsign_uriz_CloudStorageClient.sign_uriso //1!4!J!J>!Z Y 5{{9%''))"5(  rN) rrrrrrrrrrrrrrrrsc$G(IOO HH J J    rrceZdZdZeZddZdZdZe ddZ e dZ e dZ e d Ze dd Ze dd Zed Zej&d ZdZddZy)rzCTFLite model source representing a tflite model file stored in GCS.Nc2||_t||_yr3)r4_validate_gcs_tflite_uri_gcs_tflite_uri)r:rrs rr;zTFLiteGCSModelSource.__init__s 7Grcbt||jr|j|jk(SyrM)rNrOrrPs rrRzTFLiteGCSModelSource.__eq__s* eT^^ ,''5+@+@@ @rc&|j| Sr3rUrPs rrVzTFLiteGCSModelSource.__ne__rWrc^tjj|||}t||S)aUploads the model file to an existing Google Cloud Storage bucket. Args: model_file_name: The name of the model file. bucket_name: The name of an existing bucket. None to use the default bucket configured in the app. app: A Firebase app instance (or None to use the default app). Returns: TFLiteGCSModelSource: The source created from the model_file Raises: ImportError: If the Cloud Storage Library has not been installed. )rr)r_STORAGE_CLIENTr)rDrrrgcs_uris rfrom_tflite_model_filez+TFLiteGCSModelSource.from_tflite_model_files, '66==k?\_`#7DDrc0ts tdtjjj ds\tjjj ds2tdj tjjyy)Nz_Failed to import the tensorflow library for Python. Make sure to install the tensorflow module.1.z2.z5Expected tensorflow version 1.x or 2.x, but found {0}) _TF_ENABLEDrtfrVERSION startswithrrrr_assert_tf_enabledz'TFLiteGCSModelSource._assert_tf_enabledswBC Czz!!,,T22::;M;M;X;XY];^U%vbjj&8&89; ;<_2rcttjjj|}|j Sr3)rliteTFLiteConverterfrom_saved_modelconvert)saved_model_dir converters r_tf_convert_from_saved_modelz1TFLiteGCSModelSource._tf_convert_from_saved_model s-GG++<<_M   ""rctjjjdred}tjj j ||tjjj|}|jStjjj|}|jS)z4Converts the given Keras model into a TF Lite model.rzfirebase_keras_model.h5) rrrrkerasmodels save_modelr r from_keras_model_filefrom_keras_modelr) keras_model keras_filers r_tf_convert_from_keras_modelz1TFLiteGCSModelSource._tf_convert_from_keras_models ::   ( ( .2J HHOO & &{J ?//EEjQI  ""//@@MI  ""rctjtj|}t|d5}|j |dddtj |||S#1swY xYw)aCreates a Tensor Flow Lite model from the saved model, and uploads the model to GCS. Args: saved_model_dir: The saved model directory. model_file_name: The name that the tflite model will be saved as in Cloud Storage. bucket_name: The name of an existing bucket. None to use the default bucket configured in the app. app: Optional. A Firebase app instance (or None to use the default app) Returns: TFLiteGCSModelSource: The source created from the saved_model_dir Raises: ImportError: If the Tensor Flow or Cloud Storage Libraries have not been installed. wbN)rr ropenwriter)rDrrrr tflite_model model_files rr z%TFLiteGCSModelSource.from_saved_modelse$ //1+HHY /4 ( +J   \ * +#::?KY\]] + + A''A0ctjtj|}t|d5}|j |dddtj |||S#1swY xYw)aCreates a Tensor Flow Lite model from the keras model, and uploads the model to GCS. Args: keras_model: A tf.keras model. model_file_name: The name that the tflite model will be saved as in Cloud Storage. bucket_name: The name of an existing bucket. None to use the default bucket configured in the app. app: Optional. A Firebase app instance (or None to use the default app) Returns: TFLiteGCSModelSource: The source created from the keras_model Raises: ImportError: If the Tensor Flow or Cloud Storage Libraries have not been installed. rN)rr rrrr)rDrrrrrr s rrz%TFLiteGCSModelSource.from_keras_model5se$ //1+HHU /4 ( +J   \ * +#::?KY\]] + +r!c|jS)z'URI of the model file in Cloud Storage.)rr`s rrz#TFLiteGCSModelSource.gcs_tflite_uriMs###rc$t||_yr3)rr)r:rs rrz#TFLiteGCSModelSource.gcs_tflite_uriRs7Grcjtjj|j|jS)zQSigns the GCS uri, so the model file can be uploaded to Firebase ML and verified.)rrrrr4r`s r_get_signed_gcs_tflite_uriz/TFLiteGCSModelSource._get_signed_gcs_tflite_uriVs(#33<z!_validate_tags.. s 1SJsC 1szTags must be a list of strings.c3FK|]}tj|ywr3) _TAG_PATTERNrrgs rrjz!_validate_tags.. s73|!!#&7s!zTag format is invalid.)rNlistallrr)r8s rrrsE dD ! 1D 119:: 7$7 7122 KrcFtj|s td|S)Nr)rrr)rs rrrs# # ( ( -<== JrcFtj|s td|S)NrV)_AUTO_ML_MODEL_PATTERNrr)rs rr0r0s! ! ' ' .ABB Lrc<t|ts td|S)Nz*Model format must be a ModelFormat object.)rNrrr@s rrrs lK 0DEE rc@|t|ts tdyy)Nz%List filter must be a string or None.rNrfr)r,s r_validate_list_filterru$s(+s+CD D,rc|It|tur td|dks |tkDrt dj tyy)Nz#Page size must be a number or None.rMz6Page size must be a positive integer between 1 and {0})typergr_MAX_PAGE_SIZErr)r-s r_validate_page_sizery*sV  ?# %AB B q=I6))/)?A A7 rc@|t|ts tdyy)Nz$Page token must be a string or None.rt)r.s r_validate_page_tokenr{4s(*c*BC C+rc`eZdZdZdZdZdZdZdZdZ dZ dd Z d Z dd Z d ZdZdZdZy )r zFirebase ML service.z7https://firebaseml.googleapis.com/v1beta2/projects/{0}/z*https://firebaseml.googleapis.com/v1beta2/g?c|j|_|js tdtjj |j|_ddj tji}tj|jj||j |_ tj|jj|tj|_y)NzqProject ID is required to access ML service. Either set the projectId option, or use service account credentials.zX-FIREBASE-CLIENTzfire-admin-python/{0}) credentialheadersbase_url)rW _project_idrr PROJECT_URLr _project_urlfirebase_admin __version__rJsonHttpClientrget_credential_client OPERATION_URL_operation_client)r:r ml_headerss rr;z_MLService.__init__Bs>>HI I'2299$:J:JK !8!?!?@Z@Z![ $22~~446&&( ".!ModelOperationMetadataz#Unknown type of operation metadata.rN)secondsrM)rNrArrZrhandle_operation_errorr UnknownErrorraendswithr[rdrrrrr(r) r: operationrrrr metadata_typer\r#r start_timers rrz_MLService.handle_operationhs&)T*=> > == }}Z( }}Z00}}W%33IMM'4JKK))2XY Y*9==+@A==R0 Wb1 %%&>?AB B.x||F/CD 8&&**, -5T("4"4=M"NN  v)>  % %oy A**73I q O !v)> == }}Z( }}Z00}}W%33IMM'4JKK"**,,rc t| |j|jjdd|j dS#t j j$r}tj|d}~wwxYw)NpostrTrrjson) r[rrrrrrrrr)r:rrs rrz_MLService.create_modelsy D(( !!&hU]]VZ]=[!\^ ^""33 D<>2  "+22;??D D(( !!'t%--SW-:X!Y[ [""33 D<EI#: )D D<<$$U$5 5""33 D<rs '!%&L K BJJ78" #;<rzz23 $"**BD#*+$Z\$"**BD B D D V W DM&&bbJ""A%;A%H"", , ^B6,B6J 4* 4F?$?$D>B<8     E AD fDfDeLKs#E5F5F?FF F