Gabriel Lafond-Thenaille · 9512e4a8 · 9512e4a8 · 8866723b · 8866723b · 5bfc374d
--- a/src/input/decoder.c
+++ b/src/input/decoder.c
    }
 }

+static bool SubDecoderIsStt(vlc_input_decoder_t *subdec)
+{
+    return subdec->dec.fmt_in->i_cat == SPU_ES &&
+           subdec->dec.fmt_in->i_extra == sizeof(vlc_stt_extra_t) &&
+           subdec->dec.fmt_in->i_codec == VLC_CODEC_STT;
+}
--- a/modules/codec/stt_whisper.c 0 → 100644
+++ b/modules/codec/stt_whisper.c 0 → 100644
+    int model_index;
+    atomic_bool stop;
+    vlc_thread_t thread;
+} vlc_stt_loader_sys_t;
+
+/* Describe the models */
+typedef struct {
+    const char *nickname;   /* used to select the model */
+    const char *name;       /* real name of the model */
+    const char *checksum;   /* checksum of the model */
+    const char *url;        /* url to download the model */
+} vlc_stt_whisper_model_t;
+
+#define STT_WHISPER_NMODELS 9
+/* List of all available models */
+static const vlc_stt_whisper_model_t whisper_models[STT_WHISPER_NMODELS] = {
--- a/include/vlc_stt.h 0 → 100644
+++ b/include/vlc_stt.h 0 → 100644
+
+struct vlc_stt_ctx {
+    /**
+     * The context loaded and used by the module.
+     */
+    void *data;
+
+    /**
+     * The module name
+     */
+    const char *name;
+};
+
+struct vlc_stt_ctx_callbacks {
+    /**
+     * Called when the context is loaded or when an error occur in the loader
--- a/modules/codec/stt_whisper.c 0 → 100644
+++ b/modules/codec/stt_whisper.c 0 → 100644
+    if (cache_dir == NULL) {
+        ret = -ENOMEM;
+        goto ret_ctxgetmodelpath;
+    }
+    if(asprintf(&model_dir, "%s" DIR_SEP "models", cache_dir) < 0) {
+        ret = -ENOMEM;
+        goto ret_ctxgetmodelpath;
+    }
+
+    /* create the models cache dir if missing */
+    struct stat st;
+    if (vlc_stat(model_dir, &st) == -1) {
+        msg_Info(loader, "creating models folder");
+        if (vlc_mkdir_parent(model_dir, 0755) != 0) {
+            ret = VLC_EGENERIC;
+            goto ret_ctxgetmodelpath;
--- a/modules/codec/stt_whisper.c 0 → 100644
+++ b/modules/codec/stt_whisper.c 0 → 100644
+            goto ret_ctxgetmodelpath;
+        }
+    }
+
+    /* get the path of the model */
+    char *model_path = NULL;
+    ret = asprintf(&model_path, "%s" DIR_SEP "%s", model_dir,
+                   whisper_models[i].name);
+    if (ret > 0) {
+        sys->model_index = i;
+        sys->model_path = model_path;
+        sys->model_dir = model_dir;
+        ret = VLC_SUCCESS;
+    } else {
+        free(model_dir);
+        ret = VLC_EGENERIC;
--- a/modules/codec/stt_whisper.c 0 → 100644
+++ b/modules/codec/stt_whisper.c 0 → 100644
+    vlc_hex_encode_binary(hash, 32, hash_hex);
+    msg_Info(loader, "hash:\n1: %s\n2: %s", hash_hex,
+            whisper_models[index].checksum);
+
+    vlc_dialog_release(loader, dialog);
+    gcry_md_close(gctx);
+    if (strcmp(hash_hex, whisper_models[index].checksum)) {
+        msg_Info(loader, "Error wrong hash!");
+        return VLC_EGENERIC;
+    }
+    return VLC_SUCCESS;
+#else
+    VLC_UNUSED(stream);
+    VLC_UNUSED(fd);
+    VLC_UNUSED(index);
+    return VLC_EGENERIC;
-    VLC_UNUSED(index);
-    return VLC_EGENERIC;
+    VLC_UNUSED(index);
+    VLC_UNUSED(sys);
+    return VLC_EGENERIC;
--- a/modules/codec/stt_whisper.c 0 → 100644
+++ b/modules/codec/stt_whisper.c 0 → 100644
+        .use_gpu = true,
+    };
+    ctx.data = whisper_init_from_file_with_params(sys->model_path, ctxparams);
+    if (ctx.data == NULL) {
+        vlc_dialog_release(loader, dialog);
+        goto vlc_stt_whisper_CtxThread_error;
+    }
+
+    vlc_dialog_release(loader, dialog);
+
+    /* Print whisper context information. */
+    msg_Info(loader, "%s", info);
+
+vlc_stt_whisper_CtxThread_error:
+    if (ctx.data == NULL) {
+        msg_Err(loader, "Whisper context isn't load properly speech-to-text desactivated.");
-        msg_Err(loader, "Whisper context isn't load properly speech-to-text desactivated.");
+        msg_Err(loader, "Whisper context isn't load properly speech-to-text deactivated.");