Update app.py

app.py

@@ -1,13 +1,12 @@
-import os
 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 gigachat import GigaChat, SystemMessage
 
 # Авторизация в GigaChat Pro
-gc_key = os.getenv('GPT_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:
@@ -28,42 +27,71 @@ 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 generate_standard_prompt(description, advantages, key_message):
-    prompt = (
-        f"Сгенерируй смс-сообщение для клиента.\n"
-        f"Описание предложения: {description}\n"
-        f"Преимущества: {advantages}\n"
-        "В тексте смс запрещено использование запрещенных слов, обещаний, приветствий и пр. Убедись, что текст до 250 символов.\n"
-    )
-    if key_message.strip():
-        prompt += f"Ключевое сообщение: {key_message.strip()}"
-    return prompt.strip()
-
-# Функция генерации персонализированного промпта
-def generate_personalization_prompt(standard_message, gender, generation, psychotype, business_stage, industry, opf):
-    personalization_prompt = (
-        f"Адаптируй текст с учётом особенностей клиента:\n"
-        f"Пол: {gender}\nПоколение: {generation}\nПсихотип: {psychotype}\nСтадия бизнеса: {business_stage}\nОтрасль: {industry}\nОПФ: {opf}\n"
-        f"Текст для адаптации: {standard_message}"
-    )
-    return personalization_prompt.strip()
-
-# Функция генерации сообщений с GigaChat Pro
+# Функция для обрезки сообщения до последнего знака препинания
+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)
-        return res.content.strip()
+        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 = [
         "Начни сообщение с призыва к действию с продуктом.",
         "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
@@ -81,33 +109,19 @@ def generate_all_messages(non_personalized_prompt, personalized_prompt):
     # Генерация трех неперсонализированных сообщений
     for i in range(3):
         prompt = add_prefix_suffix(non_personalized_prompt, prefixes[i], suffixes[i])
-        message = generate_message_gigachat_pro(prompt)
+        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(prompt)
+        message = generate_message_gigachat_pro_with_retry(prompt)
         personalized_messages.append(message)
 
     return non_personalized_messages, personalized_messages
 
-# Функция генерации стандартного и персонализированного промптов
-def handle_generation(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
-    non_personalized_prompt = generate_standard_prompt(desc, benefits, key_message)
-    personalized_prompt = generate_personalization_prompt(non_personalized_prompt, gender, generation, psychotype, business_stage, industry, opf)
-    
-    # Генерация сообщений
-    non_personalized_messages, personalized_messages = generate_all_messages(non_personalized_prompt, personalized_prompt)
-    
-    # Вернем промпты и сообщения
-    return non_personalized_prompt, personalized_prompt, non_personalized_messages, personalized_messages
-
-# Функция для смены вкладки
-def change_tab(id):
-    return gr.Tabs(selected=id)
-
+# Интерфейс Gradio
 with gr.Blocks() as demo:
     with gr.Tabs() as tabs:
         
@@ -115,69 +129,79 @@ with gr.Blocks() as demo:
         with gr.TabItem("Исходные данные", id=0):
             with gr.Row():
                 with gr.Column():
-                    desc = gr.Textbox(label="Описание предложения", lines=6)
-                    benefits = gr.Textbox(label="Преимущества", lines=5)
-                    key_message = gr.Textbox(label="Ключевое сообщение", lines=5)
+                    desc = gr.Textbox(
+                        label="Описание предложения (предзаполненный пример можно поменять на свой)", 
+                        lines=7,
+                        value=(
+                            "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
+                            "Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
+                            "Что необходимо сделать, чтобы воспользоваться предложением:\n"
+                            "1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
+                            "2. Забрать карту.\n"
+                            "3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
+                            "4. В течение следующего месяца пользоваться ей бесплатно."
+                        )
+                    )
+                    benefits = gr.Textbox(
+                        label="Преимущества (предзаполненный пример можно поменять на свой)", 
+                        lines=5,
+                        value=(
+                            "Предложение по бесплатному обслуживанию — бессрочное.\n"
+                            "Оплата покупок без отчётов и платёжных поручений.\n"
+                            "Платёжные документы без комиссии.\n"
+                            "Лимиты на расходы сотрудников.\n"
+                            "Мгновенные переводы на карты любых банков."
+                        )
+                    )
+                    
+                    key_message = gr.Textbox(
+                        label="Ключевое сообщение (предзаполненный пример можно поменять на свой)",
+                        lines=3,
+                        value="Бесплатное обслуживание при покупках от 100 000 рублей в месяц."
+                    )
+
                 with gr.Column():
-                    gender = gr.Dropdown(label="Пол", choices=["Мужчина", "Женщина", "Не указан"])
-                    generation = gr.Dropdown(label="Поколение", choices=["Поколение Z", "Миллениалы", "Поколение X", "Бэби-бумеры"])
-                    psychotype = gr.Textbox(label="Психотип")
-                    business_stage = gr.Textbox(label="Стадия бизнеса")
-                    industry = gr.Textbox(label="Отрасль")
-                    opf = gr.Textbox(label="ОПФ")
-
-            # Кнопка "Создать" для генерации промптов и сообщений
+                    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()))
+                    business_stage = gr.Dropdown(label="Стадия бизнеса", choices=[None] + list(features.get('Стадия бизнеса', {}).keys()))
+                    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("Создать")
-            
-            non_personalized_prompt = gr.Textbox(label="Задание для копирайтера", lines=25, interactive=False)
-            personalized_prompt = gr.Textbox(label="Задание для редактора", lines=25)
-            
-            non_personalized_1 = gr.Textbox(label="Стандартное сообщение 1", lines=4, interactive=False)
-            personalized_1 = gr.Textbox(label="Персонализированное сообщение 1", lines=4, interactive=False)
-            non_personalized_2 = gr.Textbox(label="Стандартное сообщение 2", lines=4, interactive=False)
-            personalized_2 = gr.Textbox(label="Персонализированное сообщение 2", lines=4, interactive=False)
-            non_personalized_3 = gr.Textbox(label="Стандартное сообщение 3", lines=4, interactive=False)
-            personalized_3 = gr.Textbox(label="Персонализированное сообщение 3", lines=4, interactive=False)
-
-            # Привязка кнопки к функции генерации и смены вкладки
-            btn_to_prompts.click(fn=handle_generation, 
-                                 inputs=[desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf],
-                                 outputs=[non_personalized_prompt, personalized_prompt, 
-                                          non_personalized_1, personalized_1,
-                                          non_personalized_2, personalized_2,
-                                          non_personalized_3, personalized_3])
-            btn_to_prompts.click(fn=change_tab, inputs=[gr.Number(value=1)], outputs=tabs)
+            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.render()
+                    non_personalized_prompt = gr.Textbox(
+                        label="Задание для копирайтера", 
+                        lines=25,
+                        interactive=False)
                 with gr.Column():
-                    personalized_prompt.render()
+                    personalized_prompt = gr.Textbox(label="Задание для редактора", lines=25)  # Увеличенная высота
             
         # Вкладка 3: Сообщения
         with gr.TabItem("Сообщения", id=2):
-
-            # Заголовки столбцов
             with gr.Row():
                 gr.Markdown("### Копирайтер")
                 gr.Markdown("### Редактор")
                 
-            # Первый ряд
             with gr.Row():
-                non_personalized_1.render()
-                personalized_1.render()
+                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.render()
-                personalized_2.render()
+                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.render()
-                personalized_3.render()
+                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):
@@ -189,13 +213,28 @@ with gr.Blocks() as demo:
             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)
+                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])
+
 demo.launch()