Update app.py

app.py

@@ -13,13 +13,10 @@ from streamlit_mic_recorder import mic_recorder
 
 llmchain = LLM_chain()
 
-def autoplay(audio, wait: bool=True):
+def autoplay(audio: bytes, duration: float, wait: bool=True, sr: int=16000):
     if audio:
         audio_bytes = audio
-        sample_rate = 16000
-        sample_width = audio["sample_width"]
-        total_samples = len(audio_bytes) / sample_width
-        duration = total_samples / sample_rate
+        duration = duration
         audio_base64 = base64.b64encode(audio_bytes).decode('utf-8')
         if wait:
             time.sleep(duration)
@@ -32,8 +29,8 @@ def main():
         text = a2t.predict()
         response = llmchain(entity=text, id=0)
         print(sys.getsizeof(a2t), " ", sys.getsizeof(text), " ", sys.getsizeof(response), " ", sys.getsizeof(llmchain))
-        t2a = T2A(response).get_audio()
-        autoplay(t2a)
+        audio, _, duration = T2A(response).get_audio()
+        autoplay(audio=audio, duration=duration)
         del response