salomonsky commited on
Commit
f03e9a3
·
verified ·
1 Parent(s): 3a22490

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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: