Update app.py

app.py

@@ -109,7 +109,7 @@ def format_instruction_string(instr):
 
 def generate_display_prompts(description, product_name, benefits, key_message, chosen_approach,
                              gender, generation, psychotype, business_stage, industry, opf):
-    # Формируем промпты для отображения (с объединением всех подходов)
+    # Формируем промпты для отображения
     if chosen_approach == "Подход не найден для выбранных параметров.":
         return ("Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода",
                 "Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода")
@@ -151,10 +151,12 @@ def generate_display_prompts(description, product_name, benefits, key_message, c
             lines.append(f"{i}. {formatted_line}.")
         mandatory_terms = "\n".join(lines)
     else:
-        mandatory_terms = "Нет обязательных терминов, так как не выбрано ни одного параметра"
+        # Нет персональных параметров
+        return ("Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода",
+                "Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода")
 
-    if chosen_params_instructions:
-        prompt = f"""Напиши три или четыре предложения. {approach_prefix}. 
+    # Если есть персональные параметры
+    prompt = f"""Напиши три или четыре предложения. {approach_prefix}. 
 Напиши рекламное SMS для следующего продукта:
 «{description}».
 Не изменяй название продукта: «{product_name}».
@@ -166,16 +168,6 @@ def generate_display_prompts(description, product_name, benefits, key_message, c
 {approach_suffix}.
 Убедись, что УМЕСТНО использовал КАЖДЫЙ необходимый термин.
 Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
-Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
-    else:
-        prompt = f"""Напиши три или четыре предложения. {approach_prefix}. Напиши рекламное SMS для следующего продукта:
-«{description}».
-Не изменяй название продукта: «{product_name}».
-Преимущества:
-«{benefits}».
-Убедись, что написал не меньше трех и не больше четырех предложений.
-{approach_suffix}.
-Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
 Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
 
     return prompt, prompt
@@ -183,7 +175,7 @@ def generate_display_prompts(description, product_name, benefits, key_message, c
 def generate_model_prompt(description, product_name, benefits, key_message,
                           gender, generation, psychotype, business_stage, industry, opf,
                           single_approach):
-    # Генерируем промпт для обращения к модели, используя только один выбранный подход (single_approach)
+    # Генерируем промпт для обращения к модели, используя только один выбранный подход
     prefix = approach_dict[single_approach]["prefix"]
     suffix = approach_dict[single_approach]["suffix"]
 
@@ -209,6 +201,7 @@ def generate_model_prompt(description, product_name, benefits, key_message,
             lines.append(f"{i}. {formatted_line}.")
         mandatory_terms = "\n".join(lines)
     else:
+        # Не должно сюда попасться, т.к. мы проверим до вызова модели
         mandatory_terms = None
 
     if mandatory_terms:
@@ -226,15 +219,8 @@ def generate_model_prompt(description, product_name, benefits, key_message,
 Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
 Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
     else:
-        model_prompt = f"""Напиши три или четыре предложения. {prefix}. Напиши рекламное SMS для следующего продукта:
-«{description}».
-Не изменяй название продукта: «{product_name}».
-Преимущества:
-«{benefits}».
-Убедись, что написал не меньше трех и не больше четырех предложений.
-{suffix}.
-Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
-Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
+        # Теоретически сюда не попадём, если проверяем до вызова
+        model_prompt = ""
 
     return model_prompt
 
@@ -248,23 +234,29 @@ def call_model(model_prompt):
     )
     return completion.choices[0].message.content.strip()
 
-def update_all_params(description, product_name, benefits, key_message, 
-                      gender, generation, psychotype, business_stage, industry, opf):
-    # Сначала определяем подход
+def update_prompts_on_params_change(description, product_name, benefits, key_message, 
+                                    gender, generation, psychotype, business_stage, industry, opf):
+    # Обновляем подход и промпты при изменении параметров клиента
     chosen_approach = get_approaches(gender, generation, psychotype, approaches_data)
-    # Затем формируем промпты для отображения
     prompt_1, prompt_2 = generate_display_prompts(description, product_name, benefits, key_message,
                                                   chosen_approach, gender, generation, psychotype,
                                                   business_stage, industry, opf)
+    # Здесь SMS не генерируем
+    return chosen_approach, prompt_1, prompt_2
+
+def generate_personalized_sms(description, product_name, benefits, key_message,
+                              gender, generation, psychotype, business_stage, industry, opf,
+                              chosen_approach, prompt_1, prompt_2):
+    # Проверяем, есть ли персональный параметр
+    if "Для формирования промпта выберите хотя бы один личный персональный параметр" in prompt_1:
+        # Выводим предупреждение
+        return gr.update(value="Задайте хотя бы один личный персональный параметр для определения подхода"), "", ""
+    if chosen_approach == "Подход не найден для выбранных параметров.":
+        return gr.update(value="Задайте хотя бы один личный персональный параметр для определения подхода"), "", ""
 
-    # Если подход не найден или нет персональных параметров – показываем тексты как есть без генерации SMS
-    if chosen_approach == "Подход не найден для выбранных параметров." or "Для формирования промпта выберите" in prompt_1:
-        sms_1 = ""
-        sms_2 = ""
-        return chosen_approach, prompt_1, prompt_2, sms_1, sms_2
-
-    # Если подход есть
     approach_list = [a.strip() for a in chosen_approach.split(',') if a.strip()]
+    if not approach_list:
+        return gr.update(value="Задайте хотя бы один личный персональный параметр для определения подхода"), "", ""
 
     # Выбираем случайный подход для model_prompt_1
     if len(approach_list) > 1:
@@ -281,16 +273,15 @@ def update_all_params(description, product_name, benefits, key_message,
     model_prompt_1 = generate_model_prompt(description, product_name, benefits, key_message,
                                            gender, generation, psychotype, business_stage, industry, opf,
                                            chosen_single_approach_1)
-
     model_prompt_2 = generate_model_prompt(description, product_name, benefits, key_message,
                                            gender, generation, psychotype, business_stage, industry, opf,
                                            chosen_single_approach_2)
 
-    # Вызываем модель для получения SMS
     sms_1 = call_model(model_prompt_1)
     sms_2 = call_model(model_prompt_2)
 
-    return chosen_approach, prompt_1, prompt_2, sms_1, sms_2
+    # Очистим warning (presence_in_db) если было
+    return gr.update(value=""), sms_1, sms_2
 
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
@@ -327,10 +318,11 @@ with gr.Blocks(theme="default") as demo:
             industry_dropdown = gr.Dropdown(label="Отрасль", choices=["Не выбрано"]+industries, value=None)
             opf_dropdown = gr.Dropdown(label="ОПФ", choices=["Не выбрано"]+opfs, value=None)
             chosen_approach = gr.Textbox(label="Выбранный подход", lines=1, value="")
-            presence_in_db = gr.Textbox(label="Наличие в базе", lines=1, value="", interactive=False)
+            presence_in_db = gr.Textbox(label="Сообщение", lines=1, value="", interactive=False)
 
     with gr.Row():
         return_params_btn = gr.Button("Вернуть параметры предыдущего запроса")
+        set_unused_params_btn = gr.Button("Задать ранее невыставленные параметры")
         create_personal_sms_btn = gr.Button("Создать персонализированное SMS")
     
     with gr.Row():
@@ -362,15 +354,25 @@ with gr.Blocks(theme="default") as demo:
         save_sms_1_btn = gr.Button("Сохранить в базу")
         save_sms_2_btn = gr.Button("Сохранить в базу")
 
-    # Обновляем промпты, подход и генерируем SMS при изменении параметров клиента
+    # Обновляем промпты и подход при изменении параметров клиента (без генерации SMS)
     client_params = [gender_dropdown, generation_dropdown, psychotype_dropdown, business_stage_dropdown, industry_dropdown, opf_dropdown]
     for cp in client_params:
         cp.change(
-            fn=update_all_params,
+            fn=update_prompts_on_params_change,
             inputs=[description, product_name, benefits, key_message, 
                     gender_dropdown, generation_dropdown, psychotype_dropdown,
                     business_stage_dropdown, industry_dropdown, opf_dropdown],
-            outputs=[chosen_approach, prompt_1, prompt_2, sms_1, sms_2]
+            outputs=[chosen_approach, prompt_1, prompt_2]
         )
 
+    # Генерация SMS по нажатию кнопки
+    create_personal_sms_btn.click(
+        fn=generate_personalized_sms,
+        inputs=[description, product_name, benefits, key_message, 
+                gender_dropdown, generation_dropdown, psychotype_dropdown,
+                business_stage_dropdown, industry_dropdown, opf_dropdown,
+                chosen_approach, prompt_1, prompt_2],
+        outputs=[presence_in_db, sms_1, sms_2]
+    )
+
     demo.launch()