Update app.py

app.py

@@ -8,7 +8,7 @@ from langchain_community.chat_models.gigachat import GigaChat
 from openpyxl import load_workbook
 import plotly.graph_objects as go
 import random
-import pymorphy2
+import pymorphy3
 import string
 import json
 from mistralai import Mistral
@@ -17,37 +17,37 @@ import requests
 import base64
 import io
 
-#MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
+MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
 token = os.getenv('GITHUB_TOKEN')
 
 # Клиент для генерации сообщений
-#client_mistral_generate = Mistral(api_key=MISTRAL_API_KEY)
+client_mistral_generate = Mistral(api_key=MISTRAL_API_KEY)
 
 # Клиент для выполнения проверок
-#client_mistral_check = Mistral(api_key=MISTRAL_API_KEY)
+client_mistral_check = Mistral(api_key=MISTRAL_API_KEY)
 
-morph = pymorphy2.MorphAnalyzer()
+morph = pymorphy3.MorphAnalyzer()
 
 # Авторизация в GigaChat Pro
-gc_key = os.getenv('GC_KEY')
+#gc_key = os.getenv('GC_KEY')
 #chat_pro = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=1.15, verify_ssl_certs=False)
-chat_pro = GigaChat(
-    credentials=gc_key,
-    model='GigaChat-Pro-preview',
-    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
-    max_tokens=68,
-    temperature=1.15,
-    verify_ssl_certs=False
-)
-
-chat_pro_check = GigaChat(
-    credentials=gc_key,
-    model='GigaChat-Pro-preview',
-    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
-    max_tokens=3000,
-    temperature=0.8,
-    verify_ssl_certs=False
-)
+#chat_pro = GigaChat(
+#    credentials=gc_key,
+#    model='GigaChat-Pro-preview',
+#    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
+#    max_tokens=68,
+#    temperature=1.15,
+#    verify_ssl_certs=False
+#)
+
+#chat_pro_check = GigaChat(
+#    credentials=gc_key,
+#    model='GigaChat-Pro-preview',
+#    base_url='https://gigachat-preview.devices.sberbank.ru/api/v1/',
+#    max_tokens=3000,
+#    temperature=0.8,
+#    verify_ssl_certs=False
+#)
 
 approach_stats = {
     "Начни сообщение с призыва к действию с продуктом.": {"failed_checks": defaultdict(int), "total_attempts": 0},
@@ -74,30 +74,6 @@ for sheet_name, df in data.items():
         print(f"Ошибка при обработке данных листа {sheet_name}: {e}")
         features[sheet_name] = {}
 
-# Функция для создания спидометра
-def create_gauge(value):
-    fig = go.Figure(go.Indicator(
-        mode="gauge+number",
-        value=value,
-        gauge={
-            'axis': {'range': [0, 100]},
-            'bar': {'color': "black"},  # Цвет стрелки
-            'steps': [
-                {'range': [0, 40], 'color': "#55efc4"},  # Мягкий зеленый
-                {'range': [40, 70], 'color': "#ffeaa7"},  # Желтый
-                {'range': [70, 100], 'color': "#ff7675"}  # Мягкий красный
-            ],
-            'threshold': {
-                'line': {'color': "black", 'width': 4},
-                'thickness': 0.75,
-                'value': value
-            }
-        },
-        number={'font': {'size': 48}}  # Размер шрифта числа
-    ))
-    fig.update_layout(paper_bgcolor="#f8f9fa", font={'color': "#2d3436", 'family': "Arial"}, width=250, height=150)
-    return fig
-
 def save_statistics_to_github(approach_stats):
     repo = "fruitpicker01/Storage_dev"
     timestamp = int(time.time())
@@ -170,10 +146,6 @@ def save_statistics_to_github(approach_stats):
     else:
         print(f"Ошибка при сохранении CSV-файла: {response.status_code} {response.text}")
 
-# Функция для генерации случайных значений спидометров
-def generate_random_gauges():
-    return create_gauge(random.randint(80, 95)), create_gauge(random.randint(80, 95)), create_gauge(random.randint(80, 95))
-
 # Функция для смены вкладки
 def change_tab(id):
     return gr.Tabs(selected=id)
@@ -191,77 +163,77 @@ def clean_message(message):
     return message
 
 # Функция для генерации сообщения с GigaChat Pro
-def generate_message_gigachat_pro(prompt):
-    try:
-        messages = [SystemMessage(content=prompt)]
-        res = chat_pro(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Pro: {e}"
-
-#def generate_message_mistral_generate(prompt, max_retries=5):
-#    retries = 0
-#    while retries < max_retries:
-#        try:
-#            chat_response = client_mistral_generate.chat.complete(
-#                model="mistral-large-latest",
-#                messages=[
-#                    {
-#                        "role": "user",
-#                        "content": prompt,
-#                        "max_tokens": 68,
-#                        "temperature": 0.8
-#                    },
-#                ]
-#            )
-#            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
-#            return cleaned_message
-#        except Exception as e:
-#                if "Status 429" in str(e):
-#                    wait_time = 3  # Можно установить фиксированную задержку
-#                    print(f"Превышен лимит запросов. Ожидание {wait_time} секунд перед повторной попыткой...")
-#                    time.sleep(wait_time)
-#                    retries += 1
-#                else:
-#                    print(f"Ошибка при обращении к Mistral: {e}")
-#                    return None
-
-#def generate_message_mistral_check(prompt, max_retries=5):
-#    retries = 0
-#    while retries < max_retries:
-#        try:
-#            chat_response = client_mistral_check.chat.complete(
-#                model="mistral-large-latest",
-#                messages=[
-#                    {
-#                        "role": "user",
-#                        "content": prompt,
-#                        "max_tokens": 3000,
-#                        "temperature": 0.2
-#                    },
-#                ]
-#            )
-#            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
-#            return cleaned_message
-#        except Exception as e:
-#                if "Status 429" in str(e):
-#                    wait_time = 3  # Можно установить фиксированную задержку
-#                    print(f"Превышен лимит запросов. Ожидание {wait_time} секунд перед повторной попыткой...")
-#                    time.sleep(wait_time)
-#                    retries += 1
-#                else:
-#                    print(f"Ошибка при обращении к Mistral: {e}")
-#                    return None
-
-def generate_check_gigachat_pro(prompt):
-    try:
-        messages = [SystemMessage(content=prompt)]
-        res2 = chat_pro_check(messages)
-        cleaned_message = clean_message(res2.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+#def generate_message_gigachat_pro(prompt):
+#    try:
+#        messages = [SystemMessage(content=prompt)]
+#        res = chat_pro(messages)
+#        cleaned_message = clean_message(res.content.strip())
+#        return cleaned_message
+#    except Exception as e:
+#        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+
+def generate_message_mistral_generate(prompt, max_retries=5):
+    retries = 0
+    while retries < max_retries:
+        try:
+            chat_response = client_mistral_generate.chat.complete(
+                model="mistral-large-latest",
+                messages=[
+                    {
+                        "role": "user",
+                        "content": prompt,
+                        "max_tokens": 68,
+                        "temperature": 0.8
+                    },
+                ]
+            )
+            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
+            return cleaned_message
+        except Exception as e:
+                if "Status 429" in str(e):
+                    wait_time = 3  # Можно установить фиксированную задержку
+                    print(f"Превышен лимит запросов. Ожидание {wait_time} секунд перед повторной попыткой...")
+                    time.sleep(wait_time)
+                    retries += 1
+                else:
+                    print(f"Ошибка при обращении к Mistral: {e}")
+                    return None
+
+def generate_message_mistral_check(prompt, max_retries=5):
+    retries = 0
+    while retries < max_retries:
+        try:
+            chat_response = client_mistral_check.chat.complete(
+                model="mistral-large-latest",
+                messages=[
+                    {
+                        "role": "user",
+                        "content": prompt,
+                        "max_tokens": 3000,
+                        "temperature": 0.2
+                    },
+                ]
+            )
+            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
+            return cleaned_message
+        except Exception as e:
+                if "Status 429" in str(e):
+                    wait_time = 3  # Можно установить фиксированную задержку
+                    print(f"Превышен лимит запросов. Ожидание {wait_time} секунд перед повторной попыткой...")
+                    time.sleep(wait_time)
+                    retries += 1
+                else:
+                    print(f"Ошибка при обращении к Mistral: {e}")
+                    return None
+
+#def generate_check_gigachat_pro(prompt):
+#    try:
+#        messages = [SystemMessage(content=prompt)]
+#        res2 = chat_pro_check(messages)
+#        cleaned_message = clean_message(res2.content.strip())
+#        return cleaned_message
+#    except Exception as e:
+#        return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
 # Функция для замены сокращений с 'k' или 'К' на тысячи
 def replace_k_with_thousands(message):
@@ -270,7 +242,7 @@ def replace_k_with_thousands(message):
     return message
 
 def correct_dash_usage(text):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     # Step 1: Replace any dash with long dash if surrounded by spaces
     text = re.sub(r'\s[-–—]\s', ' — ', text)
 
@@ -323,7 +295,7 @@ def correct_dash_usage(text):
     text = ' '.join(sentences)
 
     def restore_yo(text):
-        morph = pymorphy2.MorphAnalyzer()
+        morph = pymorphy3.MorphAnalyzer()
         words = text.split()
         restored_words = []
         
@@ -363,9 +335,13 @@ def correct_dash_usage(text):
     text = re.sub(r'\bповышьте\b', 'повысьте', text, flags=re.IGNORECASE)
 
     text = re.sub(r'\bСбербизнес\b', 'СберБизнес', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bСбербизнеса\b', 'СберБизнес', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bСбербизнесе\b', 'СберБизнес', text, flags=re.IGNORECASE)
     text = re.sub(r'\bСбербанк\b', 'СберБанк', text, flags=re.IGNORECASE)
     text = re.sub(r'\bвашего ООО\b', 'вашей компании', text, flags=re.IGNORECASE)
-
+    text = re.sub(r'\b0₽\b', '0 р', text, flags=re.IGNORECASE)
+    text = re.sub(r'\b₽\b', 'р', text, flags=re.IGNORECASE)
+    
     # Step 11: Replace all forms of "рублей", "рубля", "руб." with "р"
     # Используем два отдельных регулярных выражения для точности
     # 1. Заменяем "руб." на "р", учитывая, что "руб." может быть перед символом "/" или другим несловесным символом
@@ -377,7 +353,8 @@ def correct_dash_usage(text):
     text = re.sub(r'(\d+)\s+тысяч(?:а|и)?(?:\s+рублей)?', r'\1 000 р', text, flags=re.IGNORECASE)
     text = re.sub(r'(\d+)\s*тыс\.\s*руб\.', r'\1 000 р', text, flags=re.IGNORECASE)
     text = re.sub(r'(\d+)\s*тыс\.\s*р\.', r'\1 000 р', text, flags=re.IGNORECASE)
-
+    text = re.sub(r'(\d+)\s*тыс\.\s*р', r'\1 000 р', text, flags=re.IGNORECASE)
+    
     # Replace millions with "млн"
     text = re.sub(r'(\d+)\s+миллиона\b|\bмиллионов\b', r'\1 млн', text, flags=re.IGNORECASE)
     text = re.sub(r'(\d+)\s*млн\s*руб\.', r'\1 млн р', text, flags=re.IGNORECASE)
@@ -500,41 +477,12 @@ def notify_failed_checks(checks):
         gr.Warning("ВСЕ ПРОВЕРКИ ПРОЙДЕНЫ")
 
 # Модифицированная функция перегенерации сообщений с уведомлениями о номере попытки
-def generate_message_gigachat_pro_with_retry(prompt, current_prefix, description, key_message):
-    global approach_stats
-    last_message = None
-    for attempt in range(30):
-        gr.Info(f"Итерация {attempt + 1}: генерируется сообщение...")
-        message = generate_message_gigachat_pro(prompt)
-        message = replace_k_with_thousands(message)
-        message = correct_dash_usage(message)
-        message_length = len(message)
-        if not notify_failed_length(message_length):
-            last_message = message
-            time.sleep(1)
-            continue
-        checks = perform_checks(message, description, key_message)
-        last_message = message
-        approach_stats[current_prefix]["total_attempts"] += 1
-        for check_name, passed in checks.items():
-            if passed is False:
-                approach_stats[current_prefix]["failed_checks"][check_name] += 1
-                break
-        notify_failed_checks(checks)  # Вызываем функцию независимо от результата проверок
-        if all(checks.values()):
-            return message
-        prompt = append_errors_to_prompt(prompt, checks)
-        time.sleep(1)
-    gr.Info("Не удалось сгенерировать сообщение, соответствующее требованиям, за 20 итераций. Возвращаем последнее сгенерированное сообщение.")
-    return last_message
-
-
-#def generate_message_mistral_with_retry(prompt, current_prefix, description, key_message):
+#def generate_message_gigachat_pro_with_retry(prompt, current_prefix, description, key_message):
 #    global approach_stats
 #    last_message = None
-#    for attempt in range(20):
+#    for attempt in range(30):
 #        gr.Info(f"Итер��ция {attempt + 1}: генерируется сообщение...")
-#        message = generate_message_mistral_generate(prompt)
+#        message = generate_message_gigachat_pro(prompt)
 #        message = replace_k_with_thousands(message)
 #        message = correct_dash_usage(message)
 #        message_length = len(message)
@@ -558,6 +506,35 @@ def generate_message_gigachat_pro_with_retry(prompt, current_prefix, description
 #    return last_message
 
 
+def generate_message_mistral_with_retry(prompt, current_prefix, description, key_message):
+    global approach_stats
+    last_message = None
+    for attempt in range(20):
+        gr.Info(f"Итерация {attempt + 1}: генерируется сообщение...")
+        message = generate_message_mistral_generate(prompt)
+        message = replace_k_with_thousands(message)
+        message = correct_dash_usage(message)
+        message_length = len(message)
+        if not notify_failed_length(message_length):
+            last_message = message
+            time.sleep(1)
+            continue
+        checks = perform_checks(message, description, key_message)
+        last_message = message
+        approach_stats[current_prefix]["total_attempts"] += 1
+        for check_name, passed in checks.items():
+            if passed is False:
+                approach_stats[current_prefix]["failed_checks"][check_name] += 1
+                break
+        notify_failed_checks(checks)  # Вызываем функцию независимо от результата проверок
+        if all(checks.values()):
+            return message
+        prompt = append_errors_to_prompt(prompt, checks)
+        time.sleep(1)
+    gr.Info("Не удалось сгенерировать сообщение, соответствующее требованиям, за 20 итераций. Возвращаем последнее сгенерированное сообщение.")
+    return last_message
+
+
 # Функция для создания задания для копирайтера
 def generate_standard_prompt(description, advantages, key_message, *selected_values):
     prompt = (
@@ -695,7 +672,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
             yield standard_prompt, display_personalization_prompt, None, None, None, None, None, None
             flag += 1
         prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
-        non_personalized_message = generate_message_gigachat_pro_with_retry(prompt, current_prefix, desc, key_message)
+        non_personalized_message = generate_message_mistral_with_retry(prompt, current_prefix, desc, key_message)
         non_personalized_length = len(non_personalized_message)
         non_personalized_display = f"{non_personalized_message}\n------\nКоличество знаков: {non_personalized_length}"
         non_personalized_messages.append(non_personalized_display)
@@ -713,7 +690,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
 
         # Генерация персонализированного сообщения
         full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
-        personalized_message = generate_message_gigachat_pro_with_retry(full_personalized_prompt, current_prefix, desc, key_message)
+        personalized_message = generate_message_mistral_with_retry(full_personalized_prompt, current_prefix, desc, key_message)
         personalized_length = len(personalized_message)
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
         personalized_messages.append(personalized_display)
@@ -739,7 +716,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
 # 1. Запрещенные слова
 
 def check_forbidden_words(message):   
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     
     # Перечень запрещённых слов и фраз
     forbidden_patterns = [
@@ -773,7 +750,7 @@ def check_forbidden_words(message):
 # 2 и #3. Обращение к клиенту и приветствие клиента
 
 def check_no_greeting(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     # Список типичных обращений и приветствий
     greeting_patterns = [
         r"привет\b", r"здравствуй", r"добрый\s(день|вечер|утро)",
@@ -793,7 +770,7 @@ def check_no_greeting(message):
 # 4. Обещания и гарантии
 
 def check_no_promises(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     promise_patterns = [
         "обещать", "обещание", "гарантировать", "обязаться", "обязать", "обязательство", "обязательный"
     ]
@@ -809,7 +786,7 @@ def check_no_promises(message):
 # 5. Составные конструкции из двух глаголов
 
 def check_no_double_verbs(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     # Разделяем текст по пробелам и знакам препинания
     words = re.split(r'\s+|[.!?]', message)
     morphs = [morph.parse(word)[0] for word in words]
@@ -827,7 +804,7 @@ def check_no_double_verbs(message):
 # 6. Причастия и причастные обороты
 
 def check_no_participles(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     words = message.split()
     exceptions = {"повышенный", "увеличенный", "пониженный", "сниженный"}
     
@@ -841,7 +818,7 @@ def check_no_participles(message):
 # 7. Деепричастия и деепричастные обороты
 
 def check_no_adverbial_participles(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     words = message.split()
     morphs = [morph.parse(word)[0] for word in words]
     
@@ -853,7 +830,7 @@ def check_no_adverbial_participles(message):
 # 8. Превосходная степень прилагательных
 
 def check_no_superlative_adjectives(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     words = message.split()
     morphs = [morph.parse(word)[0] for word in words]
     
@@ -865,7 +842,7 @@ def check_no_superlative_adjectives(message):
 # 9. Страдательный залог
 
 def check_no_passive_voice(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     words = message.split()
     morphs = [morph.parse(word)[0] for word in words]
     
@@ -877,7 +854,7 @@ def check_no_passive_voice(message):
 # 10. Порядковые числительные от 10 прописью
 
 def check_no_written_out_ordinals(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     ordinal_words = [
         "десятый", "одиннадцатый", "двенадцатый", "тринадцатый", "четырнадцатый", "пятнадцатый",
         "шестнадцатый", "семнадцатый", "восемнадцатый", "девятнадцатый", "двадцатый"
@@ -966,7 +943,7 @@ def check_no_time_parasites(message):
 def check_no_multiple_nouns(message):
     noun_count = 0
     words = re.split(r'\s+|[.!?]', message)  # Разбиваем по пробелам и знакам препинания
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     
     for word in words:
         parsed_word = morph.parse(word)[0]
@@ -1036,7 +1013,7 @@ def check_no_dates_written_out(message):
 # Доп правило. Повторы слов
 
 def check_no_word_repetitions(message):
-    morph = pymorphy2.MorphAnalyzer()
+    morph = pymorphy3.MorphAnalyzer()
     
     # Список союзов и предлогов, которые мы будем игнорировать
     ignore_words = set([
@@ -1135,9 +1112,9 @@ def check_disconnected_sentences(message):
 если таких предложений **нет**, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь **только** в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)  # Задержка в 3 секунды между запросами
-    print("GigaChat Pro response:", response)  # Выводим полный ответ модели
+    print("Mistral response:", response)  # Выводим полный ответ модели
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1162,9 +1139,9 @@ def check_synonymous_members(message):
 если таких слов или выражений нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1195,9 +1172,9 @@ def check_clickbait_phrases(message):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1241,9 +1218,9 @@ def check_abstract_claims(message):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1278,9 +1255,9 @@ def check_specialized_terms(message):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1307,9 +1284,9 @@ def check_offensive_phrases(message):
 если таких фраз нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1350,9 +1327,9 @@ def check_cliches_and_bureaucratese(message):
     если в тексте **есть** такие выражения, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}}.
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1377,9 +1354,9 @@ def check_no_contradictions(message, description):
 Если сообщение содержит факты, которые отсутствуют в описании предложения, **верни только** JSON {{"decision": true, "explanation": "<описание противоречий>"}}.
 **Не добавляй никакого дополнительного текста. Отвечай только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1404,9 +1381,9 @@ def check_contains_key_message(message, key_message):
 Если сообщение **не содержит всю** информацию из ключевого текста, **верни только** JSON {{"decision": true, "explanation": "Ключевое текст отсутствует."}}.
 **Не добавляй никакого дополнительного текста. Отвечай только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = generate_check_gigachat_pro(prompt)
+    response = generate_message_mistral_check(prompt)
     time.sleep(3)
-    print("GigaChat Pro response:", response)
+    print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
         decision = result.get("decision", False)
@@ -1429,7 +1406,7 @@ def safe_check(func, *args):
 def perform_checks(message, description, key_message):
     checks = {}
 
-    # 2. Morphological checks using pymorphy2
+    # 2. Morphological checks using pymorphy3
     morphological_checks = [
         ("forbidden_words", check_forbidden_words),
         ("client_addressing", check_no_greeting),
@@ -1542,74 +1519,6 @@ def format_checks(checks):
     return "  \n".join(formatted_results)
 
 
-# Функция для обработки нажатия кнопки "Проверить"
-def perform_all_checks_and_show_results(personalized_message_1, personalized_message_2, personalized_message_3):
-    # Моментально показываем все персонализированные сообщения
-    yield (
-        personalized_message_1, None,  # Первое сообщение без проверки
-        personalized_message_2, None,  # Второе сообщение без проверки
-        personalized_message_3, None,  # Третье сообщение без проверки
-        None, None, None  # Пустые графики для спидометров
-    )
-    
-    # Выполняем и показываем проверки с задержкой 1 секунда
-    checks_1 = perform_checks(personalized_message_1)
-    formatted_checks_1 = format_checks(checks_1)
-    time.sleep(1)  # Задержка 1 секунда перед выводом первого результата проверки
-    yield (
-        personalized_message_1, formatted_checks_1,  # Проверка для первого сообщения
-        personalized_message_2, None,  # Второе сообщение без проверки
-        personalized_message_3, None,  # Третье сообщение без проверки
-        None, None, None  # Пустые графики для спидометров
-    )
-    
-    checks_2 = perform_checks(personalized_message_2)
-    formatted_checks_2 = format_checks(checks_2)
-    time.sleep(1)  # Задержка 1 секунда перед выводом второго результата проверки
-    yield (
-        personalized_message_1, formatted_checks_1,  # Проверка для первого сообщения
-        personalized_message_2, formatted_checks_2,  # Проверка для второго сообщения
-        personalized_message_3, None,  # Третье сообщение без проверки
-        None, None, None  # Пустые графики для спидометров
-    )
-    
-    checks_3 = perform_checks(personalized_message_3)
-    formatted_checks_3 = format_checks(checks_3)
-    time.sleep(1)  # Задержка 1 секунда перед выводом третьего результата проверки
-    yield (
-        personalized_message_1, formatted_checks_1,  # Проверка для первого сообщения
-        personalized_message_2, formatted_checks_2,  # Проверка для второго сообщения
-        personalized_message_3, formatted_checks_3,  # Проверка для третьего сообщения
-        None, None, None  # Пустые графики для спидометров
-    )
-
-    # Генерация и показ графиков спидометров с задержкой 2 секунды
-    time.sleep(2)
-    gauges = generate_random_gauges()
-    yield (
-        personalized_message_1, formatted_checks_1,  # Проверка для первого сообщения
-        personalized_message_2, formatted_checks_2,  # Проверка для второго сообщения
-        personalized_message_3, formatted_checks_3,  # Проверка для третьего сообщения
-        gauges[0], None, None  # Первый график спидометра
-    )
-    
-    time.sleep(2)
-    yield (
-        personalized_message_1, formatted_checks_1,  # Проверка для первого сообщения
-        personalized_message_2, formatted_checks_2,  # Проверка для второго сообщения
-        personalized_message_3, formatted_checks_3,  # Проверка для третьего сообщения
-        gauges[0], gauges[1], None  # Первый и второй графики спидометра
-    )
-    
-    time.sleep(2)
-    yield (
-        personalized_message_1, formatted_checks_1,  # Проверка для первого сообщения
-        personalized_message_2, formatted_checks_2,  # Проверка для второго сообщения
-        personalized_message_3, formatted_checks_3,  # Проверка для третьего сообщения
-        gauges[0], gauges[1], gauges[2]  # Все три графика спидометра
-    )
-
-
 # Интерфейс Gradio
 with gr.Blocks() as demo:
     # Твой интерфейс
@@ -1681,11 +1590,6 @@ with gr.Blocks() as demo:
                 non_personalized_messages = gr.Textbox(label="Стандартные сообщения", lines=12, interactive=False)
                 personalized_messages = gr.Textbox(label="Персонализированные сообщения", lines=12, interactive=False)
 
-            # Четвертый ряд
-            with gr.Row():
-                btn_check = gr.Button("Проверить", elem_id="check3")
-                btn_check.click(fn=change_tab, inputs=[gr.Number(value=3, visible=False)], outputs=tabs)
-
             # Сначала переключаем вкладку, потом запускаем генерацию сообщений
             btn_to_prompts.click(
                 fn=change_tab,