Update app.py

app.py

@@ -822,6 +822,82 @@ def reset_button_text_2():
     time.sleep(2)
     return gr.update(value="Я предпочитаю это SMS", visible=True)
 
+def check_source_fields(description, product_name, benefits, key_message):
+
+    results = []
+
+    # Проверяем "Описание предложения"
+    desc_checks = perform_checks(description, "")
+    not_passed_desc = extract_failed_checks(desc_checks)  # функция, которая из словаря checks вытягивает только не пройденные проверки
+    if not_passed_desc:
+        results.append(f"Описание предложения:\n{not_passed_desc}")
+
+    # Проверяем "Наименование продукта"
+    name_checks = perform_checks(product_name, "")
+    not_passed_name = extract_failed_checks(name_checks)
+    if not_passed_name:
+        results.append(f"Наименование продукта:\n{not_passed_name}")
+
+    # Проверяем "Преимущества"
+    ben_checks = perform_checks(benefits, "")
+    not_passed_ben = extract_failed_checks(ben_checks)
+    if not_passed_ben:
+        results.append(f"Преимущества:\n{not_passed_ben}")
+
+    # Проверяем "Ключевое сообщение"
+    km_checks = perform_checks(key_message, "")
+    not_passed_km = extract_failed_checks(km_checks)
+    if not_passed_km:
+        results.append(f"Ключевое сообщение:\n{not_passed_km}")
+
+    if not results:
+        # Всё пройдено
+        return "Проверка исходных данных пройдена"
+    else:
+        # Склеиваем блоки
+        return "\n\n".join(results)
+
+
+def extract_failed_checks(checks_dict):
+    lines = []
+    for rule_key, result in checks_dict.items():
+        # Определяем, было ли нарушение
+        if isinstance(result, tuple):
+            passed, reason = result
+            if not passed:  # passed == False
+                # например: "Запрещенные слова: успех"
+                lines.append(f"{rule_to_str(rule_key)}: {reason}")
+        elif result is False:
+            # Если функция вернула просто False без пояснения
+            lines.append(f"{rule_to_str(rule_key)}: (без пояснения)")
+    return "\n".join(lines)
+
+
+def rule_to_str(rule_key):
+    translation = {
+        "forbidden_words": "Запрещенные слова",
+        "client_addressing": "Обращение к клиенту",
+        "promises": "Обещания и гарантии",
+        "double_verbs": "Два глагола подряд",
+        "participles": "Причастия",
+        "adverbial_participles": "Деепричастия",
+        "superlative_adjectives": "Превосходная степень",
+        "passive_voice": "Страдательный залог",
+        "written_out_ordinals": "Порядковые числительные",
+        "subordinate_clauses_chain": "Цепочки придаточных",
+        "repeating_conjunctions": "Повторяющиеся союзы",
+        "introductory_phrases": "Вводные конструкции",
+        "amplifiers": "Усилители",
+        "time_parasites": "Паразиты времени",
+        "multiple_nouns": "Несколько существительных подряд",
+        "derived_prepositions": "Производные предлоги",
+        "compound_sentences": "Сложноподчиненные предложения",
+        "dates_written_out": "Даты прописью",
+        "no_word_repetitions": "Повторы слов",
+    }
+    return translation.get(rule_key, rule_key)
+
+
 # ФУНКЦИИ ПРОВЕРОК (НАЧАЛО)
 
 # 1. Запрещенные слова
@@ -1336,13 +1412,15 @@ with gr.Blocks(theme="default") as demo:
         with gr.Column(scale=1):
             gr.Markdown("**Продукт**")
             product_dropdown = gr.Dropdown(label="Продукт", choices=products, value=products[0])
-            description = gr.Textbox(label="Описание предложения", lines=5, value="", interactive=True)
+            description = gr.Textbox(label="Описание предложения", lines=4, value="", interactive=True)
             product_name = gr.Textbox(label="Наименование продукта", lines=1, value="", interactive=True)
-            benefits = gr.Textbox(label="Преимущества", lines=9, value="", interactive=True)
+            benefits = gr.Textbox(label="Преимущества", lines=6, value="", interactive=True)
             key_message = gr.Textbox(label="Ключевое сообщение", lines=2, value="")
+            source_check_md = gr.Textbox(label="Результат проверки исходных данных", lines=2, value="")
 
             def on_product_change(selected, description, product_name, benefits, key_message,
                                   gender, generation, psychotype, business_stage, industry, opf):
+
                 if selected == "Свой продукт":
                     new_desc = ""
                     new_pname = ""
@@ -1360,15 +1438,28 @@ with gr.Blocks(theme="default") as demo:
                         new_pname = ""
                         new_ben = ""
                         new_kmsg = ""
-                chosen_approach_val, p1, p2 = update_prompts_on_params_change(new_desc, new_pname, new_ben, new_kmsg,
-                                                                              gender, generation, psychotype,
-                                                                              business_stage, industry, opf)
-                return (gr.update(value=new_desc, interactive=(selected=="Свой продукт")),
-                        gr.update(value=new_pname, interactive=(selected=="Свой продукт")),
-                        gr.update(value=new_ben, interactive=(selected=="Свой продукт")),
-                        gr.update(value=new_kmsg, interactive=(selected=="Свой продукт")),
-                        chosen_approach_val, p1, p2,
-                        "", "", "", "", "", "")
+            
+                chosen_approach_val, p1, p2 = update_prompts_on_params_change(
+                    new_desc, new_pname, new_ben, new_kmsg,
+                    gender, generation, psychotype,
+                    business_stage, industry, opf
+                )
+            
+                source_fields_report = check_source_fields(new_desc, new_pname, new_ben, new_kmsg)
+            
+                return (
+                    gr.update(value=new_desc, interactive=(selected=="Свой продукт")),
+                    gr.update(value=new_pname, interactive=(selected=="Свой продукт")),
+                    gr.update(value=new_ben, interactive=(selected=="Свой продукт")),
+                    gr.update(value=new_kmsg, interactive=(selected=="Свой продукт")),
+                    chosen_approach_val, p1, p2,
+                    "",
+                    "",
+                    "",
+                    "",
+                    "",
+                    source_fields_report
+                )
 
         with gr.Column(scale=1):
             gr.Markdown("**Клиент**")
@@ -1424,20 +1515,86 @@ with gr.Blocks(theme="default") as demo:
 
     product_dropdown.change(
         fn=on_product_change,
-        inputs=[product_dropdown, description, product_name, benefits, key_message,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[description, product_name, benefits, key_message,
-                 chosen_approach, prompt_1, prompt_2,
-                 sms_1, sms_2, comment_sms_1, comment_sms_2, corrected_sms_1, corrected_sms_2]
+        inputs=[
+            product_dropdown, description, product_name, benefits, key_message,
+            gender_dropdown, generation_dropdown, psychotype_dropdown,
+            business_stage_dropdown, industry_dropdown, opf_dropdown
+        ],
+        outputs=[
+            description, product_name, benefits, key_message,
+            chosen_approach, prompt_1, prompt_2,
+            sms_1, sms_2, comment_sms_1, comment_sms_2, corrected_sms_1, corrected_sms_2,
+            source_check_md
+        ]
+    )
+
+    description.change(
+        fn=params_change_wrapper,
+        inputs=[
+            description, product_name, benefits, key_message,
+            gender_dropdown, generation_dropdown, psychotype_dropdown,
+            business_stage_dropdown, industry_dropdown, opf_dropdown
+        ],
+        outputs=[
+            chosen_approach, prompt_1, prompt_2,
+            sms_1, sms_2, comment_sms_1, comment_sms_2, corrected_sms_1, corrected_sms_2,
+            source_check_md
+        ]
+    )
+
+    product_name.change(
+        fn=params_change_wrapper,
+        inputs=[
+            description, product_name, benefits, key_message,
+            gender_dropdown, generation_dropdown, psychotype_dropdown,
+            business_stage_dropdown, industry_dropdown, opf_dropdown
+        ],
+        outputs=[
+            chosen_approach, prompt_1, prompt_2,
+            sms_1, sms_2, comment_sms_1, comment_sms_2, corrected_sms_1, corrected_sms_2,
+            source_check_md
+        ]
+    )
+
+    benefits.change(
+        fn=params_change_wrapper,
+        inputs=[
+            description, product_name, benefits, key_message,
+            gender_dropdown, generation_dropdown, psychotype_dropdown,
+            business_stage_dropdown, industry_dropdown, opf_dropdown
+        ],
+        outputs=[
+            chosen_approach, prompt_1, prompt_2,
+            sms_1, sms_2, comment_sms_1, comment_sms_2, corrected_sms_1, corrected_sms_2,
+            source_check_md
+        ]
+    )
+
+    key_message.change(
+        fn=params_change_wrapper,
+        inputs=[
+            description, product_name, benefits, key_message,
+            gender_dropdown, generation_dropdown, psychotype_dropdown,
+            business_stage_dropdown, industry_dropdown, opf_dropdown
+        ],
+        outputs=[
+            chosen_approach, prompt_1, prompt_2,
+            sms_1, sms_2, comment_sms_1, comment_sms_2, corrected_sms_1, corrected_sms_2,
+            source_check_md
+        ]
     )
 
     def params_change_wrapper(description, product_name, benefits, key_message,
                               gender, generation, psychotype, business_stage, industry, opf):
-        chosen_approach_val, p1, p2 = update_prompts_on_params_change(description, product_name, benefits, key_message,
-                                                                     gender, generation, psychotype,
-                                                                     business_stage, industry, opf)
-        return chosen_approach_val, p1, p2, "", "", "", "", "", ""
+        chosen_approach_val, p1, p2 = update_prompts_on_params_change(
+            description, product_name, benefits, key_message,
+            gender, generation, psychotype,
+            business_stage, industry, opf
+        )
+    
+        source_fields_report = check_source_fields(description, product_name, benefits, key_message)
+    
+        return chosen_approach_val, p1, p2, "", "", "", "", "", source_fields_report
 
     client_params = [gender_dropdown, generation_dropdown, psychotype_dropdown,
                      business_stage_dropdown, industry_dropdown, opf_dropdown]