Update app.py

app.py

@@ -10,6 +10,7 @@ import base64
 import time
 from openai import OpenAI
 import string
+import math
 
 XAI_API_KEY = os.getenv("XAI_API_KEY")
 client = OpenAI(
@@ -455,21 +456,17 @@ def process_message(message, product_name):
 
     return final_message
 
-def generate_message_with_retry(model_prompt, product_name):
+def generate_message(model_prompt, product_name):
     last_message = ""
-    for _ in range(10):
-        msg = call_model(model_prompt)
-        msg = correct_dash_usage(msg)
-        msg = clean_message(msg)
-        msg = process_message(msg, product_name)
-        length = len(msg)
-        if 160 <= length <= 250:
-            msg += f"\n\n------\nКоличество знаков: {length}"
-            return msg
-        last_message = msg
-    length = len(last_message)
-    last_message += f"\n\n------\nКоличество знаков: {length}"
-    return last_message
+    msg = call_model(model_prompt)
+    msg = correct_dash_usage(msg)
+    msg = clean_message(msg)
+    msg = process_message(msg, product_name)
+    length = len(msg)
+    last_message = msg
+    length = len(msg)
+    msg += f"\n\n------\nКоличество знаков: {length}"
+    return msg
 
 def update_prompts_on_params_change(description, product_name, benefits, key_message, 
                                     gender, generation, psychotype, business_stage, industry, opf):
@@ -1193,6 +1190,16 @@ def lemmatize_word(word, morph):
     best = parsed[0]
     return best.normal_form, best.tag.POS
 
+# 0. Проверка на длину
+
+def check_length(message):
+    length = len(message)
+    if 160 <= length <= 250:
+        return True
+    else:
+        logging.warning(f"Не пройдена проверка: Длина сообщения {length} символов. Сообщение: {message}")
+        return False
+
 # 1. Запрещенные слова
 
 def check_forbidden_words(message, exceptions=None):
@@ -1756,6 +1763,112 @@ def check_no_word_repetitions(message, key_message, exceptions=None):
 
 # ФУНКЦИИ ПРОВЕРОК (КОНЕЦ)
 
+CRITICAL_CHECKS = [
+    "length_check",
+    "forbidden_words",
+    "client_addressing",
+    "promises",
+    "subordinate_clauses_chain",
+    "introductory_phrases",
+    "dates_written_out"
+]
+
+NON_CRITICAL_CHECKS = [
+    "double_verbs",
+    "participles",
+    "adverbial_participles",
+    "superlative_adjectives",
+    "passive_voice",
+    "repeating_conjunctions",
+    "amplifiers",
+    "time_parasites",
+    "multiple_nouns",
+    "derived_prepositions",
+    "compound_sentences",
+    "word_repetitions",    
+]
+
+def run_checks_critical_and_non_critical(message: str, key_message: str) -> (bool, list):
+    """
+    Возвращает (all_critical_passed: bool, failed_non_critical: list[str])
+    где failed_non_critical = список названий проверок, которые не пройдены, но не являются критическими.
+    """
+    checks = perform_checks(message, key_message)  # ваша функция, которая возвращает dict
+
+    all_critical_passed = True
+    failed_non_critical = []
+
+    for rule_name, result in checks.items():
+        # result либо True/False, либо (False, reason)
+        if isinstance(result, tuple):
+            pass_flag = result[0]
+        else:
+            pass_flag = bool(result)
+
+        if rule_name in CRITICAL_CHECKS:
+            if not pass_flag:  # Критическая не пройдена
+                all_critical_passed = False
+        elif rule_name in NON_CRITICAL_CHECKS:
+            if not pass_flag:
+                failed_non_critical.append(rule_name)
+        else:
+            # Все проверки, которых нет ни в CRITICAL_CHECKS, ни в NON_CRITICAL_CHECKS — игнорируем
+            pass
+
+    return all_critical_passed, failed_non_critical
+
+
+def attempt_generate_sms_with_checks(model_prompt: str, product_name: str, key_message: str):
+    """
+    Генерирует 1 SMS + запускает run_checks_critical_and_non_critical.
+    Возвращает (sms, all_critical_ok, failed_non_critical_list).
+    """
+    sms = generate_message(model_prompt, product_name)
+    cut_sms = cut_message(sms)
+    all_critical_ok, failed_non_critical = run_checks_critical_and_non_critical(cut_sms, key_message)
+    return sms, all_critical_ok, failed_non_critical
+
+
+def generate_sms_with_timer(model_prompt: str, product_name: str, key_message: str, max_time_sec=120):
+    """
+    Псевдо-синхронный цикл с таймером 120 секунд.
+    Возвращает "оптимальный" SMS, 
+    или строку "Не удалось за 2 минуты создать SMS, прошедшее все критические проверки".
+    """
+    start = time.time()
+    best_sms = None
+    best_non_crit_count = math.inf  # сколько некритич. проверок не пройдено (минимизируем)
+
+    while True:
+        now = time.time()
+        if now - start > max_time_sec:
+            # время вышло
+            break
+
+        sms, crit_ok, failed_non_crit = attempt_generate_sms_with_checks(model_prompt, product_name, key_message)
+
+        if crit_ok:
+            # значит SMS прошло все критические
+            non_crit_count = len(failed_non_crit)
+            if non_crit_count == 0:
+                # идеально
+                return sms  # тут же возвращаем
+            # если не 0, то проверяем, лучше ли, чем best_sms
+            if non_crit_count < best_non_crit_count:
+                best_non_crit_count = non_crit_count
+                best_sms = sms
+        else:
+            # crit fail => пропускаем
+            pass
+
+    # если цикл завершился по таймеру:
+    if best_sms is not None:
+        # вернём лучшее, что есть
+        return best_sms
+    else:
+        # значит не было ни одного SMS, прошедшего критические проверки
+        return "Не удалось за 2 минуты создать SMS, прошедшее все критические проверки."
+
 def cut_message(message: str):
     if '------' in message:
         message = message.split('------')[0].strip()