Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,11 +5,45 @@ import pandas as pd
|
|
5 |
import time
|
6 |
from langchain.schema import SystemMessage
|
7 |
from langchain.chat_models.gigachat import GigaChat
|
|
|
8 |
|
9 |
# Установка ключа API для OpenAI и GigaChat
|
10 |
openai_api_key = os.getenv('GPT_KEY')
|
11 |
gc_key = os.getenv('GC_KEY')
|
12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
# Авторизация в сервисе GigaChat
|
14 |
chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
|
15 |
chat_lite = GigaChat(credentials=gc_key, model='GigaChat', verify_ssl_certs=False)
|
@@ -188,60 +222,52 @@ def personalize_messages_with_yield(standard_message, *selected_values):
|
|
188 |
personalized_message_gigachat_plus = perform_personalization_gigachat(standard_message, personalization_prompt, "gigachat_plus")
|
189 |
yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "Все персонализированные сообщения сгенерированы"
|
190 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
191 |
# Создание интерфейса Gradio
|
192 |
with gr.Blocks() as demo:
|
193 |
gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
|
194 |
|
195 |
with gr.Row():
|
196 |
-
|
197 |
-
|
198 |
-
|
199 |
-
|
200 |
-
value=(
|
201 |
-
"Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
|
202 |
-
"Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
|
203 |
-
"Что необходимо сделать, чтобы воспользоваться предложением:\n"
|
204 |
-
"1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
|
205 |
-
"2. Забрать карту.\n"
|
206 |
-
"3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
|
207 |
-
"4. В течение следующего месяца пользоваться ей бесплатно."
|
208 |
-
)
|
209 |
-
)
|
210 |
-
advantages_input = gr.Textbox(
|
211 |
-
label="Преимущества (предзаполненный пример можно поменять на свой)",
|
212 |
-
lines=6,
|
213 |
-
value=(
|
214 |
-
"Предложение по бесплатному обслуживанию — бессрочное.\n"
|
215 |
-
"Оплата покупок без отчётов и платёжных поручений.\n"
|
216 |
-
"Платёжные документы без комиссии.\n"
|
217 |
-
"Лимиты на расходы сотрудников.\n"
|
218 |
-
"Мгновенные переводы на карты любых банков."
|
219 |
-
)
|
220 |
-
)
|
221 |
-
selections = []
|
222 |
-
for feature in features.keys():
|
223 |
-
selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
|
224 |
-
|
225 |
-
submit_btn = gr.Button("1. Создать неперсонализированное сообщение") # Оранжевая кнопка по умолчанию
|
226 |
-
|
227 |
-
with gr.Column(scale=2):
|
228 |
-
prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=16, interactive=False)
|
229 |
-
output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3)
|
230 |
-
output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3)
|
231 |
-
output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3)
|
232 |
-
output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3)
|
233 |
-
gr.Markdown("**Статус выполнения**")
|
234 |
-
status_display = gr.Textbox(label="", lines=4, interactive=False, elem_classes="status-box") # Новый блок для отображения статуса
|
235 |
-
|
236 |
-
submit_btn.click(
|
237 |
-
generate_messages,
|
238 |
-
inputs=[description_input, advantages_input] + selections,
|
239 |
-
outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
|
240 |
-
)
|
241 |
-
|
242 |
with gr.Row():
|
243 |
-
|
244 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
245 |
with gr.Row():
|
246 |
personalize_btn.click(
|
247 |
personalize_messages_with_yield,
|
@@ -255,5 +281,31 @@ with gr.Blocks() as demo:
|
|
255 |
status_display
|
256 |
]
|
257 |
)
|
258 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
259 |
demo.launch()
|
|
|
5 |
import time
|
6 |
from langchain.schema import SystemMessage
|
7 |
from langchain.chat_models.gigachat import GigaChat
|
8 |
+
#from langchain_community.chat_models.gigachat import GigaChat
|
9 |
|
10 |
# Установка ключа API для OpenAI и GigaChat
|
11 |
openai_api_key = os.getenv('GPT_KEY')
|
12 |
gc_key = os.getenv('GC_KEY')
|
13 |
|
14 |
+
# Правила для проверки
|
15 |
+
rules = (
|
16 |
+
"Проверь текст на наличие следующих ошибок:\n"
|
17 |
+
"- Запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, доступный, банкротство, долги, займ, срочно, сейчас, лучший, главный, номер 1, гарантия, успех, лидер;\n"
|
18 |
+
"- Обращение к клиенту;\n"
|
19 |
+
"- Приветствие клиента;\n"
|
20 |
+
"- Обещания и гарантии;\n"
|
21 |
+
"- Использовать составные конструкции из двух глаголов;\n"
|
22 |
+
"- Причастия и причастные обороты;\n"
|
23 |
+
"- Деепричастия и деепричастные обороты;\n"
|
24 |
+
"- Превосходная степень прилагательных;\n"
|
25 |
+
"- Страдательный залог;\n"
|
26 |
+
"- Порядковые числительные от 10 прописью;\n"
|
27 |
+
"- Цепочки с придаточными предложениями;\n"
|
28 |
+
"- Разделительные повторяющиеся союзы;\n"
|
29 |
+
"- Вводные конструкции;\n"
|
30 |
+
"- Усилители;\n"
|
31 |
+
"- Паразиты времени;\n"
|
32 |
+
"- Несколько существительных подряд, в том числе отглагольных;\n"
|
33 |
+
"- Производные предлоги;\n"
|
34 |
+
"- Сложные предложения, в которых нет связи между частями;\n"
|
35 |
+
"- Сложноподчинённые предложения;\n"
|
36 |
+
"- Даты прописью;\n"
|
37 |
+
"- Близкие по смыслу однородные члены предложения;\n"
|
38 |
+
"- Шокирующие, экстравагантные, кликбейтные фразы;\n"
|
39 |
+
"- Абстрактные заявления без поддержки фактами и отсутствие доказательства пользы для клиента;\n"
|
40 |
+
"- Гарантирующие фразы;\n"
|
41 |
+
"- Узкоспециализированные термины;\n"
|
42 |
+
"- Фразы, способные создать двойственное ощущение, обидеть;\n"
|
43 |
+
"- Речевые клише, рекламные штампы, канцеляризмы;\n"
|
44 |
+
"Текст для проверки ниже."
|
45 |
+
)
|
46 |
+
|
47 |
# Авторизация в сервисе GigaChat
|
48 |
chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
|
49 |
chat_lite = GigaChat(credentials=gc_key, model='GigaChat', verify_ssl_certs=False)
|
|
|
222 |
personalized_message_gigachat_plus = perform_personalization_gigachat(standard_message, personalization_prompt, "gigachat_plus")
|
223 |
yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "Все персонализированные сообщения сгенерированы"
|
224 |
|
225 |
+
def generate_error_check_prompt(rules):
|
226 |
+
error_prompt = (
|
227 |
+
f"{rules}\n"
|
228 |
+
"Текст для проверки ниже."
|
229 |
+
)
|
230 |
+
return error_prompt
|
231 |
+
|
232 |
+
def check_errors_with_yield(rules, *personalized_messages):
|
233 |
+
error_check_prompt = generate_error_check_prompt(rules)
|
234 |
+
yield error_check_prompt, "", "", "", "", "Промпт для проверки текста сгенерирован"
|
235 |
+
|
236 |
+
error_message_gpt4o = perform_personalization(f"{error_check_prompt}\n\n{personalized_messages[0]}", "")
|
237 |
+
yield error_check_prompt, error_message_gpt4o, "", "", "", "Результат проверки GPT-4o"
|
238 |
+
|
239 |
+
error_message_gigachat_pro = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[1]}", "", "gigachat_pro")
|
240 |
+
yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, "", "", "Результат проверки GigaChat-Pro"
|
241 |
+
|
242 |
+
error_message_gigachat_lite = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[2]}", "", "gigachat_lite")
|
243 |
+
yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, "", "Результат проверки GigaChat-Lite"
|
244 |
+
|
245 |
+
error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[3]}", "", "gigachat_plus")
|
246 |
+
yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
|
247 |
+
|
248 |
# Создание интерфейса Gradio
|
249 |
with gr.Blocks() as demo:
|
250 |
gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
|
251 |
|
252 |
with gr.Row():
|
253 |
+
submit_btn = gr.Button("1. Создать неперсонализированное сообщение")
|
254 |
+
personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)")
|
255 |
+
check_errors_btn = gr.Button("3. Проверить ошибки", elem_id="check_errors_button")
|
256 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
257 |
with gr.Row():
|
258 |
+
prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=16, interactive=False)
|
259 |
+
output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3)
|
260 |
+
output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3)
|
261 |
+
output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3)
|
262 |
+
output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3)
|
263 |
+
status_display = gr.Textbox(label="", lines=4, interactive=False, elem_classes="status-box")
|
264 |
+
|
265 |
+
submit_btn.click(
|
266 |
+
generate_messages,
|
267 |
+
inputs=[description_input, advantages_input] + selections,
|
268 |
+
outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
|
269 |
+
)
|
270 |
+
|
271 |
with gr.Row():
|
272 |
personalize_btn.click(
|
273 |
personalize_messages_with_yield,
|
|
|
281 |
status_display
|
282 |
]
|
283 |
)
|
284 |
+
|
285 |
+
with gr.Row():
|
286 |
+
error_check_prompt_display = gr.Textbox(label="Промпт для проверки текста", lines=6, interactive=False)
|
287 |
+
error_message_gpt4o = gr.Textbox(label="Результат проверки GPT-4o", lines=6)
|
288 |
+
error_message_gigachat_pro = gr.Textbox(label="Результат проверки GigaChat-Pro", lines=6)
|
289 |
+
error_message_gigachat_lite = gr.Textbox(label="Результат проверки GigaChat-Lite", lines=6)
|
290 |
+
error_message_gigachat_plus = gr.Textbox(label="Результат проверки GigaChat-Lite+", lines=6)
|
291 |
+
|
292 |
+
check_errors_btn.click(
|
293 |
+
check_errors_with_yield,
|
294 |
+
inputs=[
|
295 |
+
rules,
|
296 |
+
output_text_gpt4o,
|
297 |
+
output_text_gigachat_pro,
|
298 |
+
output_text_gigachat_lite,
|
299 |
+
output_text_gigachat_plus
|
300 |
+
],
|
301 |
+
outputs=[
|
302 |
+
error_check_prompt_display,
|
303 |
+
error_message_gpt4o,
|
304 |
+
error_message_gigachat_pro,
|
305 |
+
error_message_gigachat_lite,
|
306 |
+
error_message_gigachat_plus,
|
307 |
+
status_display
|
308 |
+
]
|
309 |
+
)
|
310 |
+
|
311 |
demo.launch()
|