Update app.py

app.py

@@ -96,22 +96,63 @@ def replace_k_with_thousands(message):
     message = re.sub(r'(\d+)[kкКK]', r'\1 000', message, flags=re.IGNORECASE)
     return message
  
-# Функция для повторной генерации сообщения, пока оно не станет короче 250 знаков и не меньше 190 знаков
-# Добавляем проверку в функцию перегенерации
-# Функция для повторной генерации сообщения с выводом номера попытки через Gradio Alerts
-# Функция для повторной генерации сообщения с выводом номера попытки через Gradio Alerts
+# Функция для добавления ошибок в промпт для перегенерации
+def append_errors_to_prompt(prompt, checks):
+    errors = []
+    
+    if not checks["forbidden_words"]:
+        errors.append("Не использовать запрещённые слова.")
+    if not checks["client_addressing"]:
+        errors.append("Не обращаться к клиенту напрямую.")
+    if not checks["promises"]:
+        errors.append("Не давать обещания и гарантии.")
+    if not checks["double_verbs"]:
+        errors.append("Не использовать составные конструкции из двух глаголов.")
+    if not checks["participles"]:
+        errors.append("Не использовать причастия.")
+    if not checks["adverbial_participles"]:
+        errors.append("Не использовать деепричастия.")
+    
+    # Добавляем ошибки в конец промпта
+    if errors:
+        error_instructions = "Следующие ошибки необходимо избежать:\n" + "\n".join(errors)
+        prompt += f"\n\n{error_instructions}"
+    
+    return prompt
+    
+def notify_failed_length(message_length):
+    if message_length < 190:
+        gr.Warning(f"Сообщение слишком короткое: {message_length} знаков. Минимум 190.")
+        return False
+    elif message_length > 250:
+        gr.Warning(f"Сообщение слишком длинное: {message_length} знаков. Максимум 250.")
+        return False
+    return True
+
+# Функция для уведомления о непройденных проверках
+def notify_failed_checks(checks):
+    failed_checks = [rule for rule, passed in checks.items() if not passed]
+    if failed_checks:
+        gr.Warning(f"Сообщение не прошло следующие проверки: {', '.join(failed_checks)}")
+
+# Модифицированная функция перегенерации сообщений с уведомлениями о номере попытки
 def generate_message_gigachat_pro_with_retry(prompt):
     last_message = None  # Храним последнее сообщение
 
     for attempt in range(10):
-        # Информация о начале новой попытки
-        if attempt > 0:  # Показываем новое уведомление только после первой попытки
-            time.sleep(1)  # Ждем, чтобы старое уведомление успело исчезнуть
-        gr.Info(f"Попытка {attempt + 1}: генерируется сообщение...")
+        # Уведомление о текущей попытке
+        gr.Info(f"Итерация {attempt + 1}: генерируется сообщение...")
 
+        # Генерация сообщения
         message = generate_message_gigachat_pro(prompt)
         message_length = len(message)
 
+        # Проверка длины сообщения. Если не проходит, сразу перегенерируем
+        if not notify_failed_length(message_length):
+            last_message = message  # Обновляем последнее сообщение
+            time.sleep(1)  # Пауза перед следующей попыткой
+            continue
+
         # Выполняем замены и проверки
         message = replace_k_with_thousands(message)
         checks = perform_checks(message)
@@ -119,16 +160,17 @@ def generate_message_gigachat_pro_with_retry(prompt):
         # Обновляем последнее сообщение
         last_message = message
 
-        if 190 <= message_length <= 250:
-            # Успешная генерация сообщения
+        # Если сообщение прошло все проверки
+        if all(checks.values()):
             return message
         
-        # Предупреждение, что сообщение не прошло проверку
-        gr.Warning(f"Попытка {attempt + 1}: сообщение не прошло проверку, пробуем снова...")
-        time.sleep(1)  # Добавляем небольшую паузу между попытками
+        # Если есть ошибки, уведомляем о непройденных проверках
+        notify_failed_checks(checks)
 
-    # Сообщаем, что возвращаем последнее сгенерированное сообщение после 10 попыток
-    gr.Info("Не удалось сгенерировать сообщение, соответствующее требованиям, за 10 попыток. Возвращаем последнее сгенерированное сообщение.")
+        time.sleep(1)  # Пауза перед следующей попыткой
+
+    # Возвращаем последнее сообщение после 10 попыток
+    gr.Info("Не удалось сгенерировать сообщение, соответствующее требованиям, за 10 итераций. Возвращаем последнее сгенерированное сообщение.")
     return last_message
 
 
@@ -240,7 +282,6 @@ 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):
     # Генерация задания для копирайтера
@@ -262,8 +303,16 @@ 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]
@@ -272,19 +321,17 @@ 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,  # Задание для редактора без префиксов, суффиксов и пустых строк
-            None, None, None, None, None, 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 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
         )
 
-        # Генерация неперсонализированного сообщения
-        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)
@@ -295,11 +342,11 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         # Выводим персонализированное сообщение
         yield (
             standard_prompt, display_personalization_prompt,  # Задание для редактора без префиксов, суффиксов и пустых строк
-            non_personalized_messages[0] if i >= 0 else None,
+            non_personalized_messages[0] if len(non_personalized_messages) > 0 else None,
             personalized_messages[0] if len(personalized_messages) > 0 else None,
-            non_personalized_messages[1] if i >= 1 else None,
+            non_personalized_messages[1] if len(non_personalized_messages) > 1 else None,
             personalized_messages[1] if len(personalized_messages) > 1 else None,
-            non_personalized_messages[2] if i >= 2 else None,
+            non_personalized_messages[2] if len(non_personalized_messages) > 2 else None,
             personalized_messages[2] if len(personalized_messages) > 2 else None
         )