audex 1.0.7a3__py3-none-any.whl

audex/lib/injectors/recorder.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from audex.config import Config
+    from audex.lib.recorder import AudioRecorder
+    from audex.lib.store import Store
+
+
+def make_recorder(config: Config, store: Store) -> AudioRecorder:
+    import pyaudio
+
+    from audex.lib.recorder import AudioConfig
+    from audex.lib.recorder import AudioRecorder
+
+    fmt = {
+        "float32": pyaudio.paFloat32,
+        "int32": pyaudio.paInt32,
+        "int16": pyaudio.paInt16,
+        "int8": pyaudio.paInt8,
+        "uint8": pyaudio.paUInt8,
+    }
+    return AudioRecorder(
+        store=store,
+        config=AudioConfig(
+            format=fmt[config.infrastructure.recorder.format],
+            channels=config.infrastructure.recorder.channels,
+            rate=config.core.audio.sample_rate,
+            chunk=config.infrastructure.recorder.chunk,
+            input_device_index=config.infrastructure.recorder.input_device_index,
+        ),
+    )

audex/lib/injectors/server.py

@@ -0,0 +1,17 @@
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from audex.lib.exporter import Exporter
+    from audex.lib.repos.doctor import DoctorRepository
+    from audex.lib.server import Server
+
+
+def make_server(
+    doctor_repo: DoctorRepository,
+    exporter: Exporter,
+) -> Server:
+    from audex.lib.server import Server
+
+    return Server(doctor_repo=doctor_repo, exporter=exporter)

audex/lib/injectors/session.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from audex.config import Config
+    from audex.lib.session import SessionManager
+
+
+def make_session_manager(config: Config) -> SessionManager:
+    import datetime
+
+    from audex.lib.session import SessionManager
+
+    return SessionManager(
+        app_name=config.core.app.app_name,
+        ttl=datetime.timedelta(hours=config.core.session.ttl_hours),
+    )

audex/lib/injectors/sqlite.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from audex.config import Config
+    from audex.lib.database.sqlite import SQLite
+
+
+def make_sqlite(config: Config) -> SQLite:
+    from audex.lib.database.sqlite import SQLite
+    from audex.lib.repos.tables import TABLES
+
+    return SQLite(
+        uri=config.infrastructure.sqlite.uri,
+        tables=list(TABLES),
+        echo=config.infrastructure.sqlite.echo,
+        pool_size=config.infrastructure.sqlite.pool_size,
+        max_overflow=config.infrastructure.sqlite.max_overflow,
+        pool_recycle=config.infrastructure.sqlite.pool_recycle,
+        pool_timeout=config.infrastructure.sqlite.pool_timeout,
+        pool_pre_ping=config.infrastructure.sqlite.pool_pre_ping,
+        create_all=config.infrastructure.sqlite.create_all,
+    )

audex/lib/injectors/store.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from audex.config import Config
+    from audex.lib.store import Store
+
+
+def make_store(config: Config) -> Store:
+    from audex.lib.store.localfile import LocalFileStore
+
+    return LocalFileStore(config.infrastructure.store.base_url)

audex/lib/injectors/transcription.py

@@ -0,0 +1,42 @@
+from __future__ import annotations
+
+import typing as t
+
+from audex.exceptions import ConfigurationError
+from audex.utils import Unset
+
+if t.TYPE_CHECKING:
+    from audex.config import Config
+    from audex.lib.transcription import Transcription
+
+
+def make_transcription(config: Config) -> Transcription:
+    if isinstance(config.provider.transcription.dashscope.credential.api_key, Unset):
+        raise ConfigurationError(
+            config_key="provider.transcription.dashscope.credential.api_key",
+            reason="missing",
+        )
+
+    from audex.lib.transcription.dashscope import DashscopeParaformer
+
+    return DashscopeParaformer(
+        model=config.provider.transcription.dashscope.model,
+        api_key=config.provider.transcription.dashscope.credential.api_key,
+        user_agent=config.provider.transcription.dashscope.user_agent,
+        workspace=config.provider.transcription.dashscope.workspace,
+        max_connections=config.provider.transcription.dashscope.websocket.max_connections,
+        idle_timeout=config.provider.transcription.dashscope.websocket.idle_timeout,
+        drain_timeout=config.provider.transcription.dashscope.websocket.drain_timeout,
+        fmt=config.provider.transcription.dashscope.session.fmt,
+        sample_rate=config.core.audio.sample_rate,
+        silence_duration_ms=config.provider.transcription.dashscope.session.silence_duration_ms,
+        vocabulary_id=config.provider.transcription.dashscope.session.vocabulary_id,
+        disfluency_removal_enabled=config.provider.transcription.dashscope.session.disfluency_removal_enabled,
+        lang_hints=config.provider.transcription.dashscope.session.lang_hints,
+        semantic_punctuation=config.provider.transcription.dashscope.session.semantic_punctuation,
+        multi_thres_mode=config.provider.transcription.dashscope.session.multi_thres_mode,
+        punctuation_pred=config.provider.transcription.dashscope.session.punctuation_pred,
+        heartbeat=config.provider.transcription.dashscope.session.heartbeat,
+        itn=config.provider.transcription.dashscope.session.itn,
+        resources=config.provider.transcription.dashscope.session.resources,
+    )

audex/lib/injectors/usb.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from audex.lib.usb import USBManager
+
+
+def make_usb_manager() -> USBManager:
+    from audex.lib.usb import USBManager
+
+    return USBManager()

audex/lib/injectors/vpr.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+import typing as t
+
+from audex.exceptions import ConfigurationError
+from audex.utils import Unset
+
+if t.TYPE_CHECKING:
+    from audex.config import Config
+    from audex.lib.vpr import VPR
+
+
+def make_vpr(config: Config) -> VPR:
+    if config.provider.vpr.provider == "xfyun":
+        if isinstance(config.provider.vpr.xfyun.credential.app_id, Unset):
+            raise ConfigurationError(
+                config_key="provider.vpr.xfyun.credential.app_id",
+                reason="missing",
+            )
+        if isinstance(config.provider.vpr.xfyun.credential.api_key, Unset):
+            raise ConfigurationError(
+                config_key="provider.vpr.xfyun.credential.api_key",
+                reason="missing",
+            )
+        if isinstance(config.provider.vpr.xfyun.credential.api_secret, Unset):
+            raise ConfigurationError(
+                config_key="provider.vpr.xfyun.credential.api_secret",
+                reason="missing",
+            )
+
+        from audex.lib.vpr.xfyun import XFYunVPR
+
+        return XFYunVPR(
+            app_id=config.provider.vpr.xfyun.credential.app_id,
+            api_key=config.provider.vpr.xfyun.credential.api_key,
+            api_secret=config.provider.vpr.xfyun.credential.api_secret,
+            group_id=config.provider.vpr.xfyun.group_id,
+            proxy=config.provider.vpr.xfyun.http.proxy,
+            timeout=config.provider.vpr.xfyun.http.timeout,
+            default_headers=config.provider.vpr.xfyun.http.default_headers,
+            default_params=config.provider.vpr.xfyun.http.default_params,
+        )
+
+    if isinstance(config.provider.vpr.unisound.credential.appkey, Unset):
+        raise ConfigurationError(
+            config_key="provider.vpr.unisound.credential.appkey",
+            reason="missing",
+        )
+    if isinstance(config.provider.vpr.unisound.credential.secret, Unset):
+        raise ConfigurationError(
+            config_key="provider.vpr.unisound.credential.secret",
+            reason="missing",
+        )
+
+    from audex.lib.vpr.unisound import UnisoundVPR
+
+    return UnisoundVPR(
+        appkey=config.provider.vpr.unisound.credential.appkey,
+        secret=config.provider.vpr.unisound.credential.secret,
+        group_id=config.provider.vpr.unisound.group_id,
+        proxy=config.provider.vpr.unisound.http.proxy,
+        timeout=config.provider.vpr.unisound.http.timeout,
+        default_headers=config.provider.vpr.unisound.http.default_headers,
+        default_params=config.provider.vpr.unisound.http.default_params,
+    )

audex/lib/injectors/wifi.py

@@ -0,0 +1,7 @@
+from __future__ import annotations
+
+from audex.lib.wifi import WiFiManager
+
+
+def make_wifi_manager() -> WiFiManager:
+    return WiFiManager()