Update app.py

app.py

@@ -268,6 +268,369 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         # Небольшая пауза между выводом каждого сообщения
         time.sleep(1)
 
+# ФУНКЦИИ ПРОВЕРОК (НАЧАЛО)
+
+# 1. Запрещенные слова
+
+def check_forbidden_words(message):   
+    morph = pymorphy2.MorphAnalyzer()
+    
+    # Перечень запрещённых слов и фраз
+    forbidden_patterns = [
+        r'№\s?1\b', r'номер\sодин\b', r'номер\s1\b',
+        r'вкусный', r'дешёвый', r'продукт', 
+        r'спам', r'доступный', r'банкротство', r'долг[и]?', r'займ',
+        r'срочный', r'сейчас', r'главный',
+        r'гарантия', r'успех', r'лидер'
+    ]
+    
+    # Удаляем знаки препинания для корректного анализа
+    message_without_punctuation = message.translate(str.maketrans('', '', string.punctuation))
+    
+    # Проверка на наличие подстроки "лучш" (без учета регистра)
+    if re.search(r'лучш', message_without_punctuation, re.IGNORECASE):
+        return False
+    
+    # Лемматизация слов сообщения
+    words = message_without_punctuation.split()
+    lemmas = [morph.parse(word)[0].normal_form for word in words]
+    normalized_message = ' '.join(lemmas)
+
+    # Проверка на запрещённые фразы и леммы
+    for pattern in forbidden_patterns:
+        if re.search(pattern, normalized_message, re.IGNORECASE):
+            return False
+
+    return True
+
+
+# 2 и #3. Обращение к клиенту и приветствие клиента
+
+def check_no_greeting(message):
+    morph = pymorphy2.MorphAnalyzer()
+    # Список типичных обращений и приветствий
+    greeting_patterns = [
+        r"привет\b", r"здравствуй", r"добрый\s(день|вечер|утро)",
+        r"дорогой\b", r"уважаемый\b", r"дорогая\b", r"уважаемая\b",
+        r"господин\b", r"госпожа\b", r"друг\b", r"коллега\b",
+        r"товарищ\b", r"приятель\b", r"друг\b", r"подруга\b"
+    ]
+    
+    # Компилируем все шаблоны в один регулярное выражение
+    greeting_regex = re.compile('|'.join(greeting_patterns), re.IGNORECASE)
+    
+    # Проверяем, начинается ли сообщение с шаблона приветствия или обращения
+    if greeting_regex.search(message.strip()):
+        return False
+    return True
+
+# 4. Обещания и гарантии
+
+def check_no_promises(message):
+    morph = pymorphy2.MorphAnalyzer()
+    promise_patterns = [
+        "обещать", "гарантировать", "обязаться"
+    ]
+    
+    words = message.split()
+    lemmas = [morph.parse(word)[0].normal_form for word in words]
+    
+    for pattern in promise_patterns:
+        if pattern in lemmas:
+            return False
+    return True
+
+# 5. Составные конструкции из двух глаголов
+
+def check_no_double_verbs(message):
+    morph = pymorphy2.MorphAnalyzer()
+    # Разделяем текст по пробелам и знакам препинания
+    words = re.split(r'\s+|[.!?]', message)
+    morphs = [morph.parse(word)[0] for word in words]
+    
+    for i in range(len(morphs) - 1):
+        # Проверяем, что оба слова являются глаголами (в любой форме, включая инфинитивы)
+        if (morphs[i].tag.POS in {'VERB', 'INFN'}) and (morphs[i+1].tag.POS in {'VERB', 'INFN'}):
+            return False
+    return True
+
+# 6. Причастия и причастные обороты
+
+def check_no_participles(message):
+    morph = pymorphy2.MorphAnalyzer()
+    words = message.split()
+    morphs = [morph.parse(word)[0] for word in words]
+    
+    for morph in morphs:
+        if 'PRTF' in morph.tag:
+            return False
+    return True
+
+# 7. Деепричастия и деепричастные обороты
+
+def check_no_adverbial_participles(message):
+    morph = pymorphy2.MorphAnalyzer()
+    words = message.split()
+    morphs = [morph.parse(word)[0] for word in words]
+    
+    for morph in morphs:
+        if 'GRND' in morph.tag:
+            return False
+    return True
+
+# 8. Превосходная степень прилагательных
+
+def check_no_superlative_adjectives(message):
+    morph = pymorphy2.MorphAnalyzer()
+    words = message.split()
+    morphs = [morph.parse(word)[0] for word in words]
+    
+    for morph in morphs:
+        if 'COMP' in morph.tag or 'Supr' in morph.tag:
+            return False
+    return True
+
+# 9. Страдательный залог
+
+def check_no_passive_voice(message):
+    morph = pymorphy2.MorphAnalyzer()
+    words = message.split()
+    morphs = [morph.parse(word)[0] for word in words]
+    
+    for morph in morphs:
+        if 'PRTF' in morph.tag and ('passive' in morph.tag or 'в' in morph.tag):
+            return False
+    return True
+
+# 10. Порядковые числительные от 10 прописью
+
+def check_no_written_out_ordinals(message):
+    morph = pymorphy2.MorphAnalyzer()
+    ordinal_words = [
+        "десятый", "одиннадцатый", "двенадцатый", "тринадцатый", "четырнадцатый", "пятнадцатый",
+        "шестнадцатый", "семнадцатый", "восемнадцатый", "девятнадцатый", "двадцатый"
+    ]
+    
+    words = message.split()
+    lemmas = [morph.parse(word)[0].normal_form for word in words]
+    
+    for word in ordinal_words:
+        if word in lemmas:
+            return False
+    return True
+
+# 11. Цепочки с придаточными предложениями
+
+def check_no_subordinate_clauses_chain(message):
+    # Регулярное выражение, которое ищет последовательности придаточных предложений
+    subordinate_clause_patterns = [
+        r'\b(который|которая|которое|которые)\b',
+        r'\b(если|потому что|так как|что|когда)\b',
+        r'\b(хотя|несмотря на то что)\b'
+    ]
+    
+    count = 0
+    for pattern in subordinate_clause_patterns:
+        if re.search(pattern, message):
+            count += 1
+    
+    # Если в предложении найдено более одного придаточного предложения подряд, возвращаем False
+    return count < 2
+
+# 12. Разделительные повторяющиеся союзы
+
+def check_no_repeating_conjunctions(message):
+    # Регулярное выражение для поиска разделительных повторяющихся союзов с запятой перед вторым союзом
+    repeating_conjunctions_patterns = r'\b(и|ни|то|не то|или|либо)\b\s*(.*?)\s*,\s*\b\1\b'
+    
+    # Разделяем сообщение на предложения по точке, вопросительному и восклицательному знакам
+    sentences = re.split(r'[.!?]\s*', message)
+    
+    # Проверяем каждое предложение отдельно
+    for sentence in sentences:
+        if re.search(repeating_conjunctions_patterns, sentence, re.IGNORECASE):
+            return False
+    return True
+
+# 13. Вводные конструкции
+
+def check_no_introductory_phrases(message):
+    introductory_phrases = [
+        r'\b(во-первых|во-вторых|с одной стороны|по сути|по правде говоря)\b',
+        r'\b(может быть|кстати|конечно|естественно|безусловно|возможно)\b'
+    ]
+    
+    for pattern in introductory_phrases:
+        if re.search(pattern, message, re.IGNORECASE):
+            return False
+    return True
+
+# 14. Усилители
+
+def check_no_amplifiers(message):
+    amplifiers = [
+        r'\b(очень|крайне|чрезвычайно|совсем|абсолютно|полностью|чисто)\b'
+    ]
+    
+    for pattern in amplifiers:
+        if re.search(pattern, message, re.IGNORECASE):
+            return False
+    return True
+
+# 15. Паразиты времени
+
+def check_no_time_parasites(message):
+    time_parasites = [
+        r'\b(сейчас|немедленно|срочно|в данный момент|теперь)\b'
+    ]
+    
+    for pattern in time_parasites:
+        if re.search(pattern, message, re.IGNORECASE):
+            return False
+    return True
+
+# 16. Несколько существительных подряд
+
+def check_no_multiple_nouns(message):
+    noun_count = 0
+    words = re.split(r'\s+|[.!?]', message)  # Разбиваем по пробелам и знакам препинания
+    morph = pymorphy2.MorphAnalyzer()
+    
+    for word in words:
+        parsed_word = morph.parse(word)[0]
+        
+        # Если слово — существительное
+        if 'NOUN' in parsed_word.tag:
+            noun_count += 1
+        # Если встречен конец предложения (точка, вопросительный знак, восклицательный знак)
+        elif re.match(r'[.!?]', word):
+            noun_count = 0
+        else:
+            noun_count = 0
+        
+        if noun_count > 2:
+            return False
+    return True
+
+# 17. Производные предлоги
+
+def check_no_derived_prepositions(message):
+    derived_prepositions = [
+        r'\b(в течение|в ходе|вследствие|в связи с|по м��ре|при помощи|согласно|вопреки|на основании|на случай|в продолжение|по причине|вблизи|вдалеке|вокруг|внутри|вдоль|посередине|вне|снаружи|благодаря|невзирая на|исходя из)\b'
+    ]
+    
+    for pattern in derived_prepositions:
+        if re.search(pattern, message, re.IGNORECASE):
+            return False
+    return True
+
+# 19. Сложноподчиненные предложения
+
+def check_no_compound_sentences(message):
+    subordinating_conjunctions = [
+        r'\bкогда\b', r'\bкак только\b', r'\bпока\b', r'\bпосле того как\b',
+        r'\bпотому что\b', r'\bтак как\b', r'\bоттого что\b', r'\bблагодаря тому что\b',
+        r'\bчтобы\b', r'\bдля того чтобы\b', r'\bесли\b', r'\bкогда бы\b', r'\bесли бы\b',
+        r'\bхотя\b', r'\bнесмотря на то что\b', r'\bкак\b', r'\bбудто\b', r'\bсловно\b', r'\bкак будто\b',
+        r'\bчто\b'
+    ]
+    
+    # Убедимся, что слово "как" используется не в вопросе
+    for pattern in subordinating_conjunctions:
+        if re.search(pattern, message) and not re.search(r'\?', message):
+            return False
+    return True
+
+# 20. Даты прописью
+
+def check_no_dates_written_out(message):
+    # Ищем упоминания месяцев или слов, связанных с датами
+    months = [
+        "января", "февраля", "марта", "апреля", "мая", "июня", 
+        "июля", "августа", "сентября", "октября", "ноября", "декабря"
+    ]
+    
+    # Слова для проверки чисел прописью
+    date_written_out_patterns = [
+        r'\b(первого|второго|третьего|четвертого|пятого|шестого|седьмого|восьмого|девятого|десятого|одиннадцатого|двенадцатого|тринадцатого|четырнадцатого|пятнадцатого|шестнадцатого|семнадцатого|восемнадцатого|девятнадцатого|двадцатого|двадцать первого|двадцать второго|двадцать третьего|двадцать четвертого|двадцать пятого|двадцать шестого|двадцать седьмого|двадцать восьмого|двадцать девятого|тридцатого|тридцать первого)\b'
+    ]
+    
+    for month in months:
+        for pattern in date_written_out_patterns:
+            if re.search(f'{pattern}\\s{month}', message, re.IGNORECASE):
+                return False
+    
+    return True
+
+# ФУНКЦИИ ПРОВЕРОК (КОНЕЦ)
+
+def perform_checks(message):
+    checks = {
+        "forbidden_words": check_forbidden_words(message),
+        "client_addressing": check_no_greeting(message),
+        "promises": check_no_promises(message),
+        "double_verbs": check_no_double_verbs(message),
+        "participles": check_no_participles(message),
+        "adverbial_participles": check_no_adverbial_participles(message),
+        "superlative_adjectives": check_no_superlative_adjectives(message),
+        "passive_voice": check_no_passive_voice(message),
+        "written_out_ordinals": check_no_written_out_ordinals(message),
+        "subordinate_clauses_chain": check_no_subordinate_clauses_chain(message),
+        "repeating_conjunctions": check_no_repeating_conjunctions(message),
+        "introductory_phrases": check_no_introductory_phrases(message),
+        "amplifiers": check_no_amplifiers(message),
+        "time_parasites": check_no_time_parasites(message),
+        "multiple_nouns": check_no_multiple_nouns(message),
+        "derived_prepositions": check_no_derived_prepositions(message),
+        "compound_sentences": check_no_compound_sentences(message),
+        "dates_written_out": check_no_dates_written_out(message)
+    }
+    return checks
+
+
+def format_checks(checks):
+    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": "Даты прописью"
+    }
+    return "  \n".join([f"{translation[rule]}: {'✔️' if result else '❌'}" for rule, result in checks.items()])
+
+
+# Функция для обработки нажатия кнопки "Проверить"
+def perform_all_checks_and_show_results(personalized_message_1, personalized_message_2, personalized_message_3):
+    # Выполняем проверки для каждого сообщения
+    checks_1 = perform_checks(personalized_message_1)
+    checks_2 = perform_checks(personalized_message_2)
+    checks_3 = perform_checks(personalized_message_3)
+    
+    # Форматируем результаты для отображения
+    formatted_checks_1 = format_checks(checks_1)
+    formatted_checks_2 = format_checks(checks_2)
+    formatted_checks_3 = format_checks(checks_3)
+    
+    # Возвращаем результаты проверок для каждого сообщения
+    return (
+        personalized_message_1, formatted_checks_1,  # Первое персонализированное сообщение и его проверка
+        personalized_message_2, formatted_checks_2,  # Второе персонализированное сообщение и его проверка
+        personalized_message_3, formatted_checks_3   # Третье персонализированное сообщение и его проверка
+    )
+
 
 # Интерфейс Gradio
 with gr.Blocks() as demo:
@@ -398,6 +761,23 @@ with gr.Blocks() as demo:
                     success_forecast_3 = gr.Plot(label="Прогноз успешности сообщения 3")
                     gr.HTML("</div>")
                 
-            btn_check.click(fn=generate_random_gauges, inputs=[], outputs=[success_forecast_1, success_forecast_2, success_forecast_3])
+            # Модифицируем нажатие кнопки "Проверить"
+            btn_check.click(
+                fn=change_tab, 
+                inputs=[gr.Number(value=3, visible=False)],  # Переключение на вкладку "Проверка"
+                outputs=tabs  # Обновляем вкладку
+            ).then(
+                fn=perform_all_checks_and_show_results,
+                inputs=[personalized_1, personalized_2, personalized_3],  # Входные персонализированные сообщения
+                outputs=[
+                    personalized_message_1, check_message_1,  # Результаты проверок для первого сообщения
+                    personalized_message_2, check_message_2,  # Результаты проверок для второго сообщения
+                    personalized_message_3, check_message_3   # Результаты проверок для третьего сообщения
+                ]
+            ).then(
+                fn=generate_random_gauges, 
+                inputs=[],  # Нет входных данных для спидометров
+                outputs=[success_forecast_1, success_forecast_2, success_forecast_3]  # Вывод значений спидометров
+            )
 
 demo.launch()