ENCOA-295

ielts_be/services/impl/third_parties/aws_polly.py

@@ -1,4 +1,5 @@
 import random
+from typing import Optional
 
 from aiobotocore.client import BaseClient
 
@@ -21,7 +22,7 @@ class AWSPolly(ITextToSpeechService):
         )
         return await tts_response['AudioStream'].read()
 
-    async def text_to_speech(self, dialog: Dialog) -> bytes:
+    async def text_to_speech(self, dialog: Dialog, include_final_clue = True) -> bytes:
         if not dialog.conversation and not dialog.monologue:
             raise ValueError("Unsupported argument for text_to_speech")
 
@@ -30,13 +31,14 @@ class AWSPolly(ITextToSpeechService):
         else:
             audio_segments = await self._conversation_to_speech(dialog)
 
-        final_message = await self.synthesize_speech(
-            "This audio recording, for the listening exercise, has finished.",
-            "Stephen"
-        )
+        if include_final_clue:
+            final_message = await self.synthesize_speech(
+                "This audio recording, for the listening exercise, has finished.",
+                "Stephen"
+            )
 
-        # Add finish message
-        audio_segments.append(final_message)
+            # Add finish message
+            audio_segments.append(final_message)
 
         # Combine the audio segments into a single audio file
         combined_audio = b"".join(audio_segments)