Update app.py

app.py

@@ -173,11 +173,9 @@ def generate_message_gigachat_plus(prompt):
         return f"Ошибка при обращении к GigaChat-Lite+: {e}"
 
 # Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
-def generate_messages(*selected_values):
-    prompt = generate_prompt(*selected_values)
-    if "Ошибка" in prompt:
-        yield prompt, "", "", "", "", "", "", "", "", "", "", "", ""
-        return
+def generate_messages(description, advantages, *selected_values):
+    # Генерация стандартного промпта с использованием вашей функции
+    prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     # Генерация стандартных сообщений от разных моделей
     message_gpt4o = generate_message_gpt4o(prompt)
@@ -220,12 +218,42 @@ def generate_messages(*selected_values):
     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:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
     
     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}"))
@@ -253,7 +281,7 @@ with gr.Blocks() as demo:
 
             submit_btn.click(
                 generate_messages,
-                inputs=selections,
+                inputs=[description_input, advantages_input] + selections,
                 outputs=[
                     prompt_display,
                     output_text_gpt4o,