Update app.py

app.py

@@ -369,28 +369,43 @@ def adapt_messages_to_best_example(
         # Adapt each message
         adapted_messages = []
         # For GigaChat-Pro
-        prompt = prompt_template.format(personalized_message=personalized_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)
         
         # For GigaChat-Lite
-        prompt = prompt_template.format(personalized_message=personalized_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)
         
         # For GigaChat-Plus
-        prompt = prompt_template.format(personalized_message=personalized_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)
         
         # For GPT-4o
-        prompt = prompt_template.format(personalized_message=personalized_gpt4o)
+        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)
         
         # For Meta-Llama-3.1-405B
-        prompt = prompt_template.format(personalized_message=personalized_meta_llama_405b)
-        adapted_meta_llama_405b = generate_message_meta_llama_3_1_405b_with_retry(prompt)
+        prompt = prompt_template.format(
+            reference_message=reference_message,
+            personalized_message=personalized_meta_llama_405b
+        )
+        adapted_meta_llama_405b = generate_message_meta_llama_405b_with_retry(prompt)
         adapted_messages.append(adapted_meta_llama_405b)
         
         return adapted_messages