Create app.py
Browse files
app.py
CHANGED
@@ -1,162 +1,67 @@
|
|
1 |
-
|
2 |
-
import
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
from
|
7 |
-
|
8 |
-
from
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
"
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
"
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
|
67 |
-
|
68 |
-
f"${bill:.2f}"
|
69 |
-
|
70 |
-
|
71 |
-
with ui.layout_columns(col_widths=[6, 6, 12]):
|
72 |
-
with ui.card(full_screen=True):
|
73 |
-
ui.card_header("Tips data")
|
74 |
-
|
75 |
-
@render.data_frame
|
76 |
-
def table():
|
77 |
-
return render.DataGrid(tips_data())
|
78 |
-
|
79 |
-
with ui.card(full_screen=True):
|
80 |
-
with ui.card_header(class_="d-flex justify-content-between align-items-center"):
|
81 |
-
"Total bill vs tip"
|
82 |
-
with ui.popover(title="Add a color variable", placement="top"):
|
83 |
-
ICONS["ellipsis"]
|
84 |
-
ui.input_radio_buttons(
|
85 |
-
"scatter_color",
|
86 |
-
None,
|
87 |
-
["none", "sex", "smoker", "day", "time"],
|
88 |
-
inline=True,
|
89 |
-
)
|
90 |
-
|
91 |
-
@render_plotly
|
92 |
-
def scatterplot():
|
93 |
-
color = input.scatter_color()
|
94 |
-
return px.scatter(
|
95 |
-
tips_data(),
|
96 |
-
x="total_bill",
|
97 |
-
y="tip",
|
98 |
-
color=None if color == "none" else color,
|
99 |
-
trendline="lowess",
|
100 |
-
)
|
101 |
-
|
102 |
-
with ui.card(full_screen=True):
|
103 |
-
with ui.card_header(class_="d-flex justify-content-between align-items-center"):
|
104 |
-
"Tip percentages"
|
105 |
-
with ui.popover(title="Add a color variable"):
|
106 |
-
ICONS["ellipsis"]
|
107 |
-
ui.input_radio_buttons(
|
108 |
-
"tip_perc_y",
|
109 |
-
"Split by:",
|
110 |
-
["sex", "smoker", "day", "time"],
|
111 |
-
selected="day",
|
112 |
-
inline=True,
|
113 |
-
)
|
114 |
-
|
115 |
-
@render_plotly
|
116 |
-
def tip_perc():
|
117 |
-
from ridgeplot import ridgeplot
|
118 |
-
|
119 |
-
dat = tips_data()
|
120 |
-
dat["percent"] = dat.tip / dat.total_bill
|
121 |
-
yvar = input.tip_perc_y()
|
122 |
-
uvals = dat[yvar].unique()
|
123 |
-
|
124 |
-
samples = [[dat.percent[dat[yvar] == val]] for val in uvals]
|
125 |
-
|
126 |
-
plt = ridgeplot(
|
127 |
-
samples=samples,
|
128 |
-
labels=uvals,
|
129 |
-
bandwidth=0.01,
|
130 |
-
colorscale="viridis",
|
131 |
-
colormode="row-index",
|
132 |
-
)
|
133 |
-
|
134 |
-
plt.update_layout(
|
135 |
-
legend=dict(
|
136 |
-
orientation="h", yanchor="bottom", y=1.02, xanchor="center", x=0.5
|
137 |
-
)
|
138 |
-
)
|
139 |
-
|
140 |
-
return plt
|
141 |
-
|
142 |
-
|
143 |
-
ui.include_css(app_dir / "styles.css")
|
144 |
-
|
145 |
-
# --------------------------------------------------------
|
146 |
-
# Reactive calculations and effects
|
147 |
-
# --------------------------------------------------------
|
148 |
-
|
149 |
-
|
150 |
-
@reactive.calc
|
151 |
-
def tips_data():
|
152 |
-
bill = input.total_bill()
|
153 |
-
idx1 = tips.total_bill.between(bill[0], bill[1])
|
154 |
-
idx2 = tips.time.isin(input.time())
|
155 |
-
return tips[idx1 & idx2]
|
156 |
-
|
157 |
-
|
158 |
-
@reactive.effect
|
159 |
-
@reactive.event(input.reset)
|
160 |
-
def _():
|
161 |
-
ui.update_slider("total_bill", value=bill_rng)
|
162 |
-
ui.update_checkbox_group("time", selected=["Lunch", "Dinner"])
|
|
|
1 |
+
from transformers import pipeline
|
2 |
+
import speech_recognition as sr
|
3 |
+
from gtts import gTTS
|
4 |
+
import os
|
5 |
+
import io
|
6 |
+
from google.oauth2.credentials import Credentials
|
7 |
+
from googleapiclient.discovery import build
|
8 |
+
from googleapiclient.http import MediaIoBaseDownload
|
9 |
+
|
10 |
+
# Инициализация модели и токенизатора
|
11 |
+
model_name = "cointegrated/rut5-base-multitask"
|
12 |
+
chatbot = pipeline("text-generation", model=model_name)
|
13 |
+
|
14 |
+
# Функция для распознавания речи
|
15 |
+
def recognize_speech():
|
16 |
+
recognizer = sr.Recognizer()
|
17 |
+
with sr.Microphone() as source:
|
18 |
+
print("Скажите что-нибудь...")
|
19 |
+
audio = recognizer.listen(source)
|
20 |
+
try:
|
21 |
+
text = recognizer.recognize_google(audio, language="ru-RU")
|
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 |
+
# Функция для загрузки документов с Google Drive
|
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)
|
41 |
+
results = service.files().list(q="mimeType='application/pdf' and trashed=false", fields="files(id, name)").execute()
|
42 |
+
items = results.get('files', [])
|
43 |
+
documents = []
|
44 |
+
for item in items:
|
45 |
+
file_id = item['id']
|
46 |
+
request = service.files().get_media(fileId=file_id)
|
47 |
+
fh = io.FileIO(item['name'], 'wb')
|
48 |
+
downloader = MediaIoBaseDownload(fh, request)
|
49 |
+
done = False
|
50 |
+
while done is False:
|
51 |
+
status, done = downloader.next_chunk()
|
52 |
+
print(f"Скачивание файла {item['name']}... {int(status.progress() * 100)}% завершено")
|
53 |
+
fh.close()
|
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, num_return_sequences=1)[0]['generated_text']
|
63 |
+
print("Чатбот: " + response)
|
64 |
+
synthesize_speech(response)
|
65 |
+
|
66 |
+
if __name__ == "__main__":
|
67 |
+
main()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|