From 2b98af7b19af3e2c3d6303d6a6ccb4f47a908826 Mon Sep 17 00:00:00 2001
From: "J.P Lee" <koreanfeel@gmail.com>
Date: Thu, 7 Dec 2023 17:00:38 +0900
Subject: [PATCH] Fix: Omitting the last chunk problem in comp_unaware mode

---
 whisper_online.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/whisper_online.py b/whisper_online.py
index 8efbbab..9eb17ea 100644
--- a/whisper_online.py
+++ b/whisper_online.py
@@ -573,10 +573,15 @@ if __name__ == "__main__":
 
             print(f"## last processed {end:.2f}s",file=logfile,flush=True)
 
-            beg = end
-            end += min_chunk
             if end >= duration:
                 break
+            
+            beg = end
+            
+            if end + min_chunk > duration:
+                end = duration
+            else:
+                end += min_chunk
         now = duration
 
     else: # online = simultaneous mode