Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
1 |
from transformers import pipeline
|
2 |
import speech_recognition as sr
|
3 |
from gtts import gTTS
|
@@ -22,19 +27,19 @@ def recognize_speech():
|
|
22 |
print("Вы сказали: " + text)
|
23 |
return text
|
24 |
except sr.UnknownValueError:
|
25 |
-
print("Не удалось
|
26 |
return ""
|
27 |
except sr.RequestError as e:
|
28 |
print("Ошибка сервиса распознавания речи; {0}".format(e))
|
29 |
return ""
|
30 |
|
31 |
-
# Функция для
|
32 |
def synthesize_speech(text):
|
33 |
tts = gTTS(text=text, lang="ru")
|
34 |
tts.save("response.mp3")
|
35 |
os.system("mpg321 response.mp3")
|
36 |
|
37 |
-
# Функция для
|
38 |
def load_documents_from_drive():
|
39 |
creds = Credentials.from_authorized_user_file('token.json', ['https://www.googleapis.com/auth/drive.readonly'])
|
40 |
service = build('drive', 'v3', credentials=creds)
|
@@ -54,12 +59,12 @@ def load_documents_from_drive():
|
|
54 |
documents.append(item['name'])
|
55 |
return documents
|
56 |
|
57 |
-
#
|
58 |
def main():
|
59 |
while True:
|
60 |
user_input = recognize_speech()
|
61 |
if user_input:
|
62 |
-
response = chatbot(user_input, max_length=100,
|
63 |
print("Чатбот: " + response)
|
64 |
synthesize_speech(response)
|
65 |
|
|
|
1 |
+
import os
|
2 |
+
|
3 |
+
# Настройка переменной окружения
|
4 |
+
os.environ['TRANSFORMERS_CACHE'] = '/path/to/writable/directory'
|
5 |
+
|
6 |
from transformers import pipeline
|
7 |
import speech_recognition as sr
|
8 |
from gtts import gTTS
|
|
|
27 |
print("Вы сказали: " + text)
|
28 |
return text
|
29 |
except sr.UnknownValueError:
|
30 |
+
print("Не удалось распознавать речь")
|
31 |
return ""
|
32 |
except sr.RequestError as e:
|
33 |
print("Ошибка сервиса распознавания речи; {0}".format(e))
|
34 |
return ""
|
35 |
|
36 |
+
# Функция для синтезиса речи
|
37 |
def synthesize_speech(text):
|
38 |
tts = gTTS(text=text, lang="ru")
|
39 |
tts.save("response.mp3")
|
40 |
os.system("mpg321 response.mp3")
|
41 |
|
42 |
+
# Функция для загрузкии документов с Google Drive
|
43 |
def load_documents_from_drive():
|
44 |
creds = Credentials.from_authorized_user_file('token.json', ['https://www.googleapis.com/auth/drive.readonly'])
|
45 |
service = build('drive', 'v3', credentials=creds)
|
|
|
59 |
documents.append(item['name'])
|
60 |
return documents
|
61 |
|
62 |
+
# Основой цикл чатбота
|
63 |
def main():
|
64 |
while True:
|
65 |
user_input = recognize_speech()
|
66 |
if user_input:
|
67 |
+
response = chatbot(user_input, max_length=100, numreturn_sequences=1)[0]['generated_text']
|
68 |
print("Чатбот: " + response)
|
69 |
synthesize_speech(response)
|
70 |
|