Update app.py

app.py

@@ -105,6 +105,8 @@ def generate_message_gigachat_pro_with_retry(prompt):
 
     for attempt in range(10):
         # Информация о начале новой попытки
+        if attempt > 0:  # Показываем новое уведомление только после первой попытки
+            time.sleep(1)  # Ждем, чтобы старое уведомление успело исчезнуть
         gr.Info(f"Попытка {attempt + 1}: генерируется сообщение...")
 
         message = generate_message_gigachat_pro(prompt)
@@ -117,7 +119,7 @@ def generate_message_gigachat_pro_with_retry(prompt):
         # Обновляем последнее сообщение
         last_message = message
 
-        if 190 <= message_length <= 250 and all(checks.values()):
+        if 190 <= message_length <= 250:
             # Успешная генерация сообщения
             return message
         
@@ -238,6 +240,7 @@ def clean_prompt_for_display(prompt, prefixes, suffixes):
 
     return cleaned_prompt.strip()
 
+# Функция для постепенной генерации всех сообщений через yield
 # Функция для постепенной генерации всех сообщений через yield
 def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
     # Генерация задания для копирайтера
@@ -259,16 +262,8 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
     non_personalized_messages = []
     personalized_messages = []
 
-    # Генерация и постепенная подача каждого сообщения
+    # Генерация задания для редактора отдельно перед генерацией сообщений
     for i in range(3):
-        # Генерация неперсонализированного сообщения
-        prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
-        non_personalized_message = generate_message_gigachat_pro_with_retry(prompt)
-        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)
-
-        # Генерация задания для редактора
         personalization_prompt = generate_personalization_prompt(
             key_message, gender, generation, psychotype, business_stage, industry, opf,
             prefix=prefixes[i], suffix=suffixes[i]
@@ -277,17 +272,19 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         # Удаляем префиксы, суффиксы и пустые строки перед выводом на экран
         display_personalization_prompt = clean_prompt_for_display(personalization_prompt, prefixes, suffixes)
 
-        # Выводим неперсонализированное сообщение и задание для редактора
+        # Сначала отображаем задание для редактора
         yield (
             standard_prompt, display_personalization_prompt,  # Задание для редактора без префиксов, суффиксов и пустых строк
-            non_personalized_messages[0] if i >= 0 else None,
-            personalized_messages[0] if len(personalized_messages) > 0 else None,
-            non_personalized_messages[1] if i >= 1 else None,
-            personalized_messages[1] if len(personalized_messages) > 1 else None,
-            non_personalized_messages[2] if i >= 2 else None,
-            personalized_messages[2] if len(personalized_messages) > 2 else None
+            None, None, None, None, None, None  # Сообщения еще не сгенерированы
         )
 
+        # Генерация неперсонализированного сообщения
+        prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
+        non_personalized_message = generate_message_gigachat_pro_with_retry(prompt)
+        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)
+
         # Генерация персонализированного сообщения
         full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
         personalized_message = generate_message_gigachat_pro_with_retry(full_personalized_prompt)
@@ -298,11 +295,11 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         # Выводим персонализированное сообщение
         yield (
             standard_prompt, display_personalization_prompt,  # Задание для редактора без префиксов, суффиксов и пустых строк
-            non_personalized_messages[0] if len(non_personalized_messages) > 0 else None,
+            non_personalized_messages[0] if i >= 0 else None,
             personalized_messages[0] if len(personalized_messages) > 0 else None,
-            non_personalized_messages[1] if len(non_personalized_messages) > 1 else None,
+            non_personalized_messages[1] if i >= 1 else None,
             personalized_messages[1] if len(personalized_messages) > 1 else None,
-            non_personalized_messages[2] if len(non_personalized_messages) > 2 else None,
+            non_personalized_messages[2] if i >= 2 else None,
             personalized_messages[2] if len(personalized_messages) > 2 else None
         )