Update app.py

app.py

@@ -23,6 +23,105 @@ for sheet_name, df in data.items():
         print(f"Ошибка при обработке данных листа {sheet_name}: {e}")
         features[sheet_name] = {}
 
+# Функция для генерации стандартного промпта
+def generate_standard_prompt(description, advantages, key_message):
+    prompt = (
+        f"Сгенерируй смс-сообщение для клиента.\n"
+        f"Описание предложения: {description}\n"
+        f"Преимущества: {advantages}\n"
+        "В тексте смс запрещено использование:\n"
+        "- Запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, доступный, банкротство, долги, займ, срочно, сейчас, лучший, главный, номер 1, гарантия, успех, лидер;\n"
+        "- Обращение к клиенту;\n"
+        "- Приветствие клиента;\n"
+        "- Обещания и гарантии;\n"
+        "- Использовать составные конструкции из двух глаголов;\n"
+        "- Причастия и причастные обороты;\n"
+        "- Деепричастия и деепричастные обороты;\n"
+        "- Превосходная степень прилагательных;\n"
+        "- Страдательный залог;\n"
+        "- Порядковые числительные от 10 прописью;\n"
+        "- Цепочки с придаточными предложениями;\n"
+        "- Разделительные повторяющиеся союзы;\n"
+        "- Вводные конструкции;\n"
+        "- Усилители;\n"
+        "- Паразиты времени;\n"
+        "- Несколько существительных подряд, в том числе отглагольных;\n"
+        "- Производные предлоги;\n"
+        "- Сложные предложения, в которых нет связи между частями;\n"
+        "- Сложноподчинённые предложения;\n"
+        "- Даты прописью;\n"
+        "- Близкие по смыслу однородные члены предложения;\n"
+        "- Шокирующие, экстравагантные, кликбейтные фразы;\n"
+        "- Абстрактные заявления без поддержки фактами и отсутствие доказательства пользы для клиента;\n"
+        "- Гарантирующие фразы;\n"
+        "- Узкоспециализированные термины;\n"
+        "- Фразы, способные создать двойственное ощущение, обидеть;\n"
+        "- Речевые клише, рекламные штампы, канцеляризмы;\n"
+        "Убедись, что в готовом тексте до 250 знаков с пробелами.\n"
+    )
+    if key_message.strip():
+        prompt += f"Убедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
+    return prompt.strip()
+
+
+# Функция для генерации персонализированного промпта
+def generate_personalization_prompt(key_message, *selected_values):
+    prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
+    gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
+    combined_instruction = ""
+    additional_instructions = ""
+
+    print(f"Выбранные значения: Пол={gender}, Поколение={generation}, Психотип={psychotype}")
+
+    # Проверяем, выбраны ли все три параметра: Пол, Поколение, Психотип
+    if gender and generation and psychotype:
+        # Получаем данные с листа "Пол Поколение Психотип"
+        sheet = features.get("Пол Поколение Психотип", {})
+
+        # Ищем ключ, соответствующий комбинации "Пол", "Поколение", "Психотип"
+        key = (gender, generation, psychotype)
+        if key in sheet:
+            combined_instruction = sheet[key]
+            print(f"Найдена комбинированная инструкция: {combined_instruction}")
+        else:
+            print(f"Комбинированная инструкция для ключа {key} не найдена.")
+
+    # Если не найдена комбинированная инструкция, добавляем индивидуальные инструкции
+    if not combined_instruction:
+        print("Добавляем индивидуальные инструкции для Пол, Поколение, Психотип.")
+        for i, feature in enumerate(["Пол", "Поколение", "Психотип"]):
+            if selected_values[i]:
+                try:
+                    instruction = features[feature][selected_values[i]]
+                    additional_instructions += f"{instruction}\n"
+                    print(f"Добавлена инструкция из {feature}: {instruction}")
+                except KeyError:
+                    return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
+
+    # Добавляем инструкции для остальных параметров (например, Отрасль)
+    for i, feature in enumerate(features.keys()):
+        if feature not in ["Пол", "Поколение", "Психотип", "Пол Поколение Психотип"]:
+            if i < len(selected_values) and selected_values[i]:
+                try:
+                    instruction = features[feature][selected_values[i]]
+                    additional_instructions += f"{instruction}\n"
+                    print(f"Добавлена инструкция из {feature}: {instruction}")
+                except KeyError:
+                    return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
+
+    # Формируем итоговый промпт
+    if combined_instruction:
+        prompt += combined_instruction  # Добавляем комбинированную инструкцию, если она есть
+    if additional_instructions:
+        prompt += additional_instructions  # Добавляем остальные инструкции
+
+    prompt += "Убедись, что в готовом тексте до 250 знаков с пробелами.\n"
+
+    prompt += f"Убедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
+    
+    return prompt.strip()
+
+
 # Функция для создания спидометра
 def create_gauge(value):
     fig = go.Figure(go.Indicator(
@@ -108,6 +207,21 @@ with gr.Blocks() as demo:
 
             btn_to_prompts = gr.Button("Создать")
             btn_to_prompts.click(fn=change_tab, inputs=[gr.Number(value=1, visible=False)], outputs=tabs)
+            
+            # Добавляем генерацию стандартного промпта при клике
+            btn_to_prompts.click(
+                fn=generate_standard_prompt,
+                inputs=[desc, benefits, key_message],
+                outputs=[non_personalized_prompt]
+            )
+            
+            # Добавляем генерацию персонализированного промпта при клике
+            btn_to_prompts.click(
+                fn=generate_personalization_prompt,
+                inputs=[key_message, gender, generation, psychotype, business_stage, industry, opf],
+                outputs=[personalized_prompt]
+            )
+
         
         # Вкладка 2: Промпты
         with gr.TabItem("Ассистент", id=1):