Update app.py

app.py

@@ -188,6 +188,62 @@ 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):
+    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"
+    )
+
+    for message in personalized_messages:
+        error_prompt += f"\n{message}\n"
+
+    return error_prompt.strip()
+
+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, "")
+    yield error_check_prompt, error_message_gpt4o, "", "", "", "Результат проверки GPT-4o"
+
+    error_message_gigachat_pro = perform_personalization_gigachat(error_check_prompt, "", "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")
+    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")
+    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-сообщений по заданным признакам")
@@ -255,5 +311,33 @@ 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_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=[
+                output_text_gpt4o,
+                output_text_gigachat_pro,
+                output_text_gigachat_lite,
+                output_text_gigachat_plus
+            ],
+            outputs=[
+                error_check_prompt_display,
+                error_message_gpt4o,
+                error_message_gigachat_pro,
+                error_message_gigachat_lite,
+                error_message_gigachat_plus,
+                status_display
+            ]
+        )
+
 
     demo.launch()