Update app.py

app.py

@@ -8,8 +8,8 @@ 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)
+gc_key = "YOUR_GIGACHAT_KEY"  # Замените на ваш ключ
+chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro-3', max_tokens=68, temperature=1, verify_ssl_certs=False)
 
 # Загрузка данных из Excel-файла
 try:
@@ -30,57 +30,80 @@ 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_random_gauges():
-    return create_gauge(random.randint(60, 90)), create_gauge(random.randint(60, 90)), create_gauge(random.randint(60, 90))
-
-# Функция для смены вкладки
-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
+# Функция для генерации стандартного промпта
+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, gender, generation, psychotype, business_stage, industry, opf):
+    prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
+    combined_instruction = ""
+    additional_instructions = ""
+
+    if gender and generation and psychotype:
+        key = (gender, generation, psychotype)
+        combined_instruction = features.get("Пол Поколение Психотип", {}).get(key, "")
+
+    if not combined_instruction:
+        additional_instructions += f"{features.get('Пол', {}).get(gender, '')}\n"
+        additional_instructions += f"{features.get('Поколение', {}).get(generation, '')}\n"
+        additional_instructions += f"{features.get('Психотип', {}).get(psychotype, '')}\n"
+
+    additional_instructions += f"{features.get('Стадия бизнеса', {}).get(business_stage, '')}\n"
+    additional_instructions += f"{features.get('Отрасль', {}).get(industry, '')}\n"
+    additional_instructions += f"{features.get('ОПФ', {}).get(opf, '')}\n"
+
+    prompt += combined_instruction if combined_instruction else additional_instructions
+    prompt += f"Убедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
+    
+    return prompt.strip()
 
 # Функция для генерации сообщения с 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
+        return res.content.strip()
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
@@ -92,6 +115,15 @@ def generate_message_gigachat_pro_with_retry(prompt):
             return message
     return message
 
+# Функция для обработки генерации промптов и сообщений
+def handle_generation(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
+    # Генерация стандартного промпта
+    standard_prompt = generate_standard_prompt(desc, benefits, key_message)
+    # Генерация персонализированного промпта
+    personalization_prompt = generate_personalization_prompt(key_message, gender, generation, psychotype, business_stage, industry, opf)
+    
+    return standard_prompt, personalization_prompt
+
 # Функция для генерации всех сообщений
 def generate_all_messages(non_personalized_prompt, personalized_prompt):
     # Варианты предложений для начала и конца
@@ -133,7 +165,7 @@ with gr.Blocks() as demo:
             with gr.Row():
                 with gr.Column():
                     desc = gr.Textbox(
-                        label="Описание предложения (предзаполненный пример можно поменять на свой)", 
+                        label="Описание предложения", 
                         lines=7,
                         value=(
                             "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
@@ -146,22 +178,11 @@ with gr.Blocks() as demo:
                         )
                     )
                     benefits = gr.Textbox(
-                        label="Преимущества (предзаполненный пример можно поменять на свой)", 
+                        label="Преимущества", 
                         lines=5,
-                        value=(
-                            "Предложение по бесплатному обслуживанию — бессрочное.\n"
-                            "Оплата ��окупок без отчётов и платёжных поручений.\n"
-                            "Платёжные документы без комиссии.\n"
-                            "Лимиты на расходы сотрудников.\n"
-                            "Мгновенные переводы на карты любых банков."
-                        )
-                    )
-                    
-                    key_message = gr.Textbox(
-                        label="Ключевое сообщение (предзаполненный пример можно поменять на свой)",
-                        lines=3,
-                        value="Бесплатное обслуживание при покупках от 100 000 рублей в месяц."
+                        value="Преимущества для теста"
                     )
+                    key_message = gr.Textbox(label="Ключевое сообщение", lines=3, value="Бесплатное обслуживание при покупках от 100 000 рублей в месяц.")
 
                 with gr.Column():
                     gender = gr.Dropdown(label="Пол", choices=[None] + list(features.get('Пол', {}).keys()))
@@ -171,20 +192,22 @@ with gr.Blocks() as demo:
                     industry = gr.Dropdown(label="Отрасль", choices=[None] + list(features.get('Отрасль', {}).keys()))
                     opf = gr.Dropdown(label="ОПФ", choices=[None] + list(features.get('ОПФ', {}).keys()))
 
-            btn_to_prompts = gr.Button("Создать")
-            btn_to_prompts.click(fn=change_tab, inputs=[gr.Number(value=1, visible=False)], outputs=tabs)
+            btn_generate_prompts = gr.Button("Создать")
+            # Привязка кнопки к функции генерации промптов
+            btn_generate_prompts.click(
+                fn=handle_generation, 
+                inputs=[desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf],
+                outputs=[gr.Textbox.update(label="Неперсонализированный промпт"), gr.Textbox.update(label="Задание для редактора")]
+            )
         
         # Вкладка 2: Промпты
         with gr.TabItem("Ассистент", id=1):
             with gr.Row():
                 with gr.Column():
-                    non_personalized_prompt = gr.Textbox(
-                        label="Неперсонализированный промпт", 
-                        lines=25,
-                        interactive=False)
+                    non_personalized_prompt = gr.Textbox(label="Неперсонализированный промпт", lines=25, interactive=False)
                 with gr.Column():
-                    personalized_prompt = gr.Textbox(label="Задание для редактора", lines=25)  # Увеличенная высота
-            
+                    personalized_prompt = gr.Textbox(label="Задание для редактора", lines=25)
+
         # Вкладка 3: Сообщения
         with gr.TabItem("Сообщения", id=2):
             with gr.Row():
@@ -203,41 +226,12 @@ with gr.Blocks() as demo:
                 non_personalized_3 = gr.Textbox(label="Стандартное сообщение 3", lines=4, interactive=False)
                 personalized_3 = gr.Textbox(label="Персонализированное сообщение 3", lines=4, interactive=False)
 
+            # Привязка кнопки для генерации сообщений
             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("### Аналитик")
-                
-            with gr.Row():
-                personalized_message_1 = gr.Textbox(label="Персонализированное сообщение 1", lines=5)
-                check_message_1 = gr.Textbox(label="Проверка сообщения 1", lines=5)
-                with gr.Column():
-                    gr.HTML("<div style='display:flex; justify-content:center; width:100%;'>")
-                    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)
-                with gr.Column():
-                    gr.HTML("<div style='display:flex; justify-content:center; width:100%;'>")
-                    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)              
-                with gr.Column():
-                    gr.HTML("<div style='display:flex; justify-content:center; width:100%;'>")
-                    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])
+            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]
+            )
 
 demo.launch()