Update app.py

app.py

@@ -131,6 +131,34 @@ def generate_message_gigachat_plus(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
+def generate_message_gpt4o_with_retry(prompt):
+    for _ in range(10):  # Максимум 10 попыток
+        message = generate_message_gpt4o(prompt)
+        if len(message) <= 250:
+            return message
+    return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
+
+def generate_message_gigachat_pro_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_gigachat_pro(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
+def generate_message_gigachat_lite_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_gigachat_lite(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
+def generate_message_gigachat_plus_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_gigachat_plus(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
 # Обновляем генерацию сообщений для отображения в интерфейсе
 def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
@@ -145,26 +173,26 @@ def generate_messages(description, advantages, *selected_values):
 
     yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
-    results["gpt4o"] = generate_message_gpt4o(standard_prompt)
+    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt)
     gpt4o_length = len(results["gpt4o"])
     gpt4o_display = f"{results['gpt4o']}\n\n------\nКоличество знаков: {gpt4o_length}"
     yield results["prompt"], gpt4o_display, "", "", "", "Сообщение GPT-4o сгенерировано"
 
-    results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
+    results["gigachat_pro"] = generate_message_gigachat_pro_with_retry(standard_prompt)
     gigachat_pro_length = len(results["gigachat_pro"])
     gigachat_pro_display = f"{results['gigachat_pro']}\n\n------\nКоличество знаков: {gigachat_pro_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "Сообщение GigaChat-Pro сгенерировано"
 
     time.sleep(2)
-    
-    results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
+
+    results["gigachat_lite"] = generate_message_gigachat_lite_with_retry(standard_prompt)
     gigachat_lite_length = len(results["gigachat_lite"])
     gigachat_lite_display = f"{results['gigachat_lite']}\n\n------\nКоличество знаков: {gigachat_lite_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Сообщение GigaChat-Lite сгенерировано"
 
     time.sleep(2)
-    
-    results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
+
+    results["gigachat_plus"] = generate_message_gigachat_plus_with_retry(standard_prompt)
     gigachat_plus_length = len(results["gigachat_plus"])
     gigachat_plus_display = f"{results['gigachat_plus']}\n\n------\nКоличество знаков: {gigachat_plus_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все сообщения сгенерированы"