Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,15 +6,18 @@ from io import BytesIO
|
|
6 |
from gtts import gTTS
|
7 |
from streamlit_webrtc import webrtc_streamer, WebRtcMode
|
8 |
import speech_recognition as sr
|
|
|
9 |
|
10 |
client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
|
11 |
pre_prompt = ""
|
12 |
pre_prompt_sent = False
|
|
|
13 |
|
14 |
def take_user_input():
|
15 |
r = sr.Recognizer()
|
16 |
|
17 |
def audio_callback(in_data, frame_count, time_info, status):
|
|
|
18 |
audio = sr.AudioData(
|
19 |
in_data.tobytes(),
|
20 |
sample_rate=webrtc_ctx.audio_sample_rate,
|
@@ -29,6 +32,7 @@ def take_user_input():
|
|
29 |
exit()
|
30 |
return query
|
31 |
|
|
|
32 |
webrtc_ctx = webrtc_streamer(
|
33 |
key="microphone",
|
34 |
mode=WebRtcMode.SENDRECV,
|
@@ -56,12 +60,12 @@ def take_user_input():
|
|
56 |
st.info('Reconociendo...')
|
57 |
|
58 |
query = transcribe_speech(audio)
|
59 |
-
|
60 |
if 'salir' in query or 'detener' in query:
|
61 |
speak("Hasta luego.")
|
62 |
exit()
|
63 |
return query
|
64 |
-
|
65 |
except sr.UnknownValueError:
|
66 |
speak('No se ha reconocido nada. Intenta de nuevo...')
|
67 |
except sr.RequestError as e:
|
|
|
6 |
from gtts import gTTS
|
7 |
from streamlit_webrtc import webrtc_streamer, WebRtcMode
|
8 |
import speech_recognition as sr
|
9 |
+
import sounddevice as sd
|
10 |
|
11 |
client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
|
12 |
pre_prompt = ""
|
13 |
pre_prompt_sent = False
|
14 |
+
webrtc_ctx = None
|
15 |
|
16 |
def take_user_input():
|
17 |
r = sr.Recognizer()
|
18 |
|
19 |
def audio_callback(in_data, frame_count, time_info, status):
|
20 |
+
global webrtc_ctx
|
21 |
audio = sr.AudioData(
|
22 |
in_data.tobytes(),
|
23 |
sample_rate=webrtc_ctx.audio_sample_rate,
|
|
|
32 |
exit()
|
33 |
return query
|
34 |
|
35 |
+
global webrtc_ctx
|
36 |
webrtc_ctx = webrtc_streamer(
|
37 |
key="microphone",
|
38 |
mode=WebRtcMode.SENDRECV,
|
|
|
60 |
st.info('Reconociendo...')
|
61 |
|
62 |
query = transcribe_speech(audio)
|
63 |
+
|
64 |
if 'salir' in query or 'detener' in query:
|
65 |
speak("Hasta luego.")
|
66 |
exit()
|
67 |
return query
|
68 |
+
|
69 |
except sr.UnknownValueError:
|
70 |
speak('No se ha reconocido nada. Intenta de nuevo...')
|
71 |
except sr.RequestError as e:
|