Update app.py

app.py

@@ -191,14 +191,17 @@ def generate_message_mistral_generate(prompt, max_retries=5):
             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
+            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} секунд перед повторной попыткой...")
+                time.sleep(wait_time)
+                retries += 1
+            else:
+                print(f"Ошибка при обращении к Mistral: {e}")
+                return None
+    print("Не удалось получить ответ от Mistral после максимального числа попыток.")
+    return None
 
 def generate_message_mistral_check(prompt, max_retries=5):
     retries = 0
@@ -237,7 +240,8 @@ def generate_message_mistral_check(prompt, max_retries=5):
 
 # Функция для замены сокращений с 'k' или 'К' на тысячи
 def replace_k_with_thousands(message):
-    # Замена и для 'k' и для 'К', с учётом регистра
+    if not isinstance(message, str):
+        return message  # or you can return an empty string
     message = re.sub(r'(\d+)[kкКK]', r'\1 000', message, flags=re.IGNORECASE)
     return message
 
@@ -512,6 +516,10 @@ def generate_message_mistral_with_retry(prompt, current_prefix, description, key
     for attempt in range(20):
         gr.Info(f"Итерация {attempt + 1}: генерируется сообщение...")
         message = generate_message_mistral_generate(prompt)
+        if message is None:
+            print("Не удалось получить сообщение от Mistral, повторная попытка...")
+            time.sleep(1)
+            continue  # Retry
         message = replace_k_with_thousands(message)
         message = correct_dash_usage(message)
         message_length = len(message)
@@ -526,7 +534,7 @@ def generate_message_mistral_with_retry(prompt, current_prefix, description, key
             if passed is False:
                 approach_stats[current_prefix]["failed_checks"][check_name] += 1
                 break
-        notify_failed_checks(checks)  # Вызываем функцию независимо от результата проверок
+        notify_failed_checks(checks)
         if all(checks.values()):
             return message
         prompt = append_errors_to_prompt(prompt, checks)