fix: prefer setting the timestamps from jigasi

jitsi · Aug 20, 2024 · f51d347 · f51d347
1 parent 9953c9a
commit f51d347
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/src/main/java/org/jitsi/jigasi/transcription/WhisperWebsocket.java b/src/main/java/org/jitsi/jigasi/transcription/WhisperWebsocket.java
@@ -45,6 +45,8 @@ public class WhisperWebsocket
 
     private Map<String, Set<TranscriptionListener>> participantListeners = new ConcurrentHashMap<>();
 
+    private final Map<String, Instant> participantTranscriptionStarts = new ConcurrentHashMap<>();
+
     private static final int maxRetryAttempts = 10;
 
 
@@ -271,6 +273,15 @@ public void onMessage(String msg)
         {
             logger.debug("Received final: " + result);
         }
+
+        Instant startTranscription = participantTranscriptionStarts.getOrDefault(participantId, null);
+        if (startTranscription == null)
+        {
+            Date now = new Date();
+            startTranscription = now.toInstant();
+            participantTranscriptionStarts.put(participantId, startTranscription);
+        }
+
         Set<TranscriptionListener> partListeners = participantListeners.getOrDefault(participantId, null);
         if (!result.isEmpty() && partListeners != null)
         {
@@ -287,13 +298,17 @@ public void onMessage(String msg)
                 TranscriptionResult tsResult = new TranscriptionResult(
                         participant,
                         id,
-                        transcriptionStart,
+                        startTranscription,
                         partial,
                         getLanguage(participant),
                         stability,
                         new TranscriptionAlternative(result));
                 l.notify(tsResult);
             }
+            if (!partial)
+            {
+                participantTranscriptionStarts.remove(participantId);
+            }
         }
     }