Add TTS to demo

apps/mic.py

@@ -4,7 +4,9 @@ import streamlit as st
 import numpy as np
 import pandas as pd
 import os
+import pyttsx3
 import matplotlib.pyplot as plt
+import re
 from mtranslate import translate
 from .utils import (
     read_markdown,
@@ -22,6 +24,10 @@ from streamlit import caching
 
 def app(state):
     mic_state = state
+    engine = pyttsx3.init()
+    engine.setProperty('rate', 100)
+    voices = engine.getProperty('voices')
+    engine.setProperty('voice', voices[1].id) 
     with st.beta_expander("Usage"):
         st.write(read_markdown("usage.md"))
     st.write("\n")
@@ -135,6 +141,21 @@ def app(state):
             "**English Translation**: "+ sequence[0] if lang_id=="en" else translate(sequence[0])
         )
 
+        try:
+            clean_text = re.sub(r'[^A-Za-z0-9 ]+', '', sequence[0])
+            engine.say(clean_text)
+            engine.runAndWait()
+            engine.stop()
+
+            engine.save_to_file(clean_text, 'temp.mp3')
+            engine.runAndWait()
+
+
+            audio_file = open('temp.mp3', 'rb')
+            audio_bytes = audio_file.read()
+            st.audio(audio_bytes, format='audio/mp3')
+        except:
+            pass
 
 
 

requirements.txt

@@ -5,4 +5,5 @@ torchvision==0.10.0
 mtranslate==1.8
 black==21.7b0
 flax==0.3.4
-sentencepiece==0.1.96
+sentencepiece==0.1.96
+pyttsx3==2.90