Spaces:
Sleeping
Sleeping
| from gtts import gTTS | |
| from deep_translator import GoogleTranslator | |
| import soundfile as sf | |
| import tempfile | |
| import numpy as np | |
| import gtts | |
| output_path = 'Audio/output.wav' | |
| translate_path = 'Audio/translate.wav' | |
| def get_lang(lang): | |
| translate = GoogleTranslator().get_supported_languages() | |
| speak = list(gtts.lang.tts_langs().values()) | |
| speak = list(map(lambda x: x.lower(), speak)) | |
| speak.sort() | |
| set_lang = list(set(speak) & set(translate)) | |
| set_lang.sort() | |
| if lang in set_lang: | |
| for key, value in gtts.lang.tts_langs().items(): | |
| if value.lower() == lang: | |
| return key | |
| else: | |
| return 'en' | |
| else: | |
| return 'en' | |
| def audio_streaming(txt=None, lang='en', to=None): | |
| # If an audio file is provided as input, use it; otherwise, use the direct file path | |
| speak = gTTS(text=txt, lang=lang, slow=False) | |
| if to == 1: | |
| audio = output_path | |
| else: | |
| audio = translate_path | |
| speak.save(audio) | |
| # Load the audio file | |
| data, samplerate = sf.read(audio) | |
| # Ensure data is in float32 format | |
| data = np.array(data, dtype=np.float32) | |
| # Save to a temporary file that Gradio can use for audio playback | |
| with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file: | |
| sf.write(tmp_file.name, data, samplerate) | |
| temp_audio_path = tmp_file.name | |
| # Return the file path to Gradio | |
| return temp_audio_path | |
| def translate_txt(lang, text): | |
| translator = GoogleTranslator(source="en", target=lang) | |
| translated_text = translator.translate(text) | |
| audio_path = audio_streaming(translated_text, lang=get_lang(lang), to=2) | |
| return translated_text, audio_path | |
| if __name__ == "__main__": | |
| # print(audio_streaming("hello world")) | |
| # os.system(f"start {audio_streaming('hello world!')}") | |
| translate = GoogleTranslator().get_supported_languages() | |
| # print(f"Can translate: {translate}", len(translate)) | |
| # print() | |
| speak = list(gtts.lang.tts_langs().values()) | |
| speak = list(map(lambda x: x.lower(), speak)) | |
| speak.sort() | |
| # print(f"Can speak: {speak}", len(speak)) | |
| # print() | |
| set_lang = list(set(speak)&set(translate)) | |
| set_lang.sort() | |
| # print(f"Set of lang: {set_lang}", len(set_lang)) | |
| not_in_speak = list(set(translate) - set(speak)) | |
| print(not_in_speak) |