Update app.py

app.py

@@ -1170,21 +1170,35 @@ def generate_message_meta_llama_3_1_405b_with_retry(prompt):
             return correct_dash_usage(message)
     return message
 
-def generate_message_mistral(prompt):
-    try:
-        chat_response = client_mistral.chat.complete(
-            model="open-mistral-nemo",
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                },
-            ]
-        )
-        cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к Mistral: {e}"
+def generate_message_mistral(prompt, max_retries=5):
+    retries = 0
+    while retries < max_retries:
+        try:
+            chat_response = client_mistral.chat.complete(
+                model="mistral-large-latest",
+                messages=[
+                    {
+                        "role": "user",
+                        "content": prompt
+                    },
+                ],
+                max_tokens=3000,  # Увеличено до 3000
+                temperature=0.2,
+                stop=["}"]  # Останавливаем генерацию после закрывающей скобки
+            )
+            cleaned_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
+    print("Ошибка: превышено количество попыток обращения к Mistral")
+    return None
 
 def generate_messages(description, advantages, key_message, approach, *selected_values):
     
@@ -2077,6 +2091,22 @@ def check_no_word_repetitions(message):
 
 # Проверки на LLM
 
+def parse_json_response(response):
+    try:
+        # Попытка найти самый длинный корректный JSON в ответе
+        import re
+        matches = re.findall(r'\{(?:[^{}]|(?R))*\}', response)
+        if matches:
+            json_str = matches[-1]  # Берем последний найденный JSON
+            result = json.loads(json_str)
+            return result
+        else:
+            print("JSON не найден в ответе")
+            return None
+    except json.JSONDecodeError as e:
+        print(f"Ошибка при разборе JSON: {e}")
+        return None
+        
 # 22. Проверка сложных предложений без логической связи
 def check_disconnected_sentences(message):
     prompt = f'''Проверь следующий текст на наличие сложных предложений, где отсутствует логическая связь между частями:
@@ -2086,19 +2116,18 @@ def check_disconnected_sentences(message):
 {{"decision": true, "explanation": "В тексте есть предложения, которые не связаны логически."}}
 Если такие предложения есть, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}};
 если таких предложений нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
-**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON.**'''
+**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
     response = generate_message_mistral(prompt)
     time.sleep(3)  # Задержка в 3 секунды между запросами
     print("Mistral response:", response)  # Выводим полный ответ модели
-    try:
-        result = json.loads(response)
+    result = parse_json_response(response)
+    if result is not None:
         decision = result.get("decision", False)
         explanation = result.get("explanation", "")
         print("Explanation:", explanation)
-        return decision
-    except Exception as e:
-        print(f"Ошибка при разборе JSON: {e}")
+        return not decision  # Инвертируем логику
+    else:
         return None
 
 # 23. Проверка на близкие по смыслу однородные члены
@@ -2110,19 +2139,18 @@ def check_synonymous_members(message):
 {{"decision": true, "explanation": "В предложении используются синонимы, передающие одну идею."}}
 Если такие слова или выражения есть, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}};
 если таких слов или выражений нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
-**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON.**'''
+**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
     response = generate_message_mistral(prompt)
     time.sleep(3)
     print("Mistral response:", response)
-    try:
-        result = json.loads(response)
+    result = parse_json_response(response)
+    if result is not None:
         decision = result.get("decision", False)
         explanation = result.get("explanation", "")
         print("Explanation:", explanation)
-        return decision
-    except Exception as e:
-        print(f"Ошибка при разборе JSON: {e}")
+        return not decision  # Инвертируем логику
+    else:
         return None
 
 # 24. Проверка на шокирующие, экстравагантные или кликбейтные фразы
@@ -2137,19 +2165,18 @@ def check_clickbait_phrases(message):
 {{"decision": false, "explanation": "Текст нейтрален и не содержит кликбейтных фраз."}}
 Если такие фразы есть, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}};
 если таких фраз нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
-**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON.**'''
+**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
     response = generate_message_mistral(prompt)
     time.sleep(3)
     print("Mistral response:", response)
-    try:
-        result = json.loads(response)
+    result = parse_json_response(response)
+    if result is not None:
         decision = result.get("decision", False)
         explanation = result.get("explanation", "")
         print("Explanation:", explanation)
-        return decision
-    except Exception as e:
-        print(f"Ошибка при разборе JSON: {e}")
+        return not decision  # Инвертируем логику
+    else:
         return None
 
 # 25. Проверка на абстрактные заявления без поддержки фактами
@@ -2161,19 +2188,18 @@ def check_abstract_claims(message):
 {{"decision": true, "explanation": "Текст содержит общие заявления без конкретики."}}
 Если такие утверждения есть, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}};
 если таких утверждений нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
-**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON.**'''
+**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
     response = generate_message_mistral(prompt)
     time.sleep(3)
     print("Mistral response:", response)
-    try:
-        result = json.loads(response)
+    result = parse_json_response(response)
+    if result is not None:
         decision = result.get("decision", False)
         explanation = result.get("explanation", "")
         print("Explanation:", explanation)
-        return decision
-    except Exception as e:
-        print(f"Ошибка при разборе JSON: {e}")
+        return not decision  # Инвертируем логику
+    else:
         return None
 
 # 26. Проверка на узкоспециализированные термины
@@ -2184,19 +2210,18 @@ def check_specialized_terms(message):
 {{"decision": false, "explanation": "В тексте отсутствуют узко��пециализированные термины."}}
 Если в тексте есть такие термины, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}};
 если таких терминов нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
-**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON.**'''
+**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
     response = generate_message_mistral(prompt)
     time.sleep(3)
     print("Mistral response:", response)
-    try:
-        result = json.loads(response)
+    result = parse_json_response(response)
+    if result is not None:
         decision = result.get("decision", False)
         explanation = result.get("explanation", "")
         print("Explanation:", explanation)
-        return decision
-    except Exception as e:
-        print(f"Ошибка при разборе JSON: {e}")
+        return not decision  # Инвертируем логику
+    else:
         return None
 
 # 27. Проверка на двусмысленные или обидные фразы
@@ -2208,19 +2233,18 @@ def check_offensive_phrases(message):
 {{"decision": true, "explanation": "Текст содержит фразу, которая может быть обидной."}}
 Если такие фразы есть, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}};
 если таких фраз нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
-**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON.**'''
+**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
     response = generate_message_mistral(prompt)
     time.sleep(3)
     print("Mistral response:", response)
-    try:
-        result = json.loads(response)
+    result = parse_json_response(response)
+    if result is not None:
         decision = result.get("decision", False)
         explanation = result.get("explanation", "")
         print("Explanation:", explanation)
-        return decision
-    except Exception as e:
-        print(f"Ошибка при разборе JSON: {e}")
+        return not decision  # Инвертируем логику
+    else:
         return None
 
 # 28. Проверка на речевые клише, рекламные штампы и канцеляризмы
@@ -2231,22 +2255,20 @@ def check_cliches_and_bureaucratese(message):
 {{"decision": false, "explanation": "Текст не содержит клише или канцеляризмов."}}
 Если в тексте есть такие выражения, **верни только** JSON {{"decision": true, "explanation": "<пояснение>"}};
 если таких выражений нет, **верни только** JSON {{"decision": false, "explanation": "<пояснение>"}}.
-**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON.**'''
+**Не добавляй никакого дополнительного текста. Перед ответом убедись, что отвечаешь только в формате JSON с закрывающими кавычками и скобками.**'''
 
     response = generate_message_mistral(prompt)
     time.sleep(3)
     print("Mistral response:", response)
-    try:
-        result = json.loads(response)
+    result = parse_json_response(response)
+    if result is not None:
         decision = result.get("decision", False)
         explanation = result.get("explanation", "")
         print("Explanation:", explanation)
-        return decision
-    except Exception as e:
-        print(f"Ошибка при разборе JSON: {e}")
+        return not decision  # Инвертируем логическое значение
+    else:
         return None
 
-
 # ФУНКЦИИ ПРОВЕРОК (КОНЕЦ)
 
 def safe_check(func, message):