Update app.py

app.py

@@ -72,7 +72,7 @@ def generate_standard_prompt(description, advantages, *selected_values):
     )
     return prompt
 
-# Функция для генерации сообщения с помощью ChatGPT-4o-Latest
+# Функции для генерации сообщений
 def generate_message_gpt4o(prompt):
     try:
         headers = {
@@ -90,7 +90,6 @@ def generate_message_gpt4o(prompt):
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
-# Функция для генерации сообщения с помощью GigaChat-Pro
 def generate_message_gigachat_pro(prompt):
     try:
         messages = [SystemMessage(content=prompt)]
@@ -99,17 +98,15 @@ def generate_message_gigachat_pro(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-# Функция для генерации сообщения с помощью GigaChat-Lite
 def generate_message_gigachat_lite(prompt):
     try:
-        time.sleep(2)  # Задержка 2 секунды перед запросом к GigaChat
+        time.sleep(2)
         messages = [SystemMessage(content=prompt)]
         res = chat_lite(messages)
         return res.content.strip()
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite: {e}"
 
-# Функция для генерации сообщения с помощью GigaChat-Lite+
 def generate_message_gigachat_plus(prompt):
     try:
         messages = [SystemMessage(content=prompt)]
@@ -144,6 +141,34 @@ def generate_messages(description, advantages, *selected_values):
     results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
     yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"]
 
+    # Сохраняем результаты для дальнейшей персонализации
+    return results
+
+# Функция для выполнения персонализации
+def personalize_messages(results, *selected_values):
+    # Промпты для персонализации
+    personalization_prompt_gpt4o = f"Пожалуйста, адаптируйте следующее сообщение: '{results['gpt4o']}' под следующие требования:\n"
+    personalization_prompt_gigachat_pro = f"Пожалуйста, адаптируйте следующее сообщение: '{results['gigachat_pro']}' под следующие требования:\n"
+    personalization_prompt_gigachat_lite = f"Пожалуйста, адаптируйте следующее сообщение: '{results['gigachat_lite']}' под следующие требования:\n"
+    personalization_prompt_gigachat_plus = f"Пожалуйста, адаптируйте следующее сообщение: '{results['gigachat_plus']}' под следующие требования:\n"
+    
+    # Добавляем требования из выбранных признаков
+    for i, feature in enumerate(features.keys()):
+        if selected_values[i]:
+            requirement = f"- {features[feature][selected_values[i]]}\n"
+            personalization_prompt_gpt4o += requirement
+            personalization_prompt_gigachat_pro += requirement
+            personalization_prompt_gigachat_lite += requirement
+            personalization_prompt_gigachat_plus += requirement
+    
+    # Персонализированные сообщения
+    personalized_message_gpt4o = generate_message_gpt4o(personalization_prompt_gpt4o)
+    personalized_message_gigachat_pro = generate_message_gigachat_pro(personalization_prompt_gigachat_pro)
+    personalized_message_gigachat_lite = generate_message_gigachat_lite(personalization_prompt_gigachat_lite)
+    personalized_message_gigachat_plus = generate_message_gigachat_plus(personalization_prompt_gigachat_plus)
+    
+    return personalization_prompt_gpt4o, personalized_message_gpt4o, personalization_prompt_gigachat_pro, personalized_message_gigachat_pro, personalization_prompt_gigachat_lite, personalized_message_gigachat_lite, personalization_prompt_gigachat_plus, personalized_message_gigachat_plus
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -187,10 +212,34 @@ with gr.Blocks() as demo:
             output_text_gigachat_lite = gr.Textbox(label="Создание сообщения GigaChat-Lite", lines=3)
             output_text_gigachat_plus = gr.Textbox(label="Создание сообщения GigaChat-Lite+", lines=3)
             
+            # Кнопка для генерации стандартных сообщений
             submit_btn.click(
                 generate_messages, 
                 inputs=[description_input, advantages_input] + selections, 
                 outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
             )
 
+            # Поля для вывода персонализированных сообщений
+            personalization_prompt_gpt4o_display = gr.Textbox(label="Промпт для персонализации GPT-4o", lines=6, interactive=False)
+            personalized_output_text_gpt4o = gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=3)
+            personalization_prompt_gigachat_pro_display = gr.Textbox(label="Промпт для персонализации GigaChat-Pro", lines=6, interactive=False)
+            personalized_output_text_gigachat_pro = gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=3)
+            personalization_prompt_gigachat_lite_display = gr.Textbox(label="Промпт для персонализации GigaChat-Lite", lines=6, interactive=False)
+            personalized_output_text_gigachat_lite = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=3)
+            personalization_prompt_gigachat_plus_display = gr.Textbox(label="Промпт для персонализации GigaChat-Lite+", lines=6, interactive=False)
+            personalized_output_text_gigachat_plus = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=3)
+
+            # Кнопка для выполнения персонализации
+            personalize_btn = gr.Button("Выполнить персонализацию")
+            personalize_btn.click(
+                personalize_messages,
+                inputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus] + selections,
+                outputs=[
+                    personalization_prompt_gpt4o_display, personalized_output_text_gpt4o,
+                    personalization_prompt_gigachat_pro_display, personalized_output_text_gigachat_pro,
+                    personalization_prompt_gigachat_lite_display, personalized_output_text_gigachat_lite,
+                    personalization_prompt_gigachat_plus_display, personalized_output_text_gigachat_plus
+                ]
+            )
+
     demo.launch()