Update app.py

app.py

@@ -10,6 +10,39 @@ from langchain.chat_models.gigachat import GigaChat
 openai_api_key = os.getenv('GPT_KEY')
 gc_key = os.getenv('GC_KEY')
 
+# Правила для проверки
+rules = (
+    "Проверь текст на наличие следующих ошибок:\n"
+    "- Запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, доступный, банкротство, долги, займ, срочно, сейчас, лучший, главный, номер 1, гарантия, успех, лидер;\n"
+    "- Обращение к клиенту;\n"
+    "- Приветствие клиента;\n"
+    "- Обещания и гарантии;\n"
+    "- Использовать составные конструкции из двух глаголов;\n"
+    "- Причастия и причастные обороты;\n"
+    "- Деепричастия и деепричастные обороты;\n"
+    "- Превосходная степень прилагательных;\n"
+    "- Страдательный залог;\n"
+    "- Порядковые числительные от 10 прописью;\n"
+    "- Цепочки с придаточными предложениями;\n"
+    "- Разделительные повторяющиеся союзы;\n"
+    "- Вводные конструкции;\n"
+    "- Усилители;\n"
+    "- Паразиты времени;\n"
+    "- Несколько существительных подряд, в том числе отглагольных;\n"
+    "- Производные предлоги;\n"
+    "- Сложные предложения, в которых нет связи между частями;\n"
+    "- Сложноподчинённые предложения;\n"
+    "- Даты прописью;\n"
+    "- Близкие по смыслу однородные члены предложения;\n"
+    "- Шокирующие, экстравагантные, кликбейтные фразы;\n"
+    "- Абстрактные заявления без поддержки фактами и отсутствие доказательства пользы для клиента;\n"
+    "- Гарантирующие фразы;\n"
+    "- Узкоспециализированные термины;\n"
+    "- Фразы, способные создать двойственное ощущение, обидеть;\n"
+    "- Речевые клише, рекламные штампы, канцеляризмы;\n"
+    "Текст для проверки ниже."
+)
+
 # Авторизация в сервисе GigaChat
 chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
 chat_lite = GigaChat(credentials=gc_key, model='GigaChat', verify_ssl_certs=False)
@@ -188,116 +221,81 @@ def personalize_messages_with_yield(standard_message, *selected_values):
     personalized_message_gigachat_plus = perform_personalization_gigachat(standard_message, personalization_prompt, "gigachat_plus")
     yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "Все персонализированные сообщения сгенерированы"
 
-def generate_error_check_prompt(*personalized_messages):
+def generate_error_check_prompt(rules):
     error_prompt = (
-        "Проверь текст на наличие следующих ошибок:\n"
-        "В тексте смс запрещено использование:\n"
-        "- Запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, доступный, банкротство, долги, займ, срочно, сейчас, лучший, главный, номер 1, гарантия, успех, лидер;\n"
-        "- Обращение к клиенту;\n"
-        "- Приветствие клиента;\n"
-        "- Обещания и гарантии;\n"
-        "- Использовать составные конструкции из двух глаголов;\n"
-        "- Причастия и причастные обороты;\n"
-        "- Деепричастия и деепричастные обороты;\n"
-        "- Превосходная степень прилагательных;\n"
-        "- Страдательный залог;\n"
-        "- Порядковые числительные от 10 прописью;\n"
-        "- Цепочки с придаточными предложениями;\n"
-        "- Разделительные повторяющиеся союзы;\n"
-        "- Вводные конструкции;\n"
-        "- Усилители;\n"
-        "- Паразиты времени;\n"
-        "- Несколько существительных подряд, в том числе отглагольных;\n"
-        "- Производные предлоги;\n"
-        "- Сложные предложения, в которых нет связи между частями;\n"
-        "- Сложноподчинённые предложения;\n"
-        "- Даты прописью;\n"
-        "- Близкие по смыслу однородные члены предложения;\n"
-        "- Шокирующие, экстравагантные, кликбейтные фразы;\n"
-        "- Абстрактные заявления без поддержки фактами и отсутствие доказательства пользы для клиента;\n"
-        "- Гарантирующие фразы;\n"
-        "- Узкоспециализированные термины;\n"
-        "- Фразы, способные создать двойственное ощущение, обидеть;\n"
-        "- Речевые клише, рекламные штампы, канцеляризмы;\n"
-        "Текст для проверки: \n"
+        f"{rules}\n"
+        "Текст для проверки ниже."
     )
+    return error_prompt
 
-    for message in personalized_messages:
-        error_prompt += f"\n{message}\n"
-
-    return error_prompt.strip()
+def check_errors_with_yield(rules, *personalized_messages):
+    error_check_prompt = generate_error_check_prompt(rules)
+    yield error_check_prompt, "", "", "", "", "Промпт для проверки текста сгенерирован"
 
-def check_errors_with_yield(*personalized_messages):
-    error_check_prompt = generate_error_check_prompt(*personalized_messages)
-    yield error_check_prompt, "", "", "", "", "Промпт для проверки ошибок сгенерирован"
-
-    error_message_gpt4o = perform_personalization(error_check_prompt, "")
+    error_message_gpt4o = perform_personalization(f"{error_check_prompt}\n\n{personalized_messages[0]}", "")
     yield error_check_prompt, error_message_gpt4o, "", "", "", "Результат проверки GPT-4o"
 
-    error_message_gigachat_pro = perform_personalization_gigachat(error_check_prompt, "", "gigachat_pro")
+    error_message_gigachat_pro = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[1]}", "", "gigachat_pro")
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, "", "", "Результат проверки GigaChat-Pro"
 
-    error_message_gigachat_lite = perform_personalization_gigachat(error_check_prompt, "", "gigachat_lite")
+    error_message_gigachat_lite = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[2]}", "", "gigachat_lite")
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, "", "Результат проверки GigaChat-Lite"
 
-    error_message_gigachat_plus = perform_personalization_gigachat(error_check_prompt, "", "gigachat_plus")
+    error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[3]}", "", "gigachat_plus")
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
 
-
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
     
     with gr.Row():
-        with gr.Column(scale=1):
-            description_input = gr.Textbox(
-                label="Описание предложения (предзаполненный пример можно поменять на свой)", 
-                lines=13,
-                value=(
-                    "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
-                    "Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
-                    "Что необходимо сделать, чтобы воспользоваться предложением:\n"
-                    "1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
-                    "2. Забрать карту.\n"
-                    "3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
-                    "4. В течение следующего месяца пользоваться ей бесплатно."
-                )
-            )
-            advantages_input = gr.Textbox(
-                label="Преимущества (предзаполненный пример можно поменять на свой)", 
-                lines=6,
-                value=(
-                    "Предложение по бесплатному обслуживанию — бессрочное.\n"
-                    "Оплата покупок без отчётов и платёжных поручений.\n"
-                    "Платёжные документы без комиссии.\n"
-                    "Лимиты на расходы сотрудников.\n"
-                    "Мгновенные переводы на карты любых банков."
-                )
+        description_input = gr.Textbox(
+            label="Описание предложения (предзаполненный пример можно поменять на свой)", 
+            lines=13,
+            value=(
+                "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
+                "Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
+                "Что необходимо сделать, чтобы воспользоваться предложением:\n"
+                "1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
+                "2. Забрать карту.\n"
+                "3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
+                "4. В течение следующего месяца пользоваться ей бесплатно."
             )
-            selections = []
-            for feature in features.keys():
-                selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
-
-            submit_btn = gr.Button("1. Создать неперсонализированное сообщение")  # Оранжевая кнопка по умолчанию
-
-        with gr.Column(scale=2):
-            prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=16, interactive=False)
-            output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3)
-            output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3)
-            output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3)
-            output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3)
-            gr.Markdown("**Статус выполнения**")
-            status_display = gr.Textbox(label="", lines=4, interactive=False, elem_classes="status-box")  # Новый блок для отображения статуса
-            
-            submit_btn.click(
-                generate_messages, 
-                inputs=[description_input, advantages_input] + selections, 
-                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
+        )
+        advantages_input = gr.Textbox(
+            label="Преимущества (предзаполненный пример можно поменять на свой)", 
+            lines=6,
+            value=(
+                "Предложение по бесплатному обслуживанию — бессрочное.\n"
+                "Оплата покупок без отчётов и платёжных поручений.\n"
+                "Платёжные документы без комиссии.\n"
+                "Лимиты на расходы сотрудников.\n"
+                "Мгновенные переводы на карты любых банков."
             )
-
+        )
+        selections = []
+        for feature in features.keys():
+            selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
+
+        submit_btn = gr.Button("1. Создать неперсонализированное сообщение")  # Оранжевая кнопка по умолчанию
+
+        prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=16, interactive=False)
+        output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3)
+        output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3)
+        output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3)
+        output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3)
+        status_display = gr.Textbox(label="Статус выполнения", lines=4, interactive=False)
+
+        submit_btn.click(
+            generate_messages, 
+            inputs=[description_input, advantages_input] + selections, 
+            outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
+        )
+    
     with gr.Row():
         personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
-        
+        check_errors_btn = gr.Button("3. Проверить ошибки", elem_id="check_errors_button")
+    
     with gr.Row():
         personalize_btn.click(
             personalize_messages_with_yield,
@@ -311,19 +309,18 @@ with gr.Blocks() as demo:
                 status_display
             ]
         )
-            
+    
     with gr.Row():
-        check_errors_btn = gr.Button("3. Проверить ошибки", elem_id="check_errors_button")
-
-        error_check_prompt_display = gr.Textbox(label="Промпт для проверки ошибок", lines=6, interactive=False)
+        error_check_prompt_display = gr.Textbox(label="Промпт для проверки текста", lines=6, interactive=False)
         error_message_gpt4o = gr.Textbox(label="Результат проверки GPT-4o", lines=6)
         error_message_gigachat_pro = gr.Textbox(label="Результат проверки GigaChat-Pro", lines=6)
         error_message_gigachat_lite = gr.Textbox(label="Результат проверки GigaChat-Lite", lines=6)
         error_message_gigachat_plus = gr.Textbox(label="Результат проверки GigaChat-Lite+", lines=6)
-
+    
         check_errors_btn.click(
             check_errors_with_yield,
             inputs=[
+                rules,
                 output_text_gpt4o,
                 output_text_gigachat_pro,
                 output_text_gigachat_lite,
@@ -338,6 +335,5 @@ with gr.Blocks() as demo:
                 status_display
             ]
         )
-
-
+    
     demo.launch()