Update app.py

app.py

@@ -18,9 +18,6 @@ import base64
 import io
 from transformers import AutoTokenizer, AutoModel
 from utils import best_text_choice
-import asyncio
-import inspect
-from together import AsyncTogether
 
 tokenizer = AutoTokenizer.from_pretrained("ai-forever/ru-en-RoSBERTa")
 model = AutoModel.from_pretrained("ai-forever/ru-en-RoSBERTa")
@@ -28,22 +25,13 @@ model = AutoModel.from_pretrained("ai-forever/ru-en-RoSBERTa")
 unique_sms_df = pd.read_parquet('unique_texts.parquet')
 
 MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
-MISTRAL_API_KEY_2 = os.getenv('MISTRAL_API_KEY_2')
-MISTRAL_API_KEY_3 = os.getenv('MISTRAL_API_KEY_3')
-MISTRAL_API_KEY_4 = os.getenv('MISTRAL_API_KEY_4')
-MISTRAL_API_KEY_5 = os.getenv('MISTRAL_API_KEY_5')
 token = os.getenv('GITHUB_TOKEN')
 
-async_client = AsyncTogether(api_key=os.environ.get("TOGETHER_API_KEY"))
-
 # Клиент для генерации сообщений
 client_mistral_generate = Mistral(api_key=MISTRAL_API_KEY)
 
 # Клиент для выполнения проверок
-client_mistral_check = Mistral(api_key=MISTRAL_API_KEY_2)
-client_mistral_check_2 = Mistral(api_key=MISTRAL_API_KEY_3)
-client_mistral_check_3 = Mistral(api_key=MISTRAL_API_KEY_4)
-client_mistral_check_4 = Mistral(api_key=MISTRAL_API_KEY_5)
+client_mistral_check = Mistral(api_key=MISTRAL_API_KEY)
 
 morph = pymorphy3.MorphAnalyzer()
 
@@ -200,8 +188,7 @@ def clean_message(message):
 #    except Exception as e:
 #        return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-async def generate_message_mistral_generate(prompt, max_retries=5):
-    
+def generate_message_mistral_generate(prompt, max_retries=5):
 #def generate_message_mistral_generate(prompt):
 #    try:
 #        messages = [SystemMessage(content=prompt)]
@@ -214,173 +201,34 @@ async def generate_message_mistral_generate(prompt, max_retries=5):
     retries = 0
     while retries < max_retries:
         try:
-            # Отправляем запрос к модели Qwen через Together API
-            response = await async_client.chat.completions.create(
-                model="Meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.9,      # Параметры можно настроить по необходимости
-                max_tokens=74
-            )
-
-            # Извлекаем и очищаем ответ
-            cleaned_message = clean_message(response.choices[0].message.content.strip())
-            return cleaned_message
-
-        except Exception as e:
-            error_message = str(e)
-            # Проверяем, связана ли ошибка с превышением лимита или разрывом соединения
-            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-                wait_time = 3  # Время ожидания перед повторной попыткой
-                print(f"Лимит запросов превышен или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
-                await asyncio.sleep(wait_time)
-                retries += 1
-            else:
-                # Для других типов ошибок выводим сообщение и прекращаем попытки
-                print(f"Ошибка при вызове Together API: {e}")
-                return None
-
-    # Если все попытки исчерпаны, сообщаем об этом
-    print("Не удалось получить ответ от Together API после максимального количества попыток.")
-    return None
-    
-#    retries = 0
-#    while retries < max_retries:
-#        try:
-#            chat_response = await client_mistral_generate.chat.complete_async(
-#                model="open-mistral-nemo",
-#                temperature=1.0,
-#                min_tokens=71,
-#                max_tokens=96,
-#                messages=[
-#                    {
-#                        "role": "user",
-#                        "content": prompt
-#                    },
-#                ]
-#            )
-#            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
-#            return cleaned_message
-#        except Exception as e:
-#            error_message = str(e)
-#            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-#                wait_time = 3
-#                print(f"Rate limit exceeded or server did not respond. Waiting {wait_time} seconds before retrying...")
-#                await asyncio.sleep(wait_time)
-#                retries += 1
-#            else:
-#                print(f"Error calling Mistral: {e}")
-#                return None
-#    print("Failed to get response from Mistral after maximum retries.")
-#    return None
-
-async def generate_message_mistral_check(prompt, max_retries=5):
-#def generate_message_mistral_check(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}"
-    retries = 0
-    while retries < max_retries:
-        try:
-            # Отправляем запрос к модели Qwen через Together API
-            response = await async_client.chat.completions.create(
-                model="Meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
-                messages=[{"role": "user", "content": prompt}],
+            chat_response = client_mistral_generate.chat.complete(
+                model="mistral-large-latest",
+                temperature=1.0,
+                min_tokens=81,
+                max_tokens=108,
+                messages=[
+                    {
+                        "role": "user",
+                        "content": prompt
+                    },
+                ]
             )
-
-            # Извлекаем и очищаем ответ
-            cleaned_message = clean_message(response.choices[0].message.content.strip())
+            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
             return cleaned_message
-
         except Exception as e:
             error_message = str(e)
-            # Проверяем, связана ли ошибка с превышением лимита или разрывом соединения
             if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-                wait_time = 3  # Время ожидания перед повторной попыткой
-                print(f"Лимит запросов превышен или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
-                await asyncio.sleep(wait_time)
+                wait_time = 3
+                print(f"Превышен лимит запросов или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
+                time.sleep(wait_time)
                 retries += 1
             else:
-                # Для других типов ошибок выводим сообщение и прекращаем попытки
-                print(f"Ошибка при вызове Together API: {e}")
+                print(f"Ошибка при обращении к Mistral: {e}")
                 return None
-
-    # Если все попытки исчерпаны, сообщаем об этом
-    print("Не удалось получить ответ от Together API после максимального количества попыток.")
+    print("Не удалось получить ответ от Mistral после максимального числа попыток.")
     return None
 
-async def generate_message_mistral_check_2(prompt, max_retries=5):
-#def generate_message_mistral_check(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}"
-
-    retries = 0
-    while retries < max_retries:
-        try:
-            # Отправляем запрос к модели Qwen через Together API
-            response = await async_client.chat.completions.create(
-                model="Meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
-                messages=[{"role": "user", "content": prompt}],
-            )
-
-            # Извлекаем и очищаем ответ
-            cleaned_message = clean_message(response.choices[0].message.content.strip())
-            return cleaned_message
-
-        except Exception as e:
-            error_message = str(e)
-            # Проверяем, связана ли ошибка с превышением лимита или разрывом соединения
-            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-                wait_time = 3  # Время ожидания перед повторной попыткой
-                print(f"Лимит запросов превышен или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
-                await asyncio.sleep(wait_time)
-                retries += 1
-            else:
-                # Для других типов ошибок выводим сообщение и прекращаем попытки
-                print(f"Ошибка при вызове Together API: {e}")
-                return None
-
-    # Если все попытки исчерпаны, сообщаем об этом
-    print("Не удалось получить ответ от Together API после максимального количества попыток.")
-    return None
-    
-#    retries = 0
-#    while retries < max_retries:
-#        try:
-#            chat_response = await client_mistral_check_2.chat.complete_async(
-#                model="open-mistral-nemo",
-#                temperature=0.2,
-#                messages=[
-#                    {
-#                        "role": "user",
-#                        "content": prompt
-#                    },
-#                ]
-#            )
-#            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
-#            return cleaned_message
-#        except Exception as e:
-#            error_message = str(e)
-#            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-#                wait_time = 3
-#                print(f"Rate limit exceeded or server did not respond. Waiting {wait_time} seconds before retrying...")
-#                await asyncio.sleep(wait_time)
-#                retries += 1
-#            else:
-#                print(f"Error calling Mistral: {e}")
-#                return None
-#    print("Failed to get response from Mistral after maximum retries.")
-#    return None
-
-async def generate_message_mistral_check_3(prompt, max_retries=5):
+def generate_message_mistral_check(prompt, max_retries=5):
 #def generate_message_mistral_check(prompt):    
 #    try:
 #        messages = [SystemMessage(content=prompt)]
@@ -389,39 +237,32 @@ async def generate_message_mistral_check_3(prompt, max_retries=5):
 #        return cleaned_message
 #    except Exception as e:
 #        return f"Ошибка при обращении к GigaChat-Pro: {e}"
-    
     retries = 0
     while retries < max_retries:
         try:
-            # Отправляем запрос к модели Qwen через Together API
-            response = await async_client.chat.completions.create(
-                model="Meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
-                messages=[{"role": "user", "content": prompt}],
+            chat_response = client_mistral_check.chat.complete(
+                model="mistral-large-latest",
+                temperature=0.2,
+                messages=[
+                    {
+                        "role": "user",
+                        "content": prompt
+                    },
+                ]
             )
-
-            # Извлекаем и очищаем ответ
-            cleaned_message = clean_message(response.choices[0].message.content.strip())
+            cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
             return cleaned_message
-
         except Exception as e:
-            error_message = str(e)
-            # Проверяем, связана ли ошибка с превышением лимита или разрывом соединения
-            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-                wait_time = 3  # Время ожидания перед повторной попыткой
-                print(f"Лимит запросов превышен или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
-                await asyncio.sleep(wait_time)
-                retries += 1
-            else:
-                # Для других типов ошибок выводим сообщение и прекращаем попытки
-                print(f"Ошибка при вызове Together API: {e}")
-                return None
-
-    # Если все попытки исчерпаны, сообщаем об этом
-    print("Не удалось получить ответ от Together API после максимального количества попыток.")
-    return None
-
-async def generate_message_mistral_check_4(prompt, max_retries=5):
-#def generate_message_mistral_check(prompt):    
+                if "Status 429" in str(e) or "Server disconnected without sending a response" 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)
@@ -429,37 +270,6 @@ async def generate_message_mistral_check_4(prompt, max_retries=5):
 #        return cleaned_message
 #    except Exception as e:
 #        return f"Ошибка при обращении к GigaChat-Pro: {e}"
-    
-    retries = 0
-    while retries < max_retries:
-        try:
-            # Отправляем запрос к модели Qwen через Together API
-            response = await async_client.chat.completions.create(
-                model="Meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
-                messages=[{"role": "user", "content": prompt}],
-            )
-
-            # Извлекаем и очищаем ответ
-            cleaned_message = clean_message(response.choices[0].message.content.strip())
-            return cleaned_message
-
-        except Exception as e:
-            error_message = str(e)
-            # Проверяем, связана ли ошибка с превышением лимита или разрывом соединения
-            if "Status 429" in error_message or "Server disconnected without sending a response" in error_message:
-                wait_time = 3  # Время ожидания перед повторной попыткой
-                print(f"Лимит запросов превышен или сервер не ответил. Ожидание {wait_time} секунд перед повторной попыткой...")
-                await asyncio.sleep(wait_time)
-                retries += 1
-            else:
-                # Для других типов ошибок выводим сообщение и прекращаем попытки
-                print(f"Ошибка при вызове Together API: {e}")
-                return None
-
-    # Если все попытки исчерпаны, сообщаем об этом
-    print("Не удалось получить ответ от Together API после максимального количества попыток.")
-    return None
-
 
 # Функция для замены сокращений с 'k' или 'К' на тысячи
 def replace_k_with_thousands(message):
@@ -628,7 +438,7 @@ def append_errors_to_prompt(prompt, checks):
         "derived_prepositions": "Не использовать производные предлоги.",
         "compound_sentences": "Избегать сложноподчиненных предложений.",
         "dates_written_out": "Не писать даты прописью.",
-#        "no_word_repetitions": "НЕ ИСПОЛЬЗУЙ ПОВТОРЯЮЩИЕСЯ СЛОВА В СООБЩЕНИИ.",
+        "no_word_repetitions": "Избегать повторов слов.",
         "disconnected_sentences": "Избегать сложных предложений без логической связи.",
         "synonymous_members": "Не использовать близкие по смыслу однородные члены предложения.",
         "clickbait_phrases": "Не использовать кликбейтные фразы.",
@@ -653,8 +463,8 @@ def append_errors_to_prompt(prompt, checks):
 
     
 def notify_failed_length(message_length):
-    if message_length < 160:
-        gr.Warning(f"Сообщение слишком короткое: {message_length} знаков. Минимум 160.")
+    if message_length < 190:
+        gr.Warning(f"Сообщение слишком короткое: {message_length} знаков. Минимум 190.")
         return False
     elif message_length > 250:
         gr.Warning(f"Сообщение слишком длинное: {message_length} знаков. Максимум 250.")
@@ -667,7 +477,7 @@ def notify_failed_checks(checks):
         "forbidden_words": "Запрещенные слова",
         "client_addressing": "Обращение к клиенту",
         "promises": "Обещания и гарантии",
-#        "double_verbs": "Два глагола подряд",
+        "double_verbs": "Два глагола подряд",
         "participles": "Причастия",
         "adverbial_participles": "Деепричастия",
         "superlative_adjectives": "Превосходная степень",
@@ -682,7 +492,7 @@ def notify_failed_checks(checks):
         "derived_prepositions": "Производные предлоги",
         "compound_sentences": "Сложноподчиненные предложения",
         "dates_written_out": "Даты прописью",
-#        "no_word_repetitions": "Повторы слов",
+        "no_word_repetitions": "Повторы слов",
         "disconnected_sentences": "Сложные предложения без логической связи",
         "synonymous_members": "Близкие по смыслу однородные члены предложения",
         "clickbait_phrases": "Кликбейтные фразы",
@@ -735,24 +545,24 @@ def notify_failed_checks(checks):
 #    return last_message
 
 
-async def generate_message_mistral_with_retry(prompt, approach_name, description, key_message, product_name, benefits):
+def generate_message_mistral_with_retry(prompt, approach_name, description, key_message, product_name, benefits):
     global approach_stats
     last_message = None
     for attempt in range(20):
-        gr.Info(f"Итерация {attempt + 1}: генерация сообщения...")
-        message = await generate_message_mistral_generate(prompt)
+        gr.Info(f"Итерация {attempt + 1}: генерируется сообщение...")
+        message = generate_message_mistral_generate(prompt)
         if message is None:
-            print("Failed to get message from Mistral, retrying...")
-#            await asyncio.sleep(1)
+            print("Не удалось получить сообщение от Mistral, повторная попытка...")
+            time.sleep(1)
             continue
         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
-#            await asyncio.sleep(1)
+            time.sleep(1)
             continue
-        checks = await perform_checks(message, description, key_message, product_name, benefits)
+        checks = perform_checks(message, description, key_message, product_name, benefits)
         last_message = message
         # Инициализируем статистику для подхода, если ее нет
         if approach_name not in approach_stats:
@@ -766,7 +576,7 @@ async def generate_message_mistral_with_retry(prompt, approach_name, description
         if all(checks.values()):
             return message
         prompt = append_errors_to_prompt(prompt, checks)
-#        await asyncio.sleep(1)
+        time.sleep(1)
     gr.Info("Не удалось сгенерировать сообщение, соответствующее требованиям, за 20 итераций. Возвращаем последнее сгенерированное сообщение.")
     return last_message
 
@@ -779,7 +589,7 @@ def generate_standard_prompt(description, benefits, key_message, *selected_value
         f"Преимущества: {benefits}\n"
         "В тексте смс запрещено использование:\n"
         "- Запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, доступный, банкротство, долги, займ, срочно, сейчас, лучший, главный, номер 1, гарантия, успех, лидер, никакой;\n"
-        "- ОДИНАКОВЫЕ СЛОВА В СООБЩЕНИИ;\n"
+        "- Повторы слов;\n"
         "- Обращение к клиенту;\n"
         "- Приветствие клиента;\n"
         "- Обещания и гарантии;\n"
@@ -806,7 +616,7 @@ def generate_standard_prompt(description, benefits, key_message, *selected_value
         "- Узкоспециализированные термины;\n"
         "- Фразы, способные создать двойственное ощущение, обидеть;\n"
         "- Речевые клише, рекламные штампы, канцеляризмы;\n"
-        "Убедись, что в готовом тексте до 250, но не менее 160 знаков с пробелами. Убедись, что в готовом тексте не менее трех предложений.\n"
+        "Убедись, что в готовом тексте до 250, но не менее 190 знаков с пробелами. Убедись, что в готовом тексте не менее трех предложений.\n"
     )        
     if key_message.strip():
         prompt += f"Убедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
@@ -818,7 +628,7 @@ def generate_standard_prompt(description, benefits, key_message, *selected_value
 def generate_personalization_prompt(key_message, *selected_values, prefix, suffix, product_name):
     prompt = f"{prefix}\n"
     prompt += f"Не изменяй название продукта: {product_name}.\n"
-    prompt += "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами (но и не менее 160 знаков с пробелами), текст с учетом следующих особенностей:\n"
+    prompt += "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами (но и не менее 190 знаков с пробелами), текст с учетом следующих особенностей:\n"
     gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
     combined_instruction = ""
     additional_instructions = ""
@@ -882,7 +692,7 @@ def clean_prompt_for_display(prompt, prefixes, suffixes):
     return cleaned_prompt.strip()
 
 # Функция для постепенной генерации всех сообщений через yield
-async def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf, product_name):
+def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf, product_name):
     standard_prompt = generate_standard_prompt(desc, benefits, key_message)
     standard_prompt_for_display = f"Не изменяй название продукта: {product_name}.\n{standard_prompt}\nУбедись, что в готовом тексте без изменений, синонимов и перестановок слов используется наименование продукта: {product_name}.\n"
     approach_mapping = {
@@ -954,7 +764,7 @@ async def generate_all_messages(desc, benefits, key_message, gender, generation,
             yield selected_approaches_text_content, standard_prompt_for_display, display_personalization_prompt, None, None
             flag += 1
         prompt = add_prefix_suffix(standard_prompt, current_prefix, current_suffix, product_name)
-        non_personalized_message = await generate_message_mistral_with_retry(prompt, approach_name, desc, key_message, product_name, benefits)
+        non_personalized_message = generate_message_mistral_with_retry(prompt, approach_name, desc, key_message, product_name, benefits)
         non_personalized_length = len(non_personalized_message)
         non_personalized_display = f"{non_personalized_message}\n------\nКоличество знаков: {non_personalized_length}"
         if non_personalized_messages:
@@ -966,7 +776,7 @@ async def generate_all_messages(desc, benefits, key_message, gender, generation,
             non_personalized_messages, personalized_messages
         )
         full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
-        personalized_message = await generate_message_mistral_with_retry(full_personalized_prompt, approach_name, desc, key_message, product_name, benefits)
+        personalized_message = generate_message_mistral_with_retry(full_personalized_prompt, approach_name, desc, key_message, product_name, benefits)
         personalized_length = len(personalized_message)
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
         if personalized_messages:
@@ -977,7 +787,7 @@ async def generate_all_messages(desc, benefits, key_message, gender, generation,
             selected_approaches_text_content, standard_prompt_for_display, display_personalization_prompt,
             non_personalized_messages, personalized_messages
         )
-#        time.sleep(1)
+        time.sleep(1)
     save_statistics_to_github(approach_stats)
 
 def rank_messages(non_personalized_messages, personalized_messages):
@@ -1371,7 +1181,7 @@ def check_no_word_repetitions(message):
         'и', 'а', 'но', 'или', 'да', 'ни', 'как', 'так',
         'в', 'на', 'под', 'над', 'за', 'к', 'до', 'по', 'из', 'у', 'о', 'про', 'для',
         'не', 'вот', 'это', 'тот', 'тем', 'при', 'чем',
-        'же', 'ли', 'бы', 'то', 'р', 'от',
+        'же', 'ли', 'бы', 'то', 'р',
     ])
     
     # Разбиваем текст на слова, удаляя знаки препинания
@@ -1451,7 +1261,7 @@ def cut_message(message):
     return message
         
 # 22. Проверка сложных предложений без логической связи
-async def check_disconnected_sentences(message):
+def check_disconnected_sentences(message):
     message_clean = cut_message(message)
     print()
     print("Проверка 22: Проверка сложных предложений без логической связи")
@@ -1466,8 +1276,8 @@ async def check_disconnected_sentences(message):
 если таких предложений **нет**, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь **только** в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
-#    await asyncio.sleep(3)  # Задержка в 3 секунды между запросами
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)  # Задержка в 3 секунды между запросами
     print("Mistral response:", response)  # Выводим полный ответ модели
     result = parse_json_response(response)
     if result is not None:
@@ -1479,7 +1289,7 @@ async def check_disconnected_sentences(message):
         return None
 
 # 23. Проверка на близкие по смыслу однородные члены
-async def check_synonymous_members(message):
+def check_synonymous_members(message):
     print()
     print("Проверка 23: Проверка на близкие по смыслу однородные члены")
     print()
@@ -1493,8 +1303,8 @@ async def check_synonymous_members(message):
 если таких слов или выражений нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check_2(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1507,7 +1317,7 @@ async def check_synonymous_members(message):
 
 
 # 24. Проверка на шокирующие, экстравагантные или кликбейтные фразы
-async def check_clickbait_phrases(message, description, benefits, key_message):
+def check_clickbait_phrases(message, description, benefits):
     message_clean = cut_message(message)
     print()
     print()
@@ -1523,7 +1333,6 @@ async def check_clickbait_phrases(message, description, benefits, key_message):
     3. Стандартные рекламные призывы к действию, такие как "купите сейчас" или "узнайте больше", не считаются кликбейтом, если они не преувеличивают преимущества или не используют явную манипуляцию эмоциями.
     4. Не считай фразы, используемые в исходном описании продукта, кликбейтными. Исходное описание: "{description}".
     5. Не считай фразы, используемые в преимуществах продукта, кликбейтными. Преимущества: "{benefits}".
-    6. Не считай фразы, используемые в ключевом сообщении, кликбейтными. Ключевое сообщение: "{key_message}".
     Пример ответа:
     {{"decision": false, "explanation": "Текст нейтрален и не содержит кликбейтных фраз."}}
     
@@ -1532,8 +1341,8 @@ async def check_clickbait_phrases(message, description, benefits, key_message):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check_4(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1546,7 +1355,7 @@ async def check_clickbait_phrases(message, description, benefits, key_message):
 
 
 # 25. Проверка на абстрактные заявления без поддержки фактами
-async def check_abstract_claims(message, description, benefits, key_message):
+def check_abstract_claims(message, description, benefits):
     print()
     print("Проверка 25: Проверка на абстрактные заявления без поддержки фактами")
     print()
@@ -1573,9 +1382,8 @@ async def check_abstract_claims(message, description, benefits, key_message):
        - "Снизьте финансовую нагрузку"
     4. Ищи общие фразы, которые не дают представления о конкретной пользе, такие как "лучшее решение", "высокое качество", "отличный сервис", если они не сопровождаются пояснением о том, почему это так.
     5. Учитывай, что в рекламных сообщениях допустимы эмоциональные и обобщённые фразы, если они достаточно конкретны для понимания аудитории, однако они должны сопровождаться фактами или подробными примерами.
-    6. Не считай фразы, используемые в исходном описании продукта, абстрактными. Исходное описание: "{description}".
-    7. Не с��итай фразы, используемые в преимуществах продукта, абстрактными. Преимущества: "{benefits}".
-    8. Не считай фразы, используемые в ключевом сообщении, абстрактными. Ключевое сообщение: "{key_message}".
+    6. Не считай фразы, используемые в исходном описании продукта, кликбейтными. Исходное описание: "{description}".
+    7. Не считай фразы, используемые в преимуществах продукта, кликбейтными. Преимущества: "{benefits}".
     
     Пример ответа:
     {{"decision": false, "explanation": "Текст не содержит абстрактные утверждения без конкретики."}}
@@ -1585,8 +1393,8 @@ async def check_abstract_claims(message, description, benefits, key_message):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check_2(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1599,7 +1407,7 @@ async def check_abstract_claims(message, description, benefits, key_message):
 
 
 # 26. Проверка на узкоспециализированные термины
-async def check_specialized_terms(message):
+def check_specialized_terms(message):
     print()
     print("Проверка 26: Проверка на узкоспециализированные термины")
     print()
@@ -1622,8 +1430,8 @@ async def check_specialized_terms(message):
     
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check_2(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1635,7 +1443,7 @@ async def check_specialized_terms(message):
         return None
 
 # 27. Проверка на двусмысленные или обидные фразы
-async def check_offensive_phrases(message):
+def check_offensive_phrases(message):
     print()
     print("Проверка 27: Проверка на двусмысленные или обидные фразы")
     print()
@@ -1651,8 +1459,8 @@ async def check_offensive_phrases(message):
 если таких фраз нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
 **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check_4(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1664,7 +1472,7 @@ async def check_offensive_phrases(message):
         return None
 
 # 28. Проверка на речевые клише, рекламные штампы и канцеляризмы
-async def check_cliches_and_bureaucratese(message, description, benefits, key_message):
+def check_cliches_and_bureaucratese(message, description, benefits):
     print()
     print("Проверка 28: Проверка на речевые клише, рекламные штампы и канцеляризмы")
     print()
@@ -1677,9 +1485,8 @@ async def check_cliches_and_bureaucratese(message, description, benefits, key_me
     - Информацию о ценах, скидках, акциях или условиях покупки (например, "при покупках от 100 000 рублей в месяц").
     - Описания способов оформления или получения услуг (например, "оформление возможно онлайн или в офисе").
     - Стандартные отраслевые термины и фразы, необходимые для понимания сообщения (например, "премиальная бизнес-карта", "Mastercard Preferred"), но **не** их использование в комбинации с общими словами, как например, "идеальное решение для вашего бизнеса".
-    - Не считай фразы, используемые в исходном описании продукта, как клише. Исходное описание: "{description}".
-    - Не считай фразы, используемые в преимуществах продукта, как клише. Преимущества: "{benefits}".
-    - Не считай фразы, используемые в ключевом сообщении, как клише. Ключевое сообщение: "{key_message}".
+    - Фразы, используемые в исходном описании продукта, кликбейтными. Исходное описание: "{description}".
+    - Фразы, используемые в преимуществах продукта, кликбейтными. Преимущества: "{benefits}".
     **Считай клише или канцеляризмами следующие типы выражений:**
     - Избитые фразы, такие как:
       - "Обеспечьте стабильность и развитие вашего бизнеса"
@@ -1697,8 +1504,8 @@ async def check_cliches_and_bureaucratese(message, description, benefits, key_me
     если в тексте **есть** такие выражения, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}}.
     **Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check_3(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1710,26 +1517,24 @@ async def check_cliches_and_bureaucratese(message, description, benefits, key_me
         return None
 
 # 29. Проверка на соответствие описанию предложения и ключевому сообщению
-async def check_no_contradictions(message, description, benefits, key_message):
+def check_no_contradictions(message, description, key_message):
     print()
     print("Проверка 29: Проверка на отсутствие противоречий с описанием предложения")
     print()
     message_clean = cut_message(message)
-    prompt = f'''Проверь, не противоречит ли следующее сообщение описанию предложения, преимуществам и ключевому сообщению. Учти, что сообщение является выжимкой из описания предложения и не может содержать столько же информации в том же объеме, сколько описание предложения - важно, чтобы в сообщении не было указано ложных фактов.
+    prompt = f'''Проверь, не противоречит ли следующее сообщение описанию предложения и ключевому сообщению. Учти, что сообщение является выжимкой из описания предложения и не может содержать столько же информации в том же объеме, сколько описание предложения - важно, чтобы в сообщении не было указано ложных фактов.
 Описание предложения:
 "{description}"
-Преимушества:
-"{benefits}"
 Ключевое сообщение:
 "{key_message}"
 Сообщение:
 "{message}"
-Если сообщение не содержит фактов, которые отсутствуют в описании предложения, преиуществах и ключевом сообщении, **верни только** JSON {{"decision": false, "explanation": "Противоречий не обнаружено."}}.
-Если сообщение содержит факты, которые отсутствуют в описании предложения, преимуществах и ключевом сообщении, **верни только** JSON {{"decision": true, "explanation": "<описание противоречий>"}}.
+Если сообщение не содержит фактов, которые отсутствуют в описании предложения и ключевом сообщении, **верни только** JSON {{"decision": false, "explanation": "Противоречий не обнаружено."}}.
+Если сообщение содержит факты, которые отсутствую�� в описании предложения и ключевом сообщении, **верни только** JSON {{"decision": true, "explanation": "<описание противоречий>"}}.
 **Не добавляй никакого дополнительного текста. Отвечай только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1741,7 +1546,7 @@ async def check_no_contradictions(message, description, benefits, key_message):
         return None
 
 # 30. Проверка на наличие ключевого сообщения
-async def check_contains_key_message(message, key_message):
+def check_contains_key_message(message, key_message):
     print()
     print("Проверка 30: Проверка на наличие ключевого сообщения")
     print()
@@ -1755,8 +1560,8 @@ async def check_contains_key_message(message, key_message):
 Если сообщение **не содержит всю** информацию из ключевого текста, **верни только** JSON {{"decision": true, "explanation": "Ключевое текст отсутствует."}}.
 **Не добавляй никакого дополнительного текста. Отвечай только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1768,7 +1573,7 @@ async def check_contains_key_message(message, key_message):
         return None
 
 # 31. Проверка на точное совпадение названий продуктов
-async def check_product_name_consistency(message, product_name):
+def check_product_name_consistency(message, product_name):
     print()
     print("Проверка 31: Проверка на точное совпадение названий продуктов")
     print()
@@ -1785,8 +1590,8 @@ async def check_product_name_consistency(message, product_name):
 Если название продукта **не** совпадает, **верни только** JSON {{"decision": true, "explanation": "<описание несоответствия>"}}.
 **Не добавляй никакого дополнительного текста. Ответ должен быть только в формате JSON с закрывающими кавычками и скобками.**'''
 
-    response = await generate_message_mistral_check_3(prompt)
-#    await asyncio.sleep(3)
+    response = generate_message_mistral_check(prompt)
+    time.sleep(3)
     print("Mistral response:", response)
     result = parse_json_response(response)
     if result is not None:
@@ -1800,63 +1605,61 @@ async def check_product_name_consistency(message, product_name):
 # ФУНКЦИИ ПРОВЕРОК (КОНЕЦ)
 
 
-async def safe_check(func, *args):
+def safe_check(func, *args):
     try:
-        if inspect.iscoroutinefunction(func):
-            return await func(*args)
-        else:
-            return func(*args)
+        return func(*args)
     except Exception as e:
-        print(f"Error in {func.__name__}: {e}")
-        return None
+        # Optionally, you can log the exception here if needed
+        print(f"Ошибка в {func.__name__}: {e}")
+        return None  # Indicate that the check could not be performed
 
-async def perform_checks(message, description, key_message, product_name, benefits):
+def perform_checks(message, description, key_message, product_name, benefits):
     checks = {}
 
     # 2. Morphological checks using pymorphy3
     morphological_checks = [
-#        ("no_word_repetitions", check_no_word_repetitions),
         ("forbidden_words", check_forbidden_words),
-#        ("double_verbs", check_no_double_verbs),
-        ("adverbial_participles", check_no_adverbial_participles),
-        ("derived_prepositions", check_no_derived_prepositions),
-        ("multiple_nouns", check_no_multiple_nouns),
-        ("introductory_phrases", check_no_introductory_phrases),
-        ("compound_sentences", check_no_compound_sentences),
-        ("superlative_adjectives", check_no_superlative_adjectives),
         ("client_addressing", check_no_greeting),
         ("promises", check_no_promises),
-        ("amplifiers", check_no_amplifiers),
+        ("double_verbs", check_no_double_verbs),
         ("participles", check_no_participles),
+        ("adverbial_participles", check_no_adverbial_participles),
+        ("superlative_adjectives", check_no_superlative_adjectives),
         ("passive_voice", check_no_passive_voice),
         ("written_out_ordinals", check_no_written_out_ordinals),
         ("subordinate_clauses_chain", check_no_subordinate_clauses_chain),
         ("repeating_conjunctions", check_no_repeating_conjunctions),
+        ("introductory_phrases", check_no_introductory_phrases),
+        ("amplifiers", check_no_amplifiers),
         ("time_parasites", check_no_time_parasites),
+        ("multiple_nouns", check_no_multiple_nouns),
+        ("derived_prepositions", check_no_derived_prepositions),
+        ("compound_sentences", check_no_compound_sentences),
         ("dates_written_out", check_no_dates_written_out),
+        ("no_word_repetitions", check_no_word_repetitions),
     ]
 
     # 3. LLM checks: check_clickbait_phrases, check_abstract_claims, check_cliches_and_bureaucratese
     llm_checks_group1 = [
+        ("product_name_consistency", check_product_name_consistency),
         ("no_contradictions", check_no_contradictions),
+        ("clickbait_phrases", check_clickbait_phrases),
         ("abstract_claims", check_abstract_claims),
         ("cliches_and_bureaucratese", check_cliches_and_bureaucratese),
-        ("clickbait_phrases", check_clickbait_phrases),
-        ("contains_key_message", check_contains_key_message),
-        ("synonymous_members", check_synonymous_members),
-        ("product_name_consistency", check_product_name_consistency)
+        ("contains_key_message", check_contains_key_message)
     ]
 
     # 4. Remaining LLM checks
     llm_checks_group2 = [
-        ("offensive_phrases", check_offensive_phrases),
         ("disconnected_sentences", check_disconnected_sentences),
-        ("specialized_terms", check_specialized_terms)
+        ("synonymous_members", check_synonymous_members),
+        ("specialized_terms", check_specialized_terms),
+        ("offensive_phrases", check_offensive_phrases),
     ]
 
     # Perform morphological checks
     for check_name, check_func in morphological_checks:
-        result = await safe_check(check_func, message)
+        result = safe_check(check_func, message)
         checks[check_name] = result
         if result is False:
             return checks  # Stop on first failure
@@ -1864,19 +1667,19 @@ async def perform_checks(message, description, key_message, product_name, benefi
     # Perform LLM checks group 1
     for check_name, check_func in llm_checks_group1:
         if check_name == "no_contradictions":
-            result = await safe_check(check_func, message, description, benefits, key_message)
+            result = safe_check(check_func, message, description, key_message)
         elif check_name == "contains_key_message":
-            result = await safe_check(check_func, message, key_message)
+            result = safe_check(check_func, message, key_message)
         elif check_name == "product_name_consistency":
-            result = await safe_check(check_func, message, product_name)
+            result = safe_check(check_func, message, product_name)
         elif check_name == "clickbait_phrases":
-            result = await safe_check(check_func, message, description, benefits, key_message)
+            result = safe_check(check_func, message, description, benefits)
         elif check_name == "abstract_claims":
-            result = await safe_check(check_func, message, description, benefits, key_message)
+            result = safe_check(check_func, message, description, benefits)
         elif check_name == "cliches_and_bureaucratese":
-            result = await safe_check(check_func, message, description, benefits, key_message)
+            result = safe_check(check_func, message, description, benefits)
         else:
-            result = await safe_check(check_func, message)
+            result = safe_check(check_func, message)
         checks[check_name] = result
         if result is False:
             return checks
@@ -1884,7 +1687,7 @@ async def perform_checks(message, description, key_message, product_name, benefi
 
     # Perform remaining LLM checks
     for check_name, check_func in llm_checks_group2:
-        result = await safe_check(check_func, message)
+        result = safe_check(check_func, message)
         checks[check_name] = result
         if result is False:
             return checks  # Stop on first failure
@@ -1897,7 +1700,7 @@ def format_checks(checks):
         "forbidden_words": "Запрещенные слова",
         "client_addressing": "Обращение к клиенту",
         "promises": "Обещания и гарантии",
-#        "double_verbs": "Два глагола подряд",
+        "double_verbs": "Два глагола подряд",
         "participles": "Причастия",
         "adverbial_participles": "Деепричастия",
         "superlative_adjectives": "Превосходная степень",
@@ -1912,7 +1715,7 @@ def format_checks(checks):
         "derived_prepositions": "Производные предлоги",
         "compound_sentences": "Сложноподчиненные предложения",
         "dates_written_out": "Даты прописью",
-#        "no_word_repetitions": "Повторы слов",
+        "no_word_repetitions": "Повторы слов",
         # Проверки на LLM
         "disconnected_sentences": "Сложные предложения без логической связи",
         "synonymous_members": "Близкие по смыслу однородные члены предложения",
@@ -1979,7 +1782,7 @@ with gr.Blocks() as demo:
                     key_message = gr.Textbox(
                         label="Ключевое сообщение (предзаполненный пример можно поменять на свой)",
                         lines=1,
-                        value="Повышенный льготный период при покупках у партнёров банка — до 365 дней и бесплатное годовое обслуживание."
+                        value="Повышенный льготный период при покупках у партнёров банка — до 365 дней и бесплатное годовое обслуживание кредитной бизнес-карты."
                     )
 
                 with gr.Column():