o
    dZŽh%  ã                   @   s<   d dl Z d dlmZ d dlmZ e  e¡ZG dd„ dƒZdS )é    N)ÚBytesIO)Ú	AudioDatac                   @   s,   e Zd Zd
dd„Zdddedefdd	„ZdS )ÚOpenAICompatibleRecognizerÚreturnNc                 C   s
   || _ d S )N)Úclient)Úselfr   © r   ú^/var/www/auris/lib/python3.10/site-packages/speech_recognition/recognizers/whisper_api/base.pyÚ__init__
   s   
z#OpenAICompatibleRecognizer.__init__Ú
audio_datar   Úmodelc                 K   sZ   t |tƒs	tdƒ‚t| ¡ ƒ}d|_d|i|¥}t |¡ | jj	j
jdd|i|¤Ž}|jS )Nz0``audio_data`` must be an ``AudioData`` instancezSpeechRecognition_audio.wavr   Úfiler   )Ú
isinstancer   Ú
ValueErrorr   Zget_wav_dataÚnameÚloggerÚdebugr   ZaudioZtranscriptionsÚcreateÚtext)r   r   r   ÚkwargsZwav_dataÚ
parametersZ
transcriptr   r   r	   Ú	recognize   s   
ÿ
ÿÿz$OpenAICompatibleRecognizer.recognize)r   N)Ú__name__Ú
__module__Ú__qualname__r
   Ústrr   r   r   r   r	   r   	   s    
r   )	ÚloggingÚior   Zspeech_recognition.audior   Ú	getLoggerr   r   r   r   r   r   r	   Ú<module>   s
    
