Update app.py

app.py

@@ -60,7 +60,6 @@ def get_approaches(gender, generation, psychotype, approaches_df):
 
     filters = []
     for param_name, param_value in [('Пол', gender), ('Поколение', generation), ('Психотип', psychotype)]:
-        # Если значение параметра "Не выбрано" или пустое
         if not param_value or param_value == "Не выбрано":
             filters.append(approaches_df[param_name].isnull() | (approaches_df[param_name].fillna('') == ''))
         else:
@@ -84,11 +83,7 @@ def get_approaches(gender, generation, psychotype, approaches_df):
     selected_approaches_text_content = ', '.join(approach_list)
     return selected_approaches_text_content
 
-def update_approach(gender, generation, psychotype):
-    return get_approaches(gender, generation, psychotype, approaches_data)
-
 def get_instructions_for_param(param_value, df, col):
-    # Если param_value пустое или "Не выбрано" – игнорируем
     if not param_value or param_value == "Не выбрано":
         return None
     row = df[df[col] == param_value]
@@ -100,7 +95,6 @@ def get_instructions_for_param(param_value, df, col):
     return instr1
 
 def format_instruction_string(instr):
-    # "слово1, слово2, слово3" -> "слово1 или слово2 или слово3"
     terms = [t.strip() for t in instr.split(',') if t.strip()]
     return " или ".join(terms) if terms else ""
 
@@ -111,7 +105,6 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
         return ("Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода",
                 "Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода")
 
-    # Собираем подходы
     approach_list = [a.strip() for a in chosen_approach.split(',') if a.strip()]
     prefix_parts = []
     suffix_parts = []
@@ -119,7 +112,7 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
         if a in approach_dict:
             prefix_parts.append(approach_dict[a]["prefix"])
             suffix_parts.append(approach_dict[a]["suffix"])
-    # Если несколько подходов, соединяем через "/"
+
     if len(prefix_parts) > 1:
         approach_prefix = " / ".join(prefix_parts)
         approach_suffix = " / ".join(suffix_parts)
@@ -127,7 +120,6 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
         approach_prefix = prefix_parts[0] if prefix_parts else ""
         approach_suffix = suffix_parts[0] if suffix_parts else ""
 
-    # Собираем выбранные параметры
     instructions_data = [
         (gender, genders_data, "Пол"),
         (generation, generations_data, "Поколение"),
@@ -143,13 +135,6 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
         if instr1:
             chosen_params_instructions.append(instr1)
 
-    # Теперь, согласно новой логике: 
-    # - Если один параметр выбран -> 1 пункт
-    # - Если два параметра -> 2 пункта (по одному с каждого параметра)
-    # И так далее. Используем только Инструкция 1.
-    # Таким образом, количество пунктов = количество выбранных параметров
-    # каждый пункт — это Инструкция 1 для параметра.
-    
     if chosen_params_instructions:
         lines = []
         for i, instr_line in enumerate(chosen_params_instructions, start=1):
@@ -157,10 +142,8 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
             lines.append(f"{i}. {formatted_line}")
         mandatory_terms = "\n".join(lines)
     else:
-        # Если нет выбранных параметров с инструкциями
         mandatory_terms = "Нет обязательных терминов, так как не выбрано ни одного параметра"
 
-    # Формируем итоговый промпт
     if chosen_params_instructions:
         prompt = f"""Напиши три или четыре предложения. {approach_prefix}. Напиши рекламное SMS для следующего продукта:
 «{description}».
@@ -175,8 +158,7 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
 Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
 Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
     else:
-        # Если нет ни одного параметра, но подход найден,
-        # просто не указываем обязательные термины.
+        # Нет персональных параметров — нет обязательных терминов
         prompt = f"""Напиши три или четыре предложения. {approach_prefix}. Напиши рекламное SMS для следующего продукта:
 «{description}».
 Не изменяй название продукта: «{product_name}».
@@ -189,12 +171,15 @@ def generate_prompts(description, product_name, benefits, key_message, chosen_ap
 
     return prompt, prompt
 
-def create_personalized_sms(description, product_name, benefits, key_message, chosen_approach,
-                            gender, generation, psychotype, business_stage, industry, opf):
+def update_all_params(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_prompts(description, product_name, benefits, key_message,
-                                           chosen_approach, gender, generation, psychotype,
-                                           business_stage, industry, opf)
-    return prompt_1, prompt_2
+                                          chosen_approach, gender, generation, psychotype,
+                                          business_stage, industry, opf)
+    return chosen_approach, prompt_1, prompt_2
 
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
@@ -232,26 +217,10 @@ with gr.Blocks(theme="default") as demo:
             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)
-            
-            # При изменении Пол, Поколение, Психотип вычисляем подход
-            def on_personal_params_change(gender, generation, psychotype):
-                return update_approach(gender, generation, psychotype)
-            
-            gender_dropdown.change(fn=on_personal_params_change, 
-                                   inputs=[gender_dropdown, generation_dropdown, psychotype_dropdown], 
-                                   outputs=[chosen_approach])
-            generation_dropdown.change(fn=on_personal_params_change, 
-                                       inputs=[gender_dropdown, generation_dropdown, psychotype_dropdown], 
-                                       outputs=[chosen_approach])
-            psychotype_dropdown.change(fn=on_personal_params_change, 
-                                       inputs=[gender_dropdown, generation_dropdown, psychotype_dropdown], 
-                                       outputs=[chosen_approach])
-    
+
     with gr.Row():
-        # Убрали кнопку "Создать персонализированное SMS" для формирования промптов
-        # Теперь промпты формируются при изменении параметров
         return_params_btn = gr.Button("Вернуть параметры предыдущего запроса")
-        create_personal_sms_btn = gr.Button("Создать персонализированное SMS")  # Можно оставить для другого функционала, но промпты будут обновляться без нее.
+        create_personal_sms_btn = gr.Button("Создать персонализированное SMS")
     
     with gr.Row():
         with gr.Column():
@@ -282,70 +251,15 @@ with gr.Blocks(theme="default") as demo:
         save_sms_1_btn = gr.Button("Сохранить в базу")
         save_sms_2_btn = gr.Button("Сохранить в базу")
 
-    # Теперь обновляем промпты при любом изменении параметров клиента и продукта
-    # Функция-обертка, чтобы получить текущие значения и сгенерировать промпты
-    def update_prompts_on_change(description, product_name, benefits, key_message, chosen_approach,
-                                 gender, generation, psychotype, business_stage, industry, opf):
-        return create_personalized_sms(description, product_name, benefits, key_message, chosen_approach,
-                                       gender, generation, psychotype, business_stage, industry, opf)
-
-    # Добавляем события при изменении всех параметров
-    # Продукт
-    product_dropdown.change(
-        fn=update_prompts_on_change,
-        inputs=[description, product_name, benefits, key_message, chosen_approach,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[prompt_1, prompt_2]
-    )
-
-    # Параметры клиента
-    gender_dropdown.change(
-        fn=update_prompts_on_change,
-        inputs=[description, product_name, benefits, key_message, chosen_approach,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[prompt_1, prompt_2]
-    )
-
-    generation_dropdown.change(
-        fn=update_prompts_on_change,
-        inputs=[description, product_name, benefits, key_message, chosen_approach,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[prompt_1, prompt_2]
-    )
-
-    psychotype_dropdown.change(
-        fn=update_prompts_on_change,
-        inputs=[description, product_name, benefits, key_message, chosen_approach,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[prompt_1, prompt_2]
-    )
-
-    business_stage_dropdown.change(
-        fn=update_prompts_on_change,
-        inputs=[description, product_name, benefits, key_message, chosen_approach,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[prompt_1, prompt_2]
-    )
-
-    industry_dropdown.change(
-        fn=update_prompts_on_change,
-        inputs=[description, product_name, benefits, key_message, chosen_approach,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[prompt_1, prompt_2]
-    )
-
-    opf_dropdown.change(
-        fn=update_prompts_on_change,
-        inputs=[description, product_name, benefits, key_message, chosen_approach,
-                gender_dropdown, generation_dropdown, psychotype_dropdown,
-                business_stage_dropdown, industry_dropdown, opf_dropdown],
-        outputs=[prompt_1, prompt_2]
-    )
+    # Обновляем промпты и подход при изменении параметров клиента
+    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,
+            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]
+        )
 
     demo.launch()