Update app.py

app.py

@@ -72,6 +72,60 @@ def generate_standard_prompt(description, advantages, *selected_values):
     )
     return prompt
 
+# Функции для генерации сообщений с персонализацией
+def create_personalization_prompt(message, *selected_values):
+    prompt = f"Пожалуйста, адаптируйте данное сообщение: '{message}' в соответствии со следующими требованиями:\n"
+    for i, feature in enumerate(features.keys()):
+        if selected_values[i]:
+            try:
+                prompt += f"- {features[feature][selected_values[i]]}\n"
+            except KeyError:
+                return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
+    return prompt.strip()
+
+def generate_personalized_message_gpt4o(message, personalization_prompt):
+    try:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {openai_api_key}"
+        }
+        data = {
+            "model": "chatgpt-4o-latest",
+            "messages": [{"role": "system", "content": f"{personalization_prompt}"}],
+            "max_tokens": 700
+        }
+        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
+        response_data = response.json()
+        return response_data["choices"][0]["message"]["content"].strip()
+    except Exception as e:
+        return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
+
+def generate_personalized_message_gigachat_pro(message, personalization_prompt):
+    try:
+        messages = [SystemMessage(content=f"{personalization_prompt}")]
+        res = chat_pro(messages)
+        return res.content.strip()
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+
+def generate_personalized_message_gigachat_lite(message, personalization_prompt):
+    try:
+        time.sleep(2)
+        messages = [SystemMessage(content=f"{personalization_prompt}")]
+        res = chat_lite(messages)
+        return res.content.strip()
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Lite: {e}"
+
+def generate_personalized_message_gigachat_plus(message, personalization_prompt):
+    try:
+        messages = [SystemMessage(content=f"{personalization_prompt}")]
+        res = chat_plus(messages)
+        return res.content.strip()
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Lite+: {e}"
+
+
 # Функция для генерации сообщения с помощью ChatGPT-4o-Latest
 def generate_message_gpt4o(prompt):
     try:
@@ -119,30 +173,52 @@ def generate_message_gigachat_plus(prompt):
         return f"Ошибка при обращении к GigaChat-Lite+: {e}"
 
 # Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
-def generate_messages(description, advantages, *selected_values):
-    standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
+def generate_messages(*selected_values):
+    prompt = generate_prompt(*selected_values)
+    if "Ошибка" в prompt:
+        yield prompt, "", "", "", "", "", "", "", "", "", "", "", ""
+        return
+
+    # Генерация стандартных сообщений от разных моделей
+    message_gpt4o = generate_message_gpt4o(prompt)
+    yield prompt, message_gpt4o, "", "", "", "", "", "", "", "", "", "", ""
+
+    message_gigachat_pro = generate_message_gigachat_pro(prompt)
+    yield prompt, message_gpt4o, message_gigachat_pro, "", "", "", "", "", "", "", "", "", ""
+
+    message_gigachat_lite = generate_message_gigachat_lite(prompt)
+    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, "", "", "", "", "", "", "", ""
+
+    message_gigachat_plus = generate_message_gigachat_plus(prompt)
+    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, "", "", "", "", "", "", "", ""
 
-    results = {
-        "prompt": standard_prompt,
-        "gpt4o": None,
-        "gigachat_pro": None,
-        "gigachat_lite": None,
-        "gigachat_plus": None
-    }
+    # Создание персонализированных промптов на основе сгенерированных сообщений и выбранных признаков
+    personalization_prompt_gpt4o = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gpt4o}' под следующие требования:\n"
+    personalization_prompt_gigachat_pro = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gigachat_pro}' под следующие требования:\n"
+    personalization_prompt_gigachat_lite = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gigachat_lite}' под следующие требования:\n"
+    personalization_prompt_gigachat_plus = f"Пожалуйста, адаптируйте следующее сообщение: '{message_gigachat_plus}' под следующие требования:\n"
 
-    yield results["prompt"], "", "", "", ""
+    # Добавляем требования, выбранные пользователем
+    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
 
-    results["gpt4o"] = generate_message_gpt4o(standard_prompt)
-    yield results["prompt"], results["gpt4o"], "", "", ""
+    # Генерация и отображение персонализированных сообщений
+    personalized_message_gpt4o = generate_personalized_message_gpt4o(message_gpt4o, personalization_prompt_gpt4o)
+    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, "", "", "", personalization_prompt_gpt4o, "", "", ""
 
-    results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", ""
+    personalized_message_gigachat_pro = generate_personalized_message_gigachat_pro(message_gigachat_pro, personalization_prompt_gigachat_pro)
+    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, personalized_message_gigachat_pro, "", "", "", personalization_prompt_gigachat_pro, "", ""
 
-    results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], ""
+    personalized_message_gigachat_lite = generate_personalized_message_gigachat_lite(message_gigachat_lite, personalization_prompt_gigachat_lite)
+    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, "", "", "", personalization_prompt_gigachat_lite, ""
 
-    results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"]
+    personalized_message_gigachat_plus = generate_personalized_message_gigachat_plus(message_gigachat_plus, personalization_prompt_gigachat_plus)
+    yield prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite, message_gigachat_plus, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "", "", "", personalization_prompt_gigachat_plus
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -150,30 +226,6 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column(scale=1):
-            description_input = gr.Textbox(
-                label="Описание предложения (предзаполненный пример можно поменять на свой)", 
-                lines=6,
-                value=(
-                    "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
-                    "Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
-                    "Что необходимо сделать, чтобы воспользоваться предложением:\n"
-                    "1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
-                    "2. Забрать карту.\n"
-                    "3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
-                    "4. В течение следующего месяца пользоваться ей бесплатно."
-                )
-            )
-            advantages_input = gr.Textbox(
-                label="Преимущества (предзаполненный пример можно поменять на свой)", 
-                lines=4,
-                value=(
-                    "Предложение по бесплатному обслуживанию — бессрочное.\n"
-                    "Оплата покупок без отчётов и платёжных поручений.\n"
-                    "Платёжные документы без комиссии.\n"
-                    "Лимиты на расходы сотрудников.\n"
-                    "Мгновенные переводы на карты любых банков."
-                )
-            )
             selections = []
             for feature in features.keys():
                 selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
@@ -181,16 +233,42 @@ with gr.Blocks() as demo:
             submit_btn = gr.Button("Submit")  # Оранжевая кнопка по умолчанию
 
         with gr.Column(scale=2):
-            prompt_display = gr.Textbox(label="Стандартный промпт", lines=15, interactive=False)
+            prompt_display = gr.Textbox(label="Сгенерированный промпт", lines=4, interactive=False)
             output_text_gpt4o = gr.Textbox(label="Создание сообщения GPT-4o", lines=3)
             output_text_gigachat_pro = gr.Textbox(label="Создание сообщения GigaChat-Pro", lines=3)
             output_text_gigachat_lite = gr.Textbox(label="Создание сообщения GigaChat-Lite", lines=3)
             output_text_gigachat_plus = gr.Textbox(label="Создание сообщения GigaChat-Lite+", lines=3)
-            
+
+            # Добавляем поля для персонализированных сообщений
+            personalized_output_text_gpt4o = gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=3)
+            personalized_output_text_gigachat_pro = gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=3)
+            personalized_output_text_gigachat_lite = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=3)
+            personalized_output_text_gigachat_plus = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=3)
+
+            # Добавляем поля для персонализированных промптов
+            personalization_prompt_gpt4o_display = gr.Textbox(label="Промпт для персонализации GPT-4o", lines=6, interactive=False)
+            personalization_prompt_gigachat_pro_display = gr.Textbox(label="Промпт для персонализации GigaChat-Pro", lines=6, interactive=False)
+            personalization_prompt_gigachat_lite_display = gr.Textbox(label="Промпт для персонализации GigaChat-Lite", lines=6, interactive=False)
+            personalization_prompt_gigachat_plus_display = gr.Textbox(label="Промпт для персонализации GigaChat-Lite+", lines=6, interactive=False)
+
             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]
+                generate_messages,
+                inputs=selections,
+                outputs=[
+                    prompt_display,
+                    output_text_gpt4o, 
+                    output_text_gigachat_pro, 
+                    output_text_gigachat_lite, 
+                    output_text_gigachat_plus,
+                    personalized_output_text_gpt4o,
+                    personalized_output_text_gigachat_pro,
+                    personalized_output_text_gigachat_lite,
+                    personalized_output_text_gigachat_plus,
+                    personalization_prompt_gpt4o_display,
+                    personalization_prompt_gigachat_pro_display,
+                    personalization_prompt_gigachat_lite_display,
+                    personalization_prompt_gigachat_plus_display
+                ]
             )
 
     demo.launch()