From 4ac0acee80c932b1de0b80c65c7e7e8062f98b4d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bjarke=20Viks=C3=B8e?= <bviksoe@hotmail.com>
Date: Wed, 3 Jul 2024 12:43:53 +0200
Subject: [PATCH] Incorrect timetstamps

Fixes #2271

- Adds consecutive timestamps after end of last segment as the new starting ts
- Add these timestamp to output when "print-special" enabled
- Fixes fflush usage in live reporting

I was not able to test this with the special "token_timestamps" option.
---
 src/whisper.cpp | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/whisper.cpp b/src/whisper.cpp
index 752d52901b1..a0ef4255c6b 100644
--- a/src/whisper.cpp
+++ b/src/whisper.cpp
@@ -6261,11 +6261,15 @@ int whisper_full_with_state(
                             }
                         }
                         text = "";
+                        t0 = t1;
                         while (i < (int) tokens_cur.size() && tokens_cur[i].id > whisper_token_beg(ctx)) {
+                            if (params.print_special) {
+                                text += whisper_token_to_str(ctx, tokens_cur[i].id);
+                            }
+                            t0 = seek + 2 * (tokens_cur[i].tid - whisper_token_beg(ctx));
                             i++;
                         }
                         i--;
-                        t0 = t1;
                         i0 = i + 1;
                         speaker_turn_next = false;
                     }
@@ -6282,8 +6286,8 @@ int whisper_full_with_state(
                             printf("[%s --> %s]  %s\n", to_timestamp(tt0).c_str(), to_timestamp(tt1).c_str(), text.c_str());
                         } else {
                             printf("%s", text.c_str());
-                            fflush(stdout);
                         }
+                        fflush(stdout);
                     }
 
                     result_all.push_back({ tt0, tt1, text, {} , speaker_turn_next });