Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import pandas as pd
 import os
 import random
+import re
+import pymorphy3
 from openai import OpenAI
 
 XAI_API_KEY = os.getenv("XAI_API_KEY")
@@ -16,7 +18,10 @@ def load_dropdown_data(file_path, sheet_name, column_name):
 
 file_path = "Исходные данные.xlsx"
 
-products, data_products = load_dropdown_data(file_path, "Продукты", "Наименование продукта"), pd.read_excel(file_path, sheet_name="Продукты")
+products_list, data_products = load_dropdown_data(file_path, "Продукты", "Наименование продукта"), pd.read_excel(file_path, sheet_name="Продукты")
+# Добавляем "Свой продукт" в начало списка
+products = ["Свой продукт"] + list(products_list)
+
 genders_data = pd.read_excel(file_path, sheet_name="Пол")
 generations_data = pd.read_excel(file_path, sheet_name="Поколение")
 psychotypes_data = pd.read_excel(file_path, sheet_name="Психотип")
@@ -49,15 +54,25 @@ approach_dict = {
 }
 
 def fill_product_details(selected_product, data):
-    if selected_product:
-        product_row = data[data["Наименование продукта"] == selected_product].iloc[0]
-        return (
-            product_row.get("Описание предложения", ""),
-            product_row.get("Наименование продукта", ""),
-            product_row.get("Преимущества", ""),
-            product_row.get("Ключевое сообщение", "")
-        )
-    return "", "", "", ""
+    # Если выбран "Свой продукт", поля очищаем и даём редактировать
+    if selected_product == "Свой продукт":
+        return gr.update(value="", interactive=True), \
+               gr.update(value="", interactive=True), \
+               gr.update(value="", interactive=True), \
+               gr.update(value="", interactive=True)
+    else:
+        # Иначе заполняем из файла и делаем поля неактивными
+        if selected_product and selected_product in data["Наименование продукта"].values:
+            product_row = data[data["Наименование продукта"] == selected_product].iloc[0]
+            return (gr.update(value=product_row.get("Описание предложения", ""), interactive=False),
+                    gr.update(value=product_row.get("Наименование продукта", ""), interactive=False),
+                    gr.update(value=product_row.get("Преимущества", ""), interactive=False),
+                    gr.update(value=product_row.get("Ключевое сообщение", ""), interactive=False))
+        else:
+            return gr.update(value="", interactive=False), \
+                   gr.update(value="", interactive=False), \
+                   gr.update(value="", interactive=False), \
+                   gr.update(value="", interactive=False)
 
 def get_approaches(gender, generation, psychotype, approaches_df):
     if approaches_df is None or approaches_df.empty:
@@ -221,6 +236,114 @@ def call_model(model_prompt):
     )
     return completion.choices[0].message.content.strip()
 
+# Функция для корректировки
+def correct_dash_usage(text):
+    morph = pymorphy3.MorphAnalyzer()
+    text = re.sub(r'\s[-–—]\s', ' — ', text)
+    text = re.sub(r'(?<=\d)[-–—](?=\d)', '–', text)
+    text = re.sub(r'(?<=[a-zA-Zа-яА-Я0-9])[-–—](?=[a-zA-Zа-яА-Я0-9])', '-', text)
+    text = re.sub(r'"([^\"]+)"', r'«\1»', text)
+    if text.count('"') == 1:
+        text = text.replace('"', '')
+    if (text.startswith('"') and text.endswith('"')) or (text.startswith('«') and text.endswith('»')):
+        text = text[1:-1].strip()
+    text = re.sub(r'(\d+)[kкКK]', r'\1 000', text, flags=re.IGNORECASE)
+
+    greeting_patterns = [
+        r"привет\b", r"здравствуй", r"добрый\s(день|вечер|утро)",
+        r"дорогой\b", r"уважаемый\b", r"дорогая\b", r"уважаемая\b",
+        r"господин\b", r"госпожа\b", r"друг\b", r"коллега\b",
+        r"товарищ\b", r"приятель\b", r"подруга\b"
+    ]
+    def is_greeting_sentence(sentence):
+        words = sentence.split()
+        if len(words) < 5:
+            for word in words:
+                parsed = morph.parse(word.lower())[0]
+                for pattern in greeting_patterns:
+                    if re.search(pattern, parsed.normal_form):
+                        return True
+        return False
+
+    sentences = re.split(r'(?<=[.!?])\s+', text)
+    if sentences and is_greeting_sentence(sentences[0]):
+        sentences = sentences[1:]
+    text = ' '.join(sentences)
+
+    def restore_yo(text):
+        morph2 = pymorphy3.MorphAnalyzer()
+        words = text.split()
+        restored_words = []
+        for word in words:
+            if word.isupper():
+                restored_words.append(word)
+                continue
+            if word.lower() == "все":
+                restored_words.append(word)
+                continue
+            parsed = morph2.parse(word)[0]
+            restored_word = parsed.word
+            if word and word[0].isupper():
+                restored_word = restored_word.capitalize()
+            restored_words.append(restored_word)
+        return ' '.join(restored_words)
+
+    text = restore_yo(text)
+    text = re.sub(r'\bИп\b', 'ИП', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bОоо\b', 'ООО', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bРф\b', 'РФ', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bпользовуйтесь\b', 'пользуйтесь', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bею\b', 'ей', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bповышьте\b', 'повысьте', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bСбербизнес(?:а|е)?\b', 'СберБизнес', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bСбербанк\b', 'СберБанк', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bвашего ООО\b', 'вашей компании', text, flags=re.IGNORECASE)
+    text = re.sub(r'\b0₽\b', '0 р', text, flags=re.IGNORECASE)
+    text = re.sub(r'\b₽\b', 'р', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bруб\.(?=\W|$)', 'р', text, flags=re.IGNORECASE)
+    text = re.sub(r'\bруб(?:ля|лей)\b', 'р', text, flags=re.IGNORECASE)
+    text = re.sub(r'(\d+)\s+тысяч(?:а|и)?(?:\s+рублей)?', r'\1 000 р', text, flags=re.IGNORECASE)
+    text = re.sub(r'(\d+)\s*тыс\.\s*руб\.', r'\1 000 р', text, flags=re.IGNORECASE)
+    text = re.sub(r'(\d+)\s*тыс\.\s*р\.?', r'\1 000 р', text, flags=re.IGNORECASE)
+    text = re.sub(r'(\d+)\s+миллиона\b|\bмиллионов\b', r'\1 млн', text, flags=re.IGNORECASE)
+    text = re.sub(r'(\d+)\s*млн\s*руб\.', r'\1 млн р', text, flags=re.IGNORECASE)
+    text = re.sub(r'(\d+)\s*р\b', r'\1 р', text)
+
+    def remove_specific_sentences(text):
+        sents = re.split(r'(?<=[.!?])\s+', text)
+        filtered = [s for s in sents if not re.search(r'\bникаких\s+(посещений|визитов)\b', s, re.IGNORECASE)]
+        return ' '.join(filtered)
+
+    text = re.sub(r'\b(\d+)\s+000\s+000\s*р\b', r'\1 млн р', text, flags=re.IGNORECASE)
+    text = re.sub(r' р р ', r' р ', text, flags=re.IGNORECASE)
+    text = remove_specific_sentences(text)
+    return text
+
+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_message_with_retry(model_prompt):
+    # До 10 попыток, чтобы длина была от 160 до 250 символов
+    last_message = ""
+    for _ in range(10):
+        msg = call_model(model_prompt)
+        msg = correct_dash_usage(msg)
+        msg = clean_message(msg)
+        length = len(msg)
+        if 160 <= length <= 250:
+            # Добавляем информацию о количестве знаков
+            msg += f"\n\n------\nКоличество знаков: {length}"
+            return msg
+        last_message = msg
+    # Если не удалось подобрать длину
+    length = len(last_message)
+    last_message += f"\n\n------\nКоличество знаков: {length}"
+    return last_message
+
 def update_prompts_on_params_change(description, product_name, benefits, key_message, 
                                     gender, generation, psychotype, business_stage, industry, opf):
     chosen_approach = get_approaches(gender, generation, psychotype, approaches_data)
@@ -232,10 +355,9 @@ def update_prompts_on_params_change(description, product_name, benefits, key_mes
 def generate_personalized_sms(description, product_name, benefits, key_message,
                               gender, generation, psychotype, business_stage, industry, opf,
                               chosen_approach, prompt_1, prompt_2):
-    # Если нет персональных параметров или подхода, показываем предупреждение:
+    # Если нет персональных параметров или подхода
     if "Для формирования промпта выберите хотя бы один личный персональный параметр" in prompt_1 or chosen_approach == "Подход не найден для выбранных параметров.":
         gr.Warning("Задайте хотя бы один личный персональный параметр для определения подхода, чтобы был сформирован промпт")
-        # Возвращаем пустые результаты для SMS
         return "", ""
 
     approach_list = [a.strip() for a in chosen_approach.split(',') if a.strip()]
@@ -253,8 +375,8 @@ def generate_personalized_sms(description, product_name, benefits, key_message,
                                            gender, generation, psychotype, business_stage, industry, opf,
                                            chosen_single_approach_2)
 
-    sms_1 = call_model(model_prompt_1)
-    sms_2 = call_model(model_prompt_2)
+    sms_1 = generate_message_with_retry(model_prompt_1)
+    sms_2 = generate_message_with_retry(model_prompt_2)
 
     return sms_1, sms_2
 
@@ -272,11 +394,11 @@ with gr.Blocks(theme="default") as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("**Продукт**")
-            product_dropdown = gr.Dropdown(label="Продукт", choices=products, value=None)
-            description = gr.Textbox(label="Описание предложения", lines=5, value="")
-            product_name = gr.Textbox(label="Наименование продукта", lines=1, value="")
-            benefits = gr.Textbox(label="Преимущества", lines=9, value="")
-            key_message = gr.Textbox(label="Ключевое сообщение", lines=2, value="")
+            product_dropdown = gr.Dropdown(label="Продукт", choices=products, value=products[0])
+            description = gr.Textbox(label="Описание предложения", lines=5, value="", interactive=True)
+            product_name = gr.Textbox(label="Наименование продукта", lines=1, value="", interactive=True)
+            benefits = gr.Textbox(label="Преимущества", lines=9, value="", interactive=True)
+            key_message = gr.Textbox(label="Ключевое сообщение", lines=2, value="", interactive=True)
         
             product_dropdown.change(
                 fn=lambda selected: fill_product_details(selected, data_products),
@@ -292,7 +414,7 @@ with gr.Blocks(theme="default") as demo:
             business_stage_dropdown = gr.Dropdown(label="Стадия бизнеса", choices=["Не выбрано"]+business_stages, value=None)
             industry_dropdown = gr.Dropdown(label="Отрасль", choices=["Не выбрано"]+industries, value=None)
             opf_dropdown = gr.Dropdown(label="ОПФ", choices=["Не выбрано"]+opfs, value=None)
-            chosen_approach = gr.Textbox(label="Выбранный подход", lines=1, value="")
+            chosen_approach = gr.Textbox(label="Выбранный подход", lines=1, value="", interactive=False)
             presence_in_db = gr.Textbox(label="Комментарий", lines=1, value="", interactive=False)
 
     with gr.Row():
@@ -304,12 +426,12 @@ with gr.Blocks(theme="default") as demo:
         with gr.Column():
             model_1_name = gr.Textbox(label="Модель 1", value="Grok-2-1212", interactive=False)
             prompt_1 = gr.Textbox(label="Промпт 1", value="", interactive=False, lines=10)
-            sms_1 = gr.Textbox(label="SMS 1", lines=3, value="")
+            sms_1 = gr.Textbox(label="SMS 1", lines=3, value="", interactive=False)
         
         with gr.Column():
             model_2_name = gr.Textbox(label="Модель 2", value="Grok-2-1212", interactive=False)
             prompt_2 = gr.Textbox(label="Промпт 2", value="", interactive=False, lines=10)
-            sms_2 = gr.Textbox(label="SMS 2", lines=3, value="")
+            sms_2 = gr.Textbox(label="SMS 2", lines=3, value="", interactive=False)
     
     with gr.Row():
         prefer_sms_1_btn = gr.Button("Я предпочитаю это SMS")
@@ -341,9 +463,8 @@ with gr.Blocks(theme="default") as demo:
         )
 
     # Генерация SMS по нажатию кнопки
-    # Если персональные параметры не выбраны - выдаём gr.Warning
     create_personal_sms_btn.click(
-        fn=generate_personalized_sms,
+        fn=generate_personal_sms,
         inputs=[description, product_name, benefits, key_message, 
                 gender_dropdown, generation_dropdown, psychotype_dropdown,
                 business_stage_dropdown, industry_dropdown, opf_dropdown,
@@ -351,5 +472,4 @@ with gr.Blocks(theme="default") as demo:
         outputs=[sms_1, sms_2]
     )
 
-    # Обязательно включаем очередь:
     demo.queue().launch()