fruitpicker01 commited on
Commit
132b01f
·
verified ·
1 Parent(s): e76af6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +47 -127
app.py CHANGED
@@ -4,45 +4,12 @@ import os
4
  import pandas as pd
5
  import time
6
  from langchain.schema import SystemMessage
7
- from langchain_community.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
- # Правила для проверки
14
- rules = (
15
- "Проверь текст на наличие следующих ошибок:\n"
16
- "- Запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, доступный, банкротство, долги, займ, срочно, сейчас, лучший, главный, номер 1, гарантия, успех, лидер;\n"
17
- "- Обращение к клиенту;\n"
18
- "- Приветствие клиента;\n"
19
- "- Обещания и гарантии;\n"
20
- "- Использовать составные конструкции из двух глаголов;\n"
21
- "- Причастия и причастные обороты;\n"
22
- "- Деепричастия и деепричастные обороты;\n"
23
- "- Превосходная степень прилагательных;\n"
24
- "- Страдательный залог;\n"
25
- "- Порядковые числительные от 10 прописью;\n"
26
- "- Цепочки с придаточными предложениями;\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
- "Текст для проверки ниже."
44
- )
45
-
46
  # Авторизация в сервисе GigaChat
47
  chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
48
  chat_lite = GigaChat(credentials=gc_key, model='GigaChat', verify_ssl_certs=False)
@@ -221,81 +188,60 @@ def personalize_messages_with_yield(standard_message, *selected_values):
221
  personalized_message_gigachat_plus = perform_personalization_gigachat(standard_message, personalization_prompt, "gigachat_plus")
222
  yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "Все персонализированные сообщения сгенерированы"
223
 
224
- def generate_error_check_prompt(rules):
225
- error_prompt = (
226
- f"{rules}\n"
227
- "Текст для проверки ниже."
228
- )
229
- return error_prompt
230
-
231
- def check_errors_with_yield(rules, *personalized_messages):
232
- error_check_prompt = generate_error_check_prompt(rules)
233
- yield error_check_prompt, "", "", "", "", "Промпт для проверки текста сгенерирован"
234
-
235
- error_message_gpt4o = perform_personalization(f"{error_check_prompt}\n\n{personalized_messages[0]}", "")
236
- yield error_check_prompt, error_message_gpt4o, "", "", "", "Результат проверки GPT-4o"
237
-
238
- error_message_gigachat_pro = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[1]}", "", "gigachat_pro")
239
- yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, "", "", "Результат проверки GigaChat-Pro"
240
-
241
- error_message_gigachat_lite = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[2]}", "", "gigachat_lite")
242
- yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, "", "Результат проверки GigaChat-Lite"
243
-
244
- error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[3]}", "", "gigachat_plus")
245
- yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
246
-
247
  # Создание интерфейса Gradio
248
  with gr.Blocks() as demo:
249
  gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
250
 
251
  with gr.Row():
252
- description_input = gr.Textbox(
253
- label="Описание предложения (предзаполненный пример можно поменять на свой)",
254
- lines=13,
255
- value=(
256
- "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
257
- "Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
258
- "Что необходимо сделать, чтобы воспользоваться предложением:\n"
259
- "1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
260
- "2. Забрать карту.\n"
261
- "3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
262
- "4. В течение следующего месяца пользоваться ей бесплатно."
 
 
263
  )
264
- )
265
- advantages_input = gr.Textbox(
266
- label="Преимущества (предзаполненный пример можно поменять на свой)",
267
- lines=6,
268
- value=(
269
- "Предложение по бесплатному обслуживанию бессрочное.\n"
270
- "Оплата покупок без отчётов и платёжных поручений.\n"
271
- "Платёжные документы без комиссии.\n"
272
- "Лимиты на расходы сотрудников.\n"
273
- "Мгновенные переводы на карты любых банков."
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
274
  )
275
- )
276
- selections = []
277
- for feature in features.keys():
278
- selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
279
-
280
- with gr.Row():
281
- submit_btn = gr.Button("1. Создать неперсонализированное сообщение")
282
- personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)")
283
- check_errors_btn = gr.Button("3. Проверить ошибки", elem_id="check_errors_button")
284
-
285
- with gr.Row():
286
- prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=16, interactive=False)
287
- output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3)
288
- output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3)
289
- output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3)
290
- output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3)
291
- status_display = gr.Textbox(label="", lines=4, interactive=False, elem_classes="status-box")
292
-
293
- submit_btn.click(
294
- generate_messages,
295
- inputs=[description_input, advantages_input] + selections,
296
- outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
297
- )
298
 
 
 
 
299
  with gr.Row():
300
  personalize_btn.click(
301
  personalize_messages_with_yield,
@@ -309,31 +255,5 @@ with gr.Blocks() as demo:
309
  status_display
310
  ]
311
  )
312
-
313
- with gr.Row():
314
- error_check_prompt_display = gr.Textbox(label="Промпт для проверки текста", lines=6, interactive=False)
315
- error_message_gpt4o = gr.Textbox(label="Результат проверки GPT-4o", lines=6)
316
- error_message_gigachat_pro = gr.Textbox(label="Результат проверки GigaChat-Pro", lines=6)
317
- error_message_gigachat_lite = gr.Textbox(label="Результат проверки GigaChat-Lite", lines=6)
318
- error_message_gigachat_plus = gr.Textbox(label="Результат проверки GigaChat-Lite+", lines=6)
319
-
320
- check_errors_btn.click(
321
- lambda *personalized_messages: check_errors_with_yield(rules, *personalized_messages), # Передаем rules внутри лямбды
322
- inputs=[
323
- output_text_gpt4o,
324
- output_text_gigachat_pro,
325
- output_text_gigachat_lite,
326
- output_text_gigachat_plus
327
- ],
328
- outputs=[
329
- error_check_prompt_display,
330
- error_message_gpt4o,
331
- error_message_gigachat_pro,
332
- error_message_gigachat_lite,
333
- error_message_gigachat_plus,
334
- status_display
335
- ]
336
- )
337
 
338
-
339
  demo.launch()
 
4
  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
  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
+ with gr.Column(scale=1):
197
+ description_input = gr.Textbox(
198
+ label="Описание предложения (предзаполненный пример можно поменять на свой)",
199
+ lines=13,
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
+ personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
244
+
245
  with gr.Row():
246
  personalize_btn.click(
247
  personalize_messages_with_yield,
 
255
  status_display
256
  ]
257
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
258
 
 
259
  demo.launch()