salomonsky commited on
Commit
0a6ca66
·
verified ·
1 Parent(s): 6036fba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -7,9 +7,14 @@ from audiorecorder import audiorecorder
7
  import speech_recognition as sr
8
  from pydub import AudioSegment
9
 
 
 
10
  if "history" not in st.session_state:
11
  st.session_state.history = []
12
 
 
 
 
13
  def recognize_speech(audio_data, show_messages=True):
14
  recognizer = sr.Recognizer()
15
  audio_recording = sr.AudioFile(audio_data)
@@ -35,6 +40,10 @@ def recognize_speech(audio_data, show_messages=True):
35
  def format_prompt(message, history):
36
  prompt = "<s>"
37
 
 
 
 
 
38
  for user_prompt, bot_response in history:
39
  prompt += f"[INST] {user_prompt} [/INST]"
40
  prompt += f" {bot_response}</s> "
@@ -97,4 +106,4 @@ def main():
97
  unsafe_allow_html=True)
98
 
99
  if __name__ == "__main__":
100
- main()
 
7
  import speech_recognition as sr
8
  from pydub import AudioSegment
9
 
10
+ pre_prompt_text = "Hablarás español, tus principios el estoicismo, eres una IA conductual, tus respuestas serán breves."
11
+
12
  if "history" not in st.session_state:
13
  st.session_state.history = []
14
 
15
+ if "pre_prompt_sent" not in st.session_state:
16
+ st.session_state.pre_prompt_sent = False
17
+
18
  def recognize_speech(audio_data, show_messages=True):
19
  recognizer = sr.Recognizer()
20
  audio_recording = sr.AudioFile(audio_data)
 
40
  def format_prompt(message, history):
41
  prompt = "<s>"
42
 
43
+ if not st.session_state.pre_prompt_sent:
44
+ prompt += f"[INST] {pre_prompt_text} [/INST]"
45
+ st.session_state.pre_prompt_sent = True
46
+
47
  for user_prompt, bot_response in history:
48
  prompt += f"[INST] {user_prompt} [/INST]"
49
  prompt += f" {bot_response}</s> "
 
106
  unsafe_allow_html=True)
107
 
108
  if __name__ == "__main__":
109
+ main()