Update app.py

app.py

@@ -2,6 +2,13 @@ import gradio as gr
 import plotly.graph_objects as go
 import random
 import pandas as pd
+from langchain.schema import SystemMessage
+from langchain_community.chat_models.gigachat import GigaChat
+from openpyxl import load_workbook
+
+# Авторизация в GigaChat Pro
+gc_key = os.getenv('GC_KEY')
+chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, temperature=1, verify_ssl_certs=False)
 
 # Загрузка данных из Excel-файла
 try:
@@ -15,7 +22,6 @@ features = {}
 for sheet_name, df in data.items():
     try:
         if sheet_name == "Пол Поколение Психотип":
-            # Создаем словарь, где ключи — это кортежи (Пол, Поколение, Психотип), а значения — инструкции
             features[sheet_name] = df.set_index(['Пол', 'Поколение', 'Психотип'])['Инструкция'].to_dict()
         else:
             features[sheet_name] = df.set_index(df.columns[0]).to_dict()[df.columns[1]]
@@ -23,105 +29,6 @@ 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(
@@ -143,11 +50,7 @@ def create_gauge(value):
         },
         number={'font': {'size': 48}}  # Размер шрифта числа
     ))
-    # Изменение размеров спидометра и центровка
-    fig.update_layout(paper_bgcolor="#f8f9fa",  # Цвет фона
-                      font={'color': "#2d3436", 'family': "Arial"},  # Цвет текста
-                      width=250,  # Уменьшенная ширина
-                      height=150)  # Уменьшенная высота
+    fig.update_layout(paper_bgcolor="#f8f9fa", font={'color': "#2d3436", 'family': "Arial"}, width=250, height=150)
     return fig
 
 # Функция для генерации случайных значений спидометров
@@ -158,6 +61,69 @@ def generate_random_gauges():
 def change_tab(id):
     return gr.Tabs(selected=id)
 
+# Вспомогательная функция для добавления префиксов и суффиксов
+def add_prefix_suffix(prompt, prefix, suffix):
+    return f"{prefix}\n{prompt}\n{suffix}"
+
+# Функция для обрезки сообщения до последнего знака препинания
+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
+
+# Функция для генерации сообщения с 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}"
+
+# Функция для повторной генерации сообщения, пока оно не станет короче 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
+
+# Функция для генерации всех сообщений
+def generate_all_messages(non_personalized_prompt, personalized_prompt):
+    # Варианты предложений для начала и конца
+    prefixes = [
+        "Начни сообщение с призыва к действию с продуктом.",
+        "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
+        "Начни сообщение с вопроса, который указывает на пользу продукта для клиента."
+    ]
+    suffixes = [
+        "Убедись, что готовый текст начинается с призыва к действию с продуктом.",
+        "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.",
+        "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
+    ]
+    
+    non_personalized_messages = []
+    personalized_messages = []
+    
+    # Генерация трех неперсонализированных сообщений
+    for i in range(3):
+        prompt = add_prefix_suffix(non_personalized_prompt, prefixes[i], suffixes[i])
+        message = generate_message_gigachat_pro_with_retry(prompt)
+        non_personalized_messages.append(message)
+    
+    # Генерация трех персонализированных сообщений
+    for i in range(3):
+        full_personalized_prompt = f"{personalized_prompt}\n\nТекст для адаптации: {non_personalized_messages[i]}"
+        prompt = add_prefix_suffix(full_personalized_prompt, prefixes[i], suffixes[i])
+        message = generate_message_gigachat_pro_with_retry(prompt)
+        personalized_messages.append(message)
+
+    return non_personalized_messages, personalized_messages
+
+# Интерфейс Gradio
 with gr.Blocks() as demo:
     with gr.Tabs() as tabs:
         
@@ -197,7 +163,6 @@ with gr.Blocks() as demo:
                     )
 
                 with gr.Column():
-                    # Используем данные из Excel для выбора
                     gender = gr.Dropdown(label="Пол", choices=[None] + list(features.get('Пол', {}).keys()))
                     generation = gr.Dropdown(label="Поколение", choices=[None] + list(features.get('Поколение', {}).keys()))
                     psychotype = gr.Dropdown(label="Психотип", choices=[None] + list(features.get('Психотип', {}).keys()))
@@ -207,70 +172,46 @@ 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)
-                 
+        
         # Вкладка 2: Промпты
         with gr.TabItem("Ассистент", id=1):
             with gr.Row():
                 with gr.Column():
                     non_personalized_prompt = gr.Textbox(
-                        label="Задание для копирайтера", 
+                        label="Неперсонализированный промпт", 
                         lines=25,
                         interactive=False)
                 with gr.Column():
                     personalized_prompt = gr.Textbox(label="Задание для редактора", lines=25)  # Увеличенная высота
-
-        # Добавляем генерацию стандартного промпта при клике
-        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]
-        )
             
         # Вкладка 3: Сообщения
         with gr.TabItem("Сообщения", id=2):
-
-            # Заголовки столбцов
             with gr.Row():
-                gr.Markdown("### Копирайтер")  # Название для столбца "Копирайтер"
-                gr.Markdown("### Редактор")  # Название для столбца "Редактор"
+                gr.Markdown("### Копирайтер")
+                gr.Markdown("### Редактор")
                 
-            # Первый ряд
             with gr.Row():
                 non_personalized_1 = gr.Textbox(label="Стандартное сообщение 1", lines=4, interactive=False)
                 personalized_1 = gr.Textbox(label="Персонализированное сообщение 1", lines=4, interactive=False)
 
-            # Второй ряд
             with gr.Row():
                 non_personalized_2 = gr.Textbox(label="Стандартное сообщение 2", lines=4, interactive=False)
                 personalized_2 = gr.Textbox(label="Персонализированное сообщение 2", lines=4, interactive=False)
 
-            # Третий ряд
             with gr.Row():
                 non_personalized_3 = gr.Textbox(label="Стандартное сообщение 3", lines=4, interactive=False)
                 personalized_3 = gr.Textbox(label="Персонализированное сообщение 3", lines=4, interactive=False)
 
-            # Четвертый ряд
-            with gr.Row():
-                btn_check = gr.Button("Проверить", elem_id="check3")
-                btn_check.click(fn=change_tab, inputs=[gr.Number(value=3, visible=False)], outputs=tabs)
+            btn_generate_messages = gr.Button("Генерировать сообщения")
+            btn_generate_messages.click(fn=generate_all_messages, inputs=[non_personalized_prompt, personalized_prompt], outputs=[non_personalized_1, personalized_1, non_personalized_2, personalized_2, non_personalized_3, personalized_3])
 
         # Вкладка 4: Проверка
         with gr.TabItem("Проверка", id=3):
-            
-            # Заголовки столбцов
             with gr.Row():
                 gr.Markdown("### Редактор")
-                gr.Markdown("### Корректор")  # Название для столбца "Копирайтер"
-                gr.Markdown("### Аналитик")  # Название для столбца "Редактор"
+                gr.Markdown("### Корректор")
+                gr.Markdown("### Аналитик")
                 
-            # Первый ряд
             with gr.Row():
                 personalized_message_1 = gr.Textbox(label="Персонализированное сообщение 1", lines=5)
                 check_message_1 = gr.Textbox(label="Проверка сообщения 1", lines=5)
@@ -279,7 +220,6 @@ with gr.Blocks() as demo:
                     success_forecast_1 = gr.Plot(label="Прогноз успешности сообщения 1")
                     gr.HTML("</div>")
             
-            # Второй ряд
             with gr.Row():
                 personalized_message_2 = gr.Textbox(label="Персонализированное сообщение 2", lines=5)
                 check_message_2 = gr.Textbox(label="Проверка сообщения 2", lines=5)
@@ -288,7 +228,6 @@ with gr.Blocks() as demo:
                     success_forecast_2 = gr.Plot(label="Прогноз успешности сообщения 2")
                     gr.HTML("</div>")
             
-            # Третий ряд
             with gr.Row():
                 personalized_message_3 = gr.Textbox(label="Персонализированное сообщение 3", lines=5)
                 check_message_3 = gr.Textbox(label="Проверка сообщения 3", lines=5)              
@@ -297,7 +236,7 @@ with gr.Blocks() as demo:
                     success_forecast_3 = gr.Plot(label="Прогноз успешности сообщения 3")
                     gr.HTML("</div>")
                 
-            # Кнопка для обновления спидометров
+            btn_check = gr.Button("Проверить")
             btn_check.click(fn=generate_random_gauges, inputs=[], outputs=[success_forecast_1, success_forecast_2, success_forecast_3])
 
 demo.launch()