Update audio_processor.py
Browse files- audio_processor.py +5 -6
audio_processor.py
CHANGED
@@ -1,25 +1,24 @@
|
|
1 |
from gtts import gTTS
|
2 |
import io
|
3 |
|
4 |
-
#
|
5 |
def text_to_speech(text, target_lang):
|
6 |
try:
|
7 |
-
if not text or not text.strip():
|
8 |
print("Error: Empty or invalid text for speech synthesis")
|
9 |
return None
|
10 |
-
# Clean and
|
11 |
-
cleaned_text = text.strip()[:200]
|
12 |
print(f"Attempting to synthesize: '{cleaned_text}' for lang: {target_lang}")
|
13 |
lang_map = {"en": "en", "fr": "fr", "es": "es", "de": "de", "hi": "hi", "zh": "zh-cn", "ar": "ar", "ru": "ru", "ja": "ja"}
|
14 |
lang_code = lang_map.get(target_lang, "en")
|
15 |
-
if lang_code not in lang_map.values():
|
16 |
print(f"Error: Unsupported language code {lang_code} for speech synthesis")
|
17 |
return None
|
18 |
tts = gTTS(text=cleaned_text, lang=lang_code, slow=False)
|
19 |
audio_buffer = io.BytesIO()
|
20 |
tts.write_to_fp(audio_buffer)
|
21 |
audio_buffer.seek(0)
|
22 |
-
if audio_buffer.getbuffer().nbytes > 0:
|
23 |
print(f"Audio generated successfully, size: {audio_buffer.getbuffer().nbytes} bytes")
|
24 |
return audio_buffer
|
25 |
print("Error: Audio buffer is empty after synthesis")
|
|
|
1 |
from gtts import gTTS
|
2 |
import io
|
3 |
|
4 |
+
# Convert translated text to speech
|
5 |
def text_to_speech(text, target_lang):
|
6 |
try:
|
7 |
+
if not text or not text.strip(): # Check for valid text
|
8 |
print("Error: Empty or invalid text for speech synthesis")
|
9 |
return None
|
10 |
+
cleaned_text = text.strip()[:200] # Clean and limit text to 200 characters
|
|
|
11 |
print(f"Attempting to synthesize: '{cleaned_text}' for lang: {target_lang}")
|
12 |
lang_map = {"en": "en", "fr": "fr", "es": "es", "de": "de", "hi": "hi", "zh": "zh-cn", "ar": "ar", "ru": "ru", "ja": "ja"}
|
13 |
lang_code = lang_map.get(target_lang, "en")
|
14 |
+
if lang_code not in lang_map.values(): # Validate language code
|
15 |
print(f"Error: Unsupported language code {lang_code} for speech synthesis")
|
16 |
return None
|
17 |
tts = gTTS(text=cleaned_text, lang=lang_code, slow=False)
|
18 |
audio_buffer = io.BytesIO()
|
19 |
tts.write_to_fp(audio_buffer)
|
20 |
audio_buffer.seek(0)
|
21 |
+
if audio_buffer.getbuffer().nbytes > 0: # Verify audio data
|
22 |
print(f"Audio generated successfully, size: {audio_buffer.getbuffer().nbytes} bytes")
|
23 |
return audio_buffer
|
24 |
print("Error: Audio buffer is empty after synthesis")
|