Update app.py

app.py

@@ -98,12 +98,12 @@ def replace_k_with_thousands(message):
  
 # Функция для повторной генерации сообщения, пока оно не станет короче 250 знаков и не меньше 190 знаков
 # Добавляем проверку в функцию перегенерации
-# Функция для повторной генерации сообщения с выводом номера попытки
+# Функция для повторной генерации сообщения с выводом номера попытки через Gradio Alerts
 def generate_message_gigachat_pro_with_retry(prompt):
     for attempt in range(10):
-        # Сообщаем о попытке
-        yield f"Попытка {attempt + 1}", None  # Выводим информацию о попытке
-
+        # Информация о начале новой попытки
+        gr.Info(f"Попытка {attempt + 1}: генерируется сообщение...")
+        
         message = generate_message_gigachat_pro(prompt)
         message_length = len(message)
 
@@ -112,12 +112,16 @@ def generate_message_gigachat_pro_with_retry(prompt):
         checks = perform_checks(message)
 
         if 190 <= message_length <= 250 and all(checks.values()):
-            yield message, None  # Возвращаем сообщение при успешной проверке
-            return  # Выходим из функции при успешной генерации
+            # Успешная генерация сообщения
+            return message
+        
+        # Предупреждение, что сообщение не прошло проверку
+        gr.Warning(f"Попытка {attempt + 1}: сообщение не прошло проверку, пробуем снова...")
         time.sleep(1)  # Добавляем небольшую паузу между попытками
 
     # Если не удалось сгенерировать подходящее сообщение за 10 попыток
-    yield "Не удалось сгенерировать сообщение за 10 попыток", None
+    raise gr.Error("Не удалось сгенерировать сообщение за 10 попыток.")
+
 
 
 # Функция для создания задания для копирайтера