Update app.py

app.py

@@ -136,37 +136,37 @@ def generate_message_gigachat_plus(prompt, temperature=0.87):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
-def generate_message_gpt4o_with_retry(prompt, gpt4o_temperature):
+def generate_message_gpt4o_with_retry(prompt):
     for _ in range(10):  # Максимум 10 попыток
-        message = generate_message_gpt4o(prompt, gpt4o_temperature)
+        message = generate_message_gpt4o(prompt)
         if len(message) <= 250:
             return message
     return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
-def generate_message_gigachat_pro_with_retry(prompt, gigachat_pro_temperature):
+def generate_message_gigachat_pro_with_retry(prompt):
     for _ in range(10):
-        message = generate_message_gigachat_pro(prompt, gigachat_pro_temperature)
+        message = generate_message_gigachat_pro(prompt)
         if len(message) <= 250:
             return message
     return message
 
-def generate_message_gigachat_lite_with_retry(prompt, gigachat_lite_temperature):
+def generate_message_gigachat_lite_with_retry(prompt):
     for _ in range(10):
-        message = generate_message_gigachat_lite(prompt, gigachat_lite_temperature)
+        message = generate_message_gigachat_lite(prompt)
         if len(message) <= 250:
             return message
     return message
 
-def generate_message_gigachat_plus_with_retry(prompt, gigachat_plus_temperature):
+def generate_message_gigachat_plus_with_retry(prompt):
     for _ in range(10):
-        message = generate_message_gigachat_plus(prompt, gigachat_plus_temperature)
+        message = generate_message_gigachat_plus(prompt)
         if len(message) <= 250:
             return message
     return message
 
 
 # Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values, gpt4o_temperature=1, gigachat_pro_temperature=1, gigachat_lite_temperature=1, gigachat_plus_temperature=1):
+def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -177,33 +177,36 @@ def generate_messages(description, advantages, *selected_values, gpt4o_temperatu
         "gigachat_plus": None
     }
 
-    # Генерация сообщения с GPT-4o
-    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt, temperature=gpt4o_temperature)
+    yield results["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 сгенерировано"
 
-    # Генерация сообщения с GigaChat-Pro
-    results["gigachat_pro"] = generate_message_gigachat_pro_with_retry(standard_prompt, temperature=gigachat_pro_temperature)
+    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)
 
-    # Генерация сообщения с GigaChat-Lite
-    results["gigachat_lite"] = generate_message_gigachat_lite_with_retry(standard_prompt, temperature=gigachat_lite_temperature)
+    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)
 
-    # Генерация сообщения с GigaChat-Plus
-    results["gigachat_plus"] = generate_message_gigachat_plus_with_retry(standard_prompt, temperature=gigachat_plus_temperature)
+    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, "Все сообщения сгенерированы"
 
     return results
 
 
+
 # Функция для генерации персонализиров��нного промпта
 def generate_personalization_prompt(*selected_values):
     prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"