Update app.py

app.py

@@ -109,6 +109,7 @@ def generate_message_gigachat_lite(prompt):
 
 def generate_message_gigachat_plus(prompt):
     try:
+        time.sleep(2)
         messages = [SystemMessage(content=prompt)]
         res = chat_plus(messages)
         return res.content.strip()
@@ -155,6 +156,11 @@ def generate_personalization_prompt(*selected_values):
                 return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
     return prompt.strip()
 
+# Функция для выполнения персонализации на основе сгенерированного промпта и сообщения
+def perform_personalization(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_gpt4o(full_prompt)
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -210,9 +216,9 @@ with gr.Blocks() as demo:
         
     with gr.Row():
         personalize_btn.click(
-            generate_personalization_prompt,
-            inputs=selections,
-            outputs=[gr.Textbox(label="Промпт для персонализации", lines=6)]
+            lambda standard_message, *selected_values: perform_personalization(standard_message, generate_personalization_prompt(*selected_values)),
+            inputs=[output_text_gpt4o] + selections,
+            outputs=[gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6)]
         )
 
     demo.launch()