qi,dZddlZddlZddlmZddlmZ ddlZejjejjejjejj ejj"ejj$dZej(j*ej(j,ej(j.ej(j0ej(j2d Zd efd Zd efd ZGddZy#e$r7Z ejde ejde de dZ [ wwxYw)z)Krisp Instance manager for pipecat audio.N)Lock)loggerz Exception: zDIn order to use the Krisp instance, you need to install krisp_audio.zMissing module: )i@i>i]i}iDi)  sample_ratec|tvrEdjdttjD}t d|d|dt|S)a Convert integer sample rate to Krisp SDK enum value. Args: sample_rate: Sample rate in Hz (e.g., 16000, 24000, 48000). Returns: Corresponding Krisp SDK SampleRate enum value. Raises: ValueError: If the sample rate is not supported by Krisp SDK. , c32K|]}t|ywNstr).0rates N/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/audio/krisp_instance.py z+int_to_krisp_sample_rate..7s#\$CI#\zUnsupported sample rate: z Hz. Supported rates: z Hz)KRISP_SAMPLE_RATESjoinsortedkeys ValueError)r supported_ratess rint_to_krisp_sample_rater*s_,,))#\&ASAXAXAZ:[#\\' }4J?J[[^ _   k **frame_duration_msc|tvrEdjdttjD}t d|d|dt|S)a3Convert integer frame duration to Krisp SDK enum value. Args: frame_duration_ms: Frame duration in milliseconds (e.g., 10, 20, 30). Returns: Corresponding Krisp SDK FrameDuration enum value. Raises: ValueError: If the frame duration is not supported by Krisp SDK. r c32K|]}t|ywrr)rdurations rrz.int_to_krisp_frame_duration..Ks( &CM( rzUnsupported frame duration: z ms. Supported durations: z ms)KRISP_FRAME_DURATIONSrrrr)rsupported_durationss rint_to_krisp_frame_durationr$>sp 55"ii( *01F1K1K1M*N(  *+<*=>$$7#8 =   !!2 33rceZdZdZdZeZdZedZ edZ e dde fdZ e dZe d efd Ze d efd Ze d Zy )KrispVivaSDKManagerz=Singleton manager for Krisp VIVA SDK with reference counting.Frc:tjd|d|y)z(Callback for Krisp SDK licensing errors.zKrisp licensing error: z - N)rerror)r( error_messages r_license_callbackz%KrispVivaSDKManager._license_callback\s  .ugSHIrc:tjd|d|y)z+Thread-safe callback for Krisp SDK logging.[z] N)rinfo) log_message log_levels r _log_callbackz!KrispVivaSDKManager._log_callbackas  a {"[M23rapi_keyc |j5|jdk(r |xs tjj dd} t j d||j|jt jjd|_ t j}tjd|j d|j"d|j$t'j(|j*|xjd z c_tjd |jd d d y #t$r<t j d|jt jjYwxYw#t,$r%}d|_ tj.d|d }~wwxYw#1swYy xYw) abAcquire a reference to the SDK (initializes if needed). Call this when creating a filter instance. Args: api_key: Krisp SDK API key. If empty, falls back to the KRISP_VIVA_API_KEY environment variable. Raises: Exception: If SDK initialization fails (propagated from krisp_audio) rKRISP_VIVA_API_KEYTz.Krisp Audio Python SDK initialized - Version: .Fz!Krisp SDK initialization failed: NKrisp SDK reference count: )_lock_reference_countosenvironget krisp_audio globalInitr*r0LogLevelOff TypeError _initialized getVersionrdebugmajorminorpatchatexitregister_force_cleanup Exceptionr()clsr1key SDK_VERSIONes racquirezKrispVivaSDKManager.acquirefsYY# O##q(!MRZZ^^4H"%MC `#..11--'0044 (,C$"-"8"8":KLLH&,,-Q{/@/@.A;CTCTBUW OOC$6$67  A % LL6s7K7K6LM NG# O# O%`#..r33D3DkFZFZF^F^_`!',C$LL#DQC!HI;# O# OsUF7$FAD>A6F>7F7>AFFFF F4 F//F44F77Gc|j5|jdkDr|xjdzc_tjd|j|jdk(r=|jr1 t j d|_tjddddy#t$r)}tjd|d|_Yd}~6d}~wwxYw#1swYyxYw)z{Release a reference to the SDK (destroys if last reference). Call this when destroying a filter instance. rr6r7Fz3Krisp Audio SDK destroyed (all references released)z Error during Krisp SDK cleanup: N) r8r9rrDrBr= globalDestroyrKr(rLrOs rreleasezKrispVivaSDKManager.releases YY 1##a'$$)$ :3;O;O:PQR''1,1A1A1#113+0( %Z[ 1 1%1 'Gs%KL+0((1 1 1s0A"C00B)) C2CCCCC'returnc^|j5|jcdddS#1swYyxYw)zwGet the current reference count. Returns: Current number of active references to the SDK. N)r8r9rLs rget_reference_countz'KrispVivaSDKManager.get_reference_counts)YY ('' ( ( ( #,c^|j5|jcdddS#1swYyxYw)z~Check if the SDK is currently initialized. Returns: True if SDK is initialized, False otherwise. N)r8rBrWs ris_initializedz"KrispVivaSDKManager.is_initializeds)YY $## $ $ $rYc8|j5|jr? tjd|jdt j d|_dddy#t$r"}tjd|Yd}~/d}~wwxYw#1swYyxYw)z)Force cleanup on program exit (failsafe).z0Force cleaning up Krisp SDK at exit (ref count: )Fz'Error during forced Krisp SDK cleanup: N) r8rBrwarningr9r=rRrKr(rSs rrJz"KrispVivaSDKManager._force_cleanupsYY PPNNJ3K_K_J``ab --/',C$ P P!PLL#J1#!NOOP P Ps. B>A"" B +BBB  BBN)r4)__name__ __module__ __qualname____doc__rBrr8r9 staticmethodr*r0 classmethodrrPrTintrXboolr[rJrrr&r&UsGL FEJJ44/Oc/O/Ob11((C(($t$$ P Prr&)rbrHr: threadingrlogururr=ModuleNotFoundErrorrOr(rK SamplingRateSr8000Hz Sr16000Hz Sr24000Hz Sr32000Hz Sr44100Hz Sr48000Hzr FrameDurationFd10msFd15msFd20msFd30msFd32msr"rerr$r&rgrrrxsG0 ,  " " + +  # # - -  # # - -  # # - -  # # - -  # # - -  !!((!!((!!((!!((!!(( +#+(434.xPxPI,FLL;qc"#FLLWX &qc* ++,sD E%2EE