Update app.py

app.py

@@ -1,30 +1,14 @@
 import io
 import base64
-import pyaudio
-import webrtcvad
 import threading
 import numpy as np
 from gtts import gTTS
 import streamlit as st
-import sounddevice as sd
 import speech_recognition as sr
 from huggingface_hub import InferenceClient
-
-def list_audio_devices():
-    p = pyaudio.PyAudio()
-    info = p.get_host_api_info_by_index(0)
-    num_devices = info.get('deviceCount')
-    devices = []
-
-    for i in range(num_devices):
-        device_info = p.get_device_info_by_host_api_device_index(0, i)
-        devices.append(device_info['name'])
-
-    p.terminate()
-    return devices
-
-devices = sd.query_devices()
-print(devices)
+from streamlit_mic_recorder import mic_recorder
+import wave
+import webrtcvad
 
 if "history" not in st.session_state:
     st.session_state.history = []
@@ -141,16 +125,24 @@ class Threader(threading.Thread):
             print("Started mythread")
             start_stream()
 
-available_audio_devices = list_audio_devices()
-print("Dispositivos de audio disponibles:")
-print(available_audio_devices)
-selected_device = st.selectbox("Selecciona el micrófono:", available_audio_devices, index=0)
-channels = [1]
-mapping = [c - 1 for c in channels]
-device_info = sd.query_devices(selected_device, 'input')
-sample_rate = int(device_info['default_samplerate'])
-
 if __name__ == "__main__":
+    # Cambios para usar streamlit_mic_recorder
+    st.write("Record your voice, and play the recorded audio:")
+    audio = mic_recorder(start_prompt="▶️", stop_prompt="🛑", key='recorder')
+
+    if audio:       
+        st.audio(audio['bytes'])
+        
+        audio_bytes = audio["bytes"]
+        sample_width = audio["sample_width"]
+        sample_rate = audio["sample_rate"]
+        num_channels = 1
+
+        with wave.open(temp_audio_file_path, 'w') as wave_file:
+            wave_file.setnchannels(num_channels)
+            wave_file.setsampwidth(sample_width)
+            wave_file.setframerate(sample_rate)
+            wave_file.writeframes(audio_bytes)
 
     vad = webrtcvad.Vad(1)
 
@@ -163,4 +155,4 @@ if __name__ == "__main__":
     block_size = int(sample_rate * interval_size / 1000)
 
     Threader(name='mythread')
-    st.button("Detener Stream")
+    st.button("Detener Stream")