examples : remove explicit parakeet_state from example

Author: danbev

examples/parakeet-cli/parakeet-cli.cpp

@@ -145,17 +145,11 @@ int main(int argc, char ** argv) {
         ctx_params.flash_attn  = params.flash_attn;
         ctx_params.gpu_device  = params.gpu_device;
 
-        struct parakeet_context * pctx = parakeet_init_from_file_with_params_no_state(params.model.c_str(), ctx_params);
+        struct parakeet_context * pctx = parakeet_init_from_file_with_params(params.model.c_str(), ctx_params);
         if (pctx == nullptr) {
             fprintf(stderr, "error: failed to load Parakeet model from '%s'\n", params.model.c_str());
             return 1;
         }
-        struct parakeet_state * state = parakeet_init_state(pctx);
-        if (state == nullptr) {
-            fprintf(stderr, "error: failed to initialize parakeet state\n");
-            parakeet_free(pctx);
-            return 2;
-        }
 
         fprintf(stderr, "Successfully loaded Parakeet model\n");
         fprintf(stderr, "Processing audio (%zu samples, %.2f seconds)\n",
@@ -169,41 +163,36 @@ int main(int argc, char ** argv) {
         full_params.new_token_callback  = token_callback;
         full_params.new_token_callback_user_data = nullptr;
 
-        const int mel_frames = pcmf32.size() / PARAKEET_HOP_LENGTH;
-        const int model_max_ctx = parakeet_n_audio_ctx(pctx);
-        const bool fits_single_chunk = mel_frames <= model_max_ctx;
-
-        int ret;
-        if (fits_single_chunk) {
-            ret = parakeet_chunk(pctx, state, full_params, pcmf32.data(), pcmf32.size());
-        } else {
-            ret = parakeet_full_with_state(pctx, state, full_params, pcmf32.data(), pcmf32.size());
+        const int mel_frames = (int)(pcmf32.size() / PARAKEET_HOP_LENGTH);
+        if (mel_frames <= parakeet_n_audio_ctx(pctx)) {
+            full_params.chunk_length_ms = 0;
         }
 
+        int ret = parakeet_full(pctx, full_params, pcmf32.data(), pcmf32.size());
+
         if (ret != 0) {
             fprintf(stderr, "error: failed to process audio file '%s'\n", fname.c_str());
-            parakeet_free_state(state);
             parakeet_free(pctx);
             continue;
         }
 
         printf("\n");
 
         if (params.print_segments) {
-            const int n_segments = parakeet_full_n_segments_from_state(state);
+            const int n_segments = parakeet_full_n_segments(pctx);
             fprintf(stderr, "\nSegments (%d):\n", n_segments);
 
             for (int i = 0; i < n_segments; i++) {
-                const char * text = parakeet_full_get_segment_text_from_state(state, i);
-                const int64_t t0 = parakeet_full_get_segment_t0_from_state(state, i);
-                const int64_t t1 = parakeet_full_get_segment_t1_from_state(state, i);
-                const int n_tokens = parakeet_full_n_tokens_from_state(state, i);
+                const char * text = parakeet_full_get_segment_text(pctx, i);
+                const int64_t t0 = parakeet_full_get_segment_t0(pctx, i);
+                const int64_t t1 = parakeet_full_get_segment_t1(pctx, i);
+                const int n_tokens = parakeet_full_n_tokens(pctx, i);
 
                 fprintf(stderr, "Segment %d: [%lld -> %lld] \"%s\"\n", i, (long long)t0, (long long)t1, text);
                 fprintf(stderr, "Tokens [%d]:\n", n_tokens);
 
                 for (int j = 0; j < n_tokens; j++) {
-                    parakeet_token_data token_data = parakeet_full_get_token_data_from_state(state, i, j);
+                    parakeet_token_data token_data = parakeet_full_get_token_data(pctx, i, j);
                     const char * token_str = parakeet_token_to_str(pctx, token_data.id);
 
                     fprintf(stderr, "  [%2d] id=%5d frame=%3d dur_idx=%2d dur_val=%2d p=%.4f plog=%.4f t0=%4lld t1=%4lld word_start=%s \"%s\"\n",
@@ -222,7 +211,6 @@ int main(int argc, char ** argv) {
             }
         }
 
-        parakeet_free_state(state);
         parakeet_free(pctx);
     }