Update app.py

app.py

@@ -1,3 +1,4 @@
+import re  # Добавляем импорт re для возможного редактирования текста
 import gradio as gr
 import os
 import pandas as pd
@@ -31,70 +32,124 @@ for sheet_name, df in data.items():
         print(f"Ошибка при обработке данных листа {sheet_name}: {e}")
         features[sheet_name] = {}
 
-# Функция для создания спидометра
-def create_gauge(value):
-    fig = go.Figure(go.Indicator(
-        mode="gauge+number",
-        value=value,
-        gauge={
-            'axis': {'range': [0, 100]},
-            'bar': {'color': "black"},  # Цвет стрелки
-            'steps': [
-                {'range': [0, 40], 'color': "#55efc4"},  # Мягкий зеленый
-                {'range': [40, 70], 'color': "#ffeaa7"},  # Желтый
-                {'range': [70, 100], 'color': "#ff7675"}  # Мягкий красный
-            ],
-            'threshold': {
-                'line': {'color': "black", 'width': 4},
-                'thickness': 0.75,
-                'value': value
-            }
-        },
-        number={'font': {'size': 48}}  # Размер шрифта числа
-    ))
-    fig.update_layout(paper_bgcolor="#f8f9fa", font={'color': "#2d3436", 'family': "Arial"}, width=250, height=150)
-    return fig
+# Функция для создания задания для копирайтера
+def generate_standard_prompt(description, advantages, key_message, *selected_values):
+    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
 
-# Функция для генерации случайных значений спидометров
-def generate_random_gauges():
-    return create_gauge(random.randint(60, 90)), create_gauge(random.randint(60, 90)), create_gauge(random.randint(60, 90))
+# Функция для создания задания для редактора
+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 = ""
 
-# Функция для смены вкладки
-def change_tab(id):
-    return gr.Tabs(selected=id)
+    print(f"Выбранные значения: Пол={gender}, Поколение={generation}, Психотип={psychotype}")
 
-# Вспомогательная функция для добавления префиксов и суффиксов
-def add_prefix_suffix(prompt, prefix, suffix):
-    return f"{prefix}\n{prompt}\n{suffix}"
+    # Проверяем, выбраны ли все три параметра: Пол, Поколение, Психотип
+    if gender and generation and psychotype:
+        # Получаем данные с листа "Пол Поколение Психотип"
+        sheet = features.get("Пол Поколение Психотип", {})
 
-# Функция для обрезки сообщения до последнего знака препинания
-def clean_message(message):
-    if not message.endswith(('.', '!', '?')):
-        last_period = max(message.rfind('.'), message.rfind('!'), message.rfind('?'))
-        if last_period != -1:
-            message = message[:last_period + 1]
-    return message
+        # Ищем ключ, соответствующий комбинации "Пол", "Поколение", "Психотип"
+        key = (gender, generation, psychotype)
+        if key in sheet:
+            combined_instruction = sheet[key]
+            print(f"Найдена комбинированная инструкция: {combined_instruction}")
+        else:
+            print(f"Комбинированная инструкция для ключа {key} не найдена.")
 
-# Функция для генерации сообщения с GigaChat Pro
-def generate_message_gigachat_pro(prompt):
-    try:
-        messages = [SystemMessage(content=prompt)]
-        res = chat_pro(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+    # Если не найдена комбинированная инструкция, добавляем индивидуальные инструкции
+    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()}"
+    
+    if "призыва к действию" в prompt and "минимум прямых призывов к действию" в prompt:
+        prompt = re.sub(r"Убедись, что готовый текст начинается с призыва к действию с продуктом.\n", "", prompt)
+        
+    return prompt.strip()
 
-# Функция для повторной генерации сообщения, пока оно не станет короче 250 знаков
-def generate_message_gigachat_pro_with_retry(prompt):
-    for _ in range(10):
-        message = generate_message_gigachat_pro(prompt)
-        if len(message) <= 250:
-            return message
-    return message
 
 # Функция для постепенной генерации всех сообщений через yield
-def generate_all_messages(desc, benefits, key_message):
+def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
+    # Генерация задания для копирайтера
+    standard_prompt = generate_standard_prompt(desc, benefits, key_message)
+    yield standard_prompt, None, None, None, None, None, None, None
+
+    # Небольшая пауза для демонстрации постепенной генерации
+    time.sleep(1)
+
+    # Генерация задания для редактора
+    personalization_prompt = generate_personalization_prompt(key_message, gender, generation, psychotype, business_stage, industry, opf)
+    yield standard_prompt, personalization_prompt, None, None, None, None, None, None
+    
+    # Небольшая пауза для демонстрации постепенной генерации
+    time.sleep(1)
+
     # Варианты предложений для начала и конца
     prefixes = [
         "Начни сообщение с призыва к действию с продуктом.",
@@ -110,32 +165,16 @@ def generate_all_messages(desc, benefits, key_message):
     non_personalized_messages = []
     personalized_messages = []
     
-    # Формируем общий неперсонализированный текст для использования в сообщениях
-    non_personalized_prompt = f"{desc}\n\nПреимущества:\n{benefits}\n\nКлючевое сообщение: {key_message}"
-    personalized_prompt = f"Персонализированное сообщение для клиента: {desc}\n\n{benefits}\n\nКлючевое сообщение: {key_message}"
-
-    # Выводим задание для копирайтера
-    yield non_personalized_prompt, None, None, None, None, None, None, None
-    
-    # Небольшая пауза для демонстрации постепенной генерации
-    time.sleep(1)
-    
-    # Выводим задание для редактора
-    yield non_personalized_prompt, personalized_prompt, None, None, None, None, None, None
-    
-    # Небольшая пауза для демонстрации постепенной генерации
-    time.sleep(1)
-    
     # Генерация и постепенная подача каждого сообщения
     for i in range(3):
         # Генерация неперсонализированного сообщения
-        prompt = add_prefix_suffix(non_personalized_prompt, prefixes[i], suffixes[i])
+        prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
         non_personalized_message = generate_message_gigachat_pro_with_retry(prompt)
         non_personalized_messages.append(non_personalized_message)
         
         # Выводим неперсонализированное сообщение
         yield (
-            non_personalized_prompt, personalized_prompt,  # Задания для копирайтера и редактора
+            standard_prompt, personalization_prompt,  # Задания для копирайтера и редактора
             non_personalized_messages[0] if i >= 0 else None,  # Первое неперсонализированное сообщение
             personalized_messages[0] if len(personalized_messages) > 0 else None,  # Первое персонализированное сообщение, если оно уже есть
             non_personalized_messages[1] if i >= 1 else None,  # Второе неперсонализированное сообщение
@@ -145,13 +184,13 @@ def generate_all_messages(desc, benefits, key_message):
         )
         
         # Генерация персонализированного сообщения
-        full_personalized_prompt = f"{personalized_prompt}\n\nТекст для адаптации: {non_personalized_message}"
+        full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
         personalized_message = generate_message_gigachat_pro_with_retry(full_personalized_prompt)
         personalized_messages.append(personalized_message)
 
         # Выводим персонализированное сообщение
         yield (
-            non_personalized_prompt, personalized_prompt,  # Задания для копирайтера и редактора
+            standard_prompt, personalization_prompt,  # Задания для копирайтера и редактора
             non_personalized_messages[0] if len(non_personalized_messages) > 0 else None,  # Первое неперсонализированное сообщение
             personalized_messages[0] if len(personalized_messages) > 0 else None,  # Первое персонализированное сообщение
             non_personalized_messages[1] if len(non_personalized_messages) > 1 else None,  # Второе неперсонализированное сообщение
@@ -249,11 +288,11 @@ with gr.Blocks() as demo:
             # Сначала переключаем вкладку, потом запускаем генерацию сообщений
             btn_to_prompts.click(
                 fn=change_tab, 
-                inputs=[gr.Number(value=1, visible=False)],  # Переключение на вкладку "Сообщения" (id=1)
+                inputs=[gr.Number(value=1, visible=False)],  # Переключение на вкладку "Ассистент" (id=1)
                 outputs=tabs  # Обновляем вкладку
             ).then(
                 fn=generate_all_messages, 
-                inputs=[desc, benefits, key_message],  # Входные текстовые поля
+                inputs=[desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf],  # Входные текстовые поля
                 outputs=[
                     non_personalized_prompt, personalized_prompt,  # Поля для задания копирайтера и редактора (на вкладке "Ассистент")
                     non_personalized_1, personalized_1,  # Сообщения на вкладке "Сообщения"