Update app.py

app.py

@@ -230,27 +230,43 @@ def perform_personalization_gigachat(standard_message, personalization_prompt, m
         result = generate_message_gigachat_plus(full_prompt)
     return clean_message(result)
 
+
+def perform_personalization_with_retry(standard_message, personalization_prompt):
+    for _ in range(10):  # Максимум 10 попыток
+        message = perform_personalization(standard_message, personalization_prompt)
+        if len(message) <= 250:
+            return message
+    return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
+
+def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model):
+    for _ in range(10):
+        message = perform_personalization_gigachat(standard_message, personalization_prompt, model)
+        if len(message) <= 250:
+            return message
+    return message
+
+
 # Обновляем блок персонализации
 def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, *selected_values):
     personalization_prompt = generate_personalization_prompt(*selected_values)
     yield personalization_prompt, "", "", "", "", "Промпт для персонализации сгенерирован"
 
-    personalized_message_gpt4o = perform_personalization(gpt4o_message, personalization_prompt)
+    personalized_message_gpt4o = perform_personalization_with_retry(gpt4o_message, personalization_prompt)
     gpt4o_length = len(personalized_message_gpt4o)
     gpt4o_display = f"{personalized_message_gpt4o}\n\n------\nКоличество знаков: {gpt4o_length}"
     yield personalization_prompt, gpt4o_display, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
 
-    personalized_message_gigachat_pro = perform_personalization_gigachat(gigachat_pro_message, personalization_prompt, "gigachat_pro")
+    personalized_message_gigachat_pro = perform_personalization_gigachat_with_retry(gigachat_pro_message, personalization_prompt, "gigachat_pro")
     gigachat_pro_length = len(personalized_message_gigachat_pro)
     gigachat_pro_display = f"{personalized_message_gigachat_pro}\n\n------\nКоличество знаков: {gigachat_pro_length}"
     yield personalization_prompt, gpt4o_display, gigachat_pro_display, "", "", "Персонализированное сообщение GigaChat-Pro сгенерировано"
 
-    personalized_message_gigachat_lite = perform_personalization_gigachat(gigachat_lite_message, personalization_prompt, "gigachat_lite")
+    personalized_message_gigachat_lite = perform_personalization_gigachat_with_retry(gigachat_lite_message, personalization_prompt, "gigachat_lite")
     gigachat_lite_length = len(personalized_message_gigachat_lite)
     gigachat_lite_display = f"{personalized_message_gigachat_lite}\n\n------\nКоличество знаков: {gigachat_lite_length}"
     yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Персонализированное сообщение GigaChat-Lite сгенерировано"
 
-    personalized_message_gigachat_plus = perform_personalization_gigachat(gigachat_plus_message, personalization_prompt, "gigachat_plus")
+    personalized_message_gigachat_plus = perform_personalization_gigachat_with_retry(gigachat_plus_message, personalization_prompt, "gigachat_plus")
     gigachat_plus_length = len(personalized_message_gigachat_plus)
     gigachat_plus_display = f"{personalized_message_gigachat_plus}\n\n------\nКоличество знаков: {gigachat_plus_length}"
     yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все персонализированные сообщения сгенерированы"