o
    ¢Äi|  ã                   @   sœ  d dl Z d dlmZ d dlmZ d dlmZ g d¢Ze j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d	¡d
dœdœZde j 	e j 
e j e¡¡d¡ddœiZe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœdœZddddd d!d"d#œiZd'd%d&„ZdS )(é    N)ÚModel)ÚVAD)Útrain_custom_verifier)r   r   r   z'resources/models/embedding_model.tflitezXhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/embedding_model.tflite)Ú
model_pathZdownload_urlz&resources/models/melspectrogram.tflitezWhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/melspectrogram.tflite)Z	embeddingZmelspectrogramZ
silero_vadz resources/models/silero_vad.onnxzQhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/silero_vad.onnxz"resources/models/alexa_v0.1.tflitezShttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/alexa_v0.1.tflitez(resources/models/hey_mycroft_v0.1.tflitezYhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/hey_mycroft_v0.1.tflitez'resources/models/hey_jarvis_v0.1.tflitezXhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/hey_jarvis_v0.1.tflitez(resources/models/hey_rhasspy_v0.1.tflitezYhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/hey_rhasspy_v0.1.tflitez"resources/models/timer_v0.1.tflitezShttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/timer_v0.1.tflitez$resources/models/weather_v0.1.tflitezUhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/weather_v0.1.tflite)ZalexaZhey_mycroftZ
hey_jarvisZhey_rhasspyÚtimerÚweatherr   Z1_minute_timerZ5_minute_timerZ10_minute_timerZ20_minute_timerZ30_minute_timerZ1_hour_timer)Ú1Ú2Ú3Ú4Ú5Ú6Útflitec                 C   s8   | dkrdd„ t  ¡ D ƒS | dkrdd„ t  ¡ D ƒS d S )Nr   c                 S   s   g | ]}t | d  ‘qS )r   )ÚMODELS©Ú.0Úi© r   úO/home/kim/smarthome/.venv/lib/python3.10/site-packages/openwakeword/__init__.pyÚ
<listcomp>C   s    z.get_pretrained_model_paths.<locals>.<listcomp>Zonnxc                 S   s    g | ]}t | d   dd¡‘qS )r   z.tflitez.onnx)r   Úreplacer   r   r   r   r   E   s     )r   Úkeys)Zinference_frameworkr   r   r   Úget_pretrained_model_pathsA   s
   ÿr   )r   )ÚosZopenwakeword.modelr   Zopenwakeword.vadr   Z"openwakeword.custom_verifier_modelr   Ú__all__ÚpathÚjoinÚdirnameÚabspathÚ__file__ZFEATURE_MODELSZ
VAD_MODELSr   Zmodel_class_mappingsr   r   r   r   r   Ú<module>   s\    þþûþÿ	þþþþþþëúÿ