Adding fix for validating audit length

brunodmartins · brunodmartins · commit 187477a6e227 · 2025-03-28T17:06:45.000-03:00
diff --git a/localstack-core/localstack/services/transcribe/provider.py b/localstack-core/localstack/services/transcribe/provider.py
@@ -44,6 +44,8 @@
 from localstack.utils.run import run
 from localstack.utils.threads import start_thread
 
+MAX_AUDIO_DURATION_SECONDS = 60 * 60 * 4
+
 LOG = logging.getLogger(__name__)
 
 VOSK_MODELS_URL = f"{HUGGING_FACE_ENDPOINT}/vosk-models/resolve/main/"
@@ -305,6 +307,16 @@ def _run_transcription_job(self, args: Tuple[TranscribeStore, str]):
             LOG.debug("Media format detected as: %s", format)
             job["MediaFormat"] = SUPPORTED_FORMAT_NAMES[format]
 
+            duration = ffprobe_output["format"]["duration"]
+            LOG.debug("Media duration: %s (seconds)", duration)
+
+            if duration >= MAX_AUDIO_DURATION_SECONDS:
+                failure_reason = (
+                    "Invalid file size: file size too large. Maximum audio duration is 4.000000 hours. Check the length of the file and try your request again"
+                )
+                raise RuntimeError()
+
+
             # Determine the sample rate of input audio if possible
             for stream in ffprobe_output["streams"]:
                 if stream["codec_type"] == "audio":
