Update app.py

app.py

@@ -1,27 +1,21 @@
 # version - ArticMonkey:19.03.24:1743
 
 import psutil
-import base64
-import time
 import sys
 
 import streamlit as st 
+
+# my modules 
 from audio_processing.A2T import A2T
 from audio_processing.T2A import T2A
 from llm.llm import LLM_chain
+
+# libraries from other authors 
 from streamlit_mic_recorder import mic_recorder
+from streamlit_TTS import auto_play
 
 llmchain = LLM_chain()
 
-# def autoplay(audio: bytes, duration: float, wait: bool=True, sr: int = 16_000):
-#     if audio:
-#         audio_bytes = audio
-#         duration = duration
-#         audio_base64 = base64.b64encode(audio_bytes).decode('utf-8')
-#         st.audio(audio_base64, sample_rate=sr)
-#         if wait:
-#             time.sleep(2)
-
 def main():
     mic = mic_recorder(start_prompt="Record",stop_prompt="Stop", just_once=True)
 
@@ -29,10 +23,13 @@ def main():
         a2t = A2T(mic["bytes"])
         text = a2t.predict()
         response = llmchain(entity=text, id=0)
+    
+        audio, _, _ = T2A(response).get_audio()
+        auto_play(audio)
+
         print(sys.getsizeof(response))
-        audio, sr, duration = T2A(response).get_audio()
-        st.audio(audio, sample_rate=sr)
         print(sys.getsizeof(audio))
+        
         del response
         del audio