o Ďi#@sddlZddlZddlZddlZddlmZddlmZddlmZddl Z ddl m Z ddl m Z mZmZmZddlmZddlmZdd lmZmZdd lmZd Zdd d ZdddZdadddZddZdddZ GdddZ!dS)N)getpass)Path)urlparse)rfapi)API_URLAPP_URL DEMO_KEYSload_roboflow_api_key)Project) Workspace) CLIPModel GazeModel) write_linez1.2.13cCst|tstdttdtt|tdd|Dr^|tvr$|Sttd|}|j dkr7t|j |j dkrX|dkrTt d t d |d 7}t||||Std |}|Stjd td |d|ddS)Nz.API Key is of Incorrect Type Expected Type: z Input Type: css|] }|r|VqdSN)islower).0crM/home/jeff/fluffinator/venv/lib/python3.10/site-packages/roboflow/__init__.py szcheck_key..z /?api_key=iz retrying...z?There was an error validating the api key with Roboflow server.z8upload and label your dataset, and get an API KEY here: z/?model=z&ref= onboarding) isinstancestr RuntimeErrortypeanyrrequestspostr status_codetextprinttimesleep check_keyjsonsysstdoutwriter)api_keymodelnotebook num_retriesresponserrrrr'sF      r'Fc Cstj}|dkrttdd}n ttddd}tjd|d}tj|r3|s3tddStj|r@|r@t ||durMtdt d n tdt d |d t d }t t d |}|jdkr|}|durwtdtjtj|sttj|d|i}t|dd}|d|}|d|d<t|d}tj||ddWddS1swYdS|dS)Nntroboflowz config.jsonz.configROBOFLOW_CONFIG_DIR)defaultz_You are already logged into Roboflow. To make a different login,run roboflow.login(force=True).zvisit z+/auth-cli to get your authentication token.z/auth-cli/?workspace=z" to get your authentication token.z%Paste the authentication token here: z/query/cliAuthToken/rz9Invalid API key. Please check your API key and try again. workspacesrurl RF_WORKSPACEwindent)osnamerrhomegetenvpathisfilerremoverrr getr"r( ValueErrorexistsdirnamemakedirslistkeysopendumpraise_for_status) workspaceforceos_name default_path conf_locationtokenr_logindefault_workspace_idfrrrlogin?sB      rWcCs&|dur tatSt|atS)zHigh level function to initialize Roboflow. Args: the_workspace: the workspace url to initialize. If None, the default workspace will be used. Returns: None N)RoboflowrNactive_workspace the_workspacerrrinitialize_roboflowys   r\cCsft}d|vs d|vr t|}|jd}|d}t|d}ntd||}||}|j}|S)zHigh level function to load Roboflow models. Args: model_url: the model url to load. Must be from either app.roboflow.com or universe.roboflow.com Returns: the model object to use for inference universe.roboflow.comapp.roboflow.com/r:GModel URL must be from either app.roboflow.com or universe.roboflow.com) r\rrAsplitintrEprojectversionr-) model_urloperate_workspace parsed_url path_partsrdrer-rrr load_models    rjc Cstd|vsd|vr!t|}|jd}|d}t|d}|d}ntdt|d}||}||}|||S) aHigh level function to download data from Roboflow. Args: dataset_url: the dataset url to download. Must be from either app.roboflow.com or universe.roboflow.com model_format: the format the dataset will be downloaded in location: the location the dataset will be downloaded to Returns: The dataset object with location available as dataset.location r]r^r_r:r`rrarZ) rrArbrcrEr\rdredownload) dataset_url model_formatlocationrhrirdrer[rgrrrdownload_datasets        roc@s@eZdZ   d ddZddZdddZdd d Zd d ZdS)rXN undefinedcCs6||_|jdur t|_||_||_d|_|dSNF)r,r rmr.rauth)selfr,rmr.rrr__init__s  zRoboflow.__init__cCsHt|j|j|j}|dkrd|_|S|tvrd|_|S|d}||_|S)NrTrN)r'r,rmr.rruniversecurrent_workspace)rsr1r9rrrrrsz Roboflow.authcCs|tjdtjdtj|dur|j}|jr,|j}t||}t||||j S|jt vr:ti|j||j St d)Nz loading Roboflow workspace...rz!A valid API key must be provided.) r)r*r+flushrvr,r get_workspacer rmrrE)rsr[r, list_projectsrrrrNs     zRoboflow.workspacecCs|durd|vr|d}|d|d}}n|j}ttd|d|d|j}|jdkr6t|j| d}t |j|S)zFunction that takes in the name of the project and returns the project object :param project_name api_key: project name :param the_workspace workspace name :return project object Nr_rrz ?api_key=rrd) rsplitrvr rDrr,r"rr#r(r )rs project_namer[splitted_project dataset_inforrrrds $    zRoboflow.projectcCs|j|jd}tj|ddS)zto string function)r,rNr:r;)r,rNr(dumps)rs json_valuerrr__str__szRoboflow.__str__)Nrprpr)__name__ __module__ __qualname__rtrrrNrdrrrrrrXs    rX)rrqr)"r(r=r)r%rpathlibr urllib.parserr roboflow.adaptersrroboflow.configrrrr roboflow.core.projectr roboflow.core.workspacer roboflow.modelsr r roboflow.util.generalr __version__r'rWrYr\rjrorXrrrrs,         *7