Update app.py

app.py

@@ -342,7 +342,7 @@ def adapt_messages_to_best_example(
     approach,
     *selected_values
 ):
-    # Extract personalization parameters
+    # Извлечение параметров персонализации (без изменений)
     gender = selected_values[0]
     generation = selected_values[1]
     psychotype = selected_values[2]
@@ -350,15 +350,15 @@ def adapt_messages_to_best_example(
     industry = selected_values[4]
     legal_form = selected_values[5]
     
-    # Retrieve the reference message
+    # Получение эталонного сообщения (без изменений)
     reference_message = get_reference_message(gender, generation, psychotype, business_stage, industry, legal_form)
     
     if not reference_message:
-        # No reference message found
-        adapted_messages = ["Эталонное сообщение не найдено для выбранных параметров персонализации."] * 5
-        return adapted_messages
+        # Если эталонное сообщение не найдено
+        adapted_message = "Эталонное сообщение не найдено для выбранных параметров персонализации."
+        yield adapted_message, adapted_message, adapted_message, adapted_message, adapted_message
     else:
-        # Construct the prompt template
+        # Шаблон промпта (без изменений)
         prompt_template = (
             "Пожалуйста, перепиши следующее сообщение, сохраняя его смысл, но используя стиль, похожий на приведённое ниже эталонное сообщение. Не добавляй новую информацию и не изменяй факты.\n\n"
             "Эталонное сообщение:\n\"{reference_message}\"\n\n"
@@ -366,49 +366,52 @@ def adapt_messages_to_best_example(
             "Перепиши сообщение для адаптации в стиле эталонного сообщения."
         )
         
-        # Adapt each message
-        adapted_messages = []
-        # For GigaChat-Pro
+        # Инициализация пустых переменных для сообщений
+        adapted_gigachat_pro = ""
+        adapted_gigachat_lite = ""
+        adapted_gigachat_plus = ""
+        adapted_gpt4o = ""
+        adapted_meta_llama_405b = ""
+        
+        # Адаптация сообщения для GigaChat-Pro
         prompt = prompt_template.format(
             reference_message=reference_message,
             personalized_message=personalized_gigachat_pro
         )
         adapted_gigachat_pro = generate_message_gigachat_pro_with_retry(prompt)
-        adapted_messages.append(adapted_gigachat_pro)
+        yield adapted_gigachat_pro, "", "", "", ""
         
-        # For GigaChat-Lite
+        # Адаптация сообщения для GigaChat-Lite
         prompt = prompt_template.format(
             reference_message=reference_message,
             personalized_message=personalized_gigachat_lite
         )
         adapted_gigachat_lite = generate_message_gigachat_lite_with_retry(prompt)
-        adapted_messages.append(adapted_gigachat_lite)
+        yield adapted_gigachat_pro, adapted_gigachat_lite, "", "", ""
         
-        # For GigaChat-Plus
+        # Адаптация сообщения для GigaChat-Plus
         prompt = prompt_template.format(
             reference_message=reference_message,
             personalized_message=personalized_gigachat_plus
         )
         adapted_gigachat_plus = generate_message_gigachat_plus_with_retry(prompt)
-        adapted_messages.append(adapted_gigachat_plus)
+        yield adapted_gigachat_pro, adapted_gigachat_lite, adapted_gigachat_plus, "", ""
         
-        # For GPT-4o
+        # Адаптация сообщения для GPT-4o
         prompt = prompt_template.format(
             reference_message=reference_message,
             personalized_message=personalized_gpt4o
         )
         adapted_gpt4o = generate_message_gpt4o_with_retry(prompt)
-        adapted_messages.append(adapted_gpt4o)
+        yield adapted_gigachat_pro, adapted_gigachat_lite, adapted_gigachat_plus, adapted_gpt4o, ""
         
-        # For Meta-Llama-3.1-405B
+        # Адаптация сообщения для Meta-Llama-3.1-405B
         prompt = prompt_template.format(
             reference_message=reference_message,
             personalized_message=personalized_meta_llama_405b
         )
         adapted_meta_llama_405b = generate_message_meta_llama_3_1_405b_with_retry(prompt)
-        adapted_messages.append(adapted_meta_llama_405b)
-        
-        return adapted_messages
+        yield adapted_gigachat_pro, adapted_gigachat_lite, adapted_gigachat_plus, adapted_gpt4o, adapted_meta_llama_405b
 
 
 def update_best_example_prompt(*selected_values):