Create app.py

app.py

@@ -0,0 +1,406 @@
+import gradio as gr
+import requests
+import os
+import json
+import pandas as pd
+import time
+from langchain.schema import SystemMessage
+from langchain_community.chat_models.gigachat import GigaChat
+from openpyxl import load_workbook
+import base64
+
+# Установка ключа API для OpenAI и GigaChat
+openai_api_key = os.getenv('GPT_KEY')
+gc_key = os.getenv('GC_KEY')
+token = os.getenv('GITHUB_TOKEN')
+
+# Авторизация в сервисе GigaChat
+chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
+chat_lite = GigaChat(credentials=gc_key, model='GigaChat', verify_ssl_certs=False)
+chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', verify_ssl_certs=False)
+
+# Загрузка данных из Excel-файла
+try:
+    data = pd.read_excel('Признаки.xlsx', sheet_name=None)
+except Exception as e:
+    print(f"Ошибка при загрузке Excel-файла: {e}")
+    data = {}
+
+# Создание списка признаков и их значений
+features = {}
+for sheet_name, df in data.items():
+    try:
+        features[sheet_name] = df.set_index(df.columns[0]).to_dict()[df.columns[1]]
+    except Exception as e:
+        print(f"Ошибка при обработке данных листа {sheet_name}: {e}")
+        features[sheet_name] = {}
+
+# Функция для генерации стандартного промпта
+def generate_standard_prompt(description, advantages, *selected_values):
+    prompt = (
+        "Сгенерируй смс-сообщение для клиента.\n"
+        "Объем готового текста: до 250 знаков с пробелами.\n"
+        "Начни сообщение с призыва к действию с продуктом.\n"
+        f"Описание предложения: {description}\n"
+        f"Преимущества: {advantages}\n"
+        "Вклад на короткий срок.\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 знаков с пробелами."
+    )
+    return prompt
+
+# Функции для генерации сообщений
+def generate_message_gpt4o(prompt):
+    try:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {openai_api_key}"
+        }
+        data = {
+            "model": "chatgpt-4o-latest",
+            "messages": [{"role": "system", "content": prompt}],
+            "max_tokens": 700
+        }
+        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
+        response_data = response.json()
+        return response_data["choices"][0]["message"]["content"].strip()
+    except Exception as e:
+        return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
+
+def generate_message_gigachat_pro(prompt):
+    try:
+        messages = [SystemMessage(content=prompt)]
+        res = chat_pro(messages)
+        return res.content.strip()
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Pro: {e}"
+
+def generate_message_gigachat_lite(prompt):
+    try:
+        time.sleep(2)
+        messages = [SystemMessage(content=prompt)]
+        res = chat_lite(messages)
+        return res.content.strip()
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Lite: {e}"
+
+def generate_message_gigachat_plus(prompt):
+    try:
+        time.sleep(2)
+        messages = [SystemMessage(content=prompt)]
+        res = chat_plus(messages)
+        return res.content.strip()
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Lite+: {e}"
+
+# Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
+def generate_messages(description, advantages, *selected_values):
+    standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
+
+    results = {
+        "prompt": standard_prompt,
+        "gpt4o": None,
+        "gigachat_pro": None,
+        "gigachat_lite": None,
+        "gigachat_plus": None
+    }
+
+    yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
+
+    results["gpt4o"] = generate_message_gpt4o(standard_prompt)
+    yield results["prompt"], results["gpt4o"], "", "", "", "Сообщение GPT-4o сгенерировано"
+
+    results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", "", "Сообщение GigaChat-Pro сгенерировано"
+    time.sleep(2)
+    
+    results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], "", "Сообщение GigaChat-Lite сгенерировано"
+    time.sleep(2)
+    
+    results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"], "Все сообщения сгенерированы"
+
+    return results
+
+# Функция для генерации персонализированного промпта
+def generate_personalization_prompt(*selected_values):
+    prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
+    for i, feature in enumerate(features.keys()):
+        if selected_values[i]:
+            try:
+                prompt += f"{features[feature][selected_values[i]]}\n"
+            except KeyError:
+                return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
+                
+    prompt += "Убедись, что в готовом тексте до 250 знаков с пробелами."
+    
+    return prompt.strip()
+
+# Функция для выполнения персонализации на основе сгенерированного промпта и сообщения
+def perform_personalization(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_gpt4o(full_prompt)
+
+def perform_personalization_gigachat(standard_message, personalization_prompt, model):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    if model == "gigachat_pro":
+        return generate_message_gigachat_pro(full_prompt)
+    elif model == "gigachat_lite":
+        return generate_message_gigachat_lite(full_prompt)
+    elif model == "gigachat_plus":
+        return generate_message_gigachat_plus(full_prompt)
+
+# Функция для выполнения персонализации с использованием yield
+def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, *selected_values):
+    personalization_prompt = generate_personalization_prompt(*selected_values)
+    yield personalization_prompt, "", "", "", "", "Промпт для персонализации сгенерирован"
+
+    personalized_message_gpt4o = perform_personalization(gpt4o_message, personalization_prompt)
+    yield personalization_prompt, personalized_message_gpt4o, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
+
+    personalized_message_gigachat_pro = perform_personalization_gigachat(gigachat_pro_message, personalization_prompt, "gigachat_pro")
+    yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, "", "", "Персонализированное сообщение GigaChat-Pro сгенерировано"
+
+    personalized_message_gigachat_lite = perform_personalization_gigachat(gigachat_lite_message, personalization_prompt, "gigachat_lite")
+    yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, "", "Персонализированное сообщение GigaChat-Lite сгенерировано"
+
+    personalized_message_gigachat_plus = perform_personalization_gigachat(gigachat_plus_message, personalization_prompt, "gigachat_plus")
+    yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "Все персонализированные сообщения сгенерированы"
+
+# Функция для генерации промпта проверки текста
+def generate_error_check_prompt():
+    prompt = (
+        "Проверь текст SMS-сообщения на соответствие установленным правилам и ограничениям, касающимся его формирования. На основе выявленных несоответствий предоставь рекомендации по исправлению текста. "
+        "Особое внимание удели проверке: количества символов в тексте SMS-сообщения, орфографическим и пунктуационным ошибкам, определению частей речи (причастия, деепричастия, причастный оборот, деепричастный оборот). "
+        "Анализируй только текст SMS-сообщения, ничего не придумывай и не добавляй лишнего. "
+        "Правила и ограничения, которым должен соответствовать текст SMS-сообщения:\n"
+        "1. Количество символов в SMS-сообщении должно быть до 250 знаков с учетом пробелов.\n"
+        "2. В тексте должен быть призыв к действию с использованием глагола в повелительном наклонении (например: оформите, получите, разместите, размещайте, откройте, подключите, подайте заявку).\n"
+        "3. Должно соблюдаться соответствие фактов о продукте.\n"
+        "4. В генерациях смс запрещено использовать обещания и гарантии.\n"
+        "5. В генерациях смс запрещено использовать составные конструкции из двух глаголов.\n"
+        "6. В генерациях смс запрещено использовать причастия и причастные обороты.\n"
+        "7. В генерациях смс запрещено использовать деепричастия и деепричастные обороты.\n"
+        "8. В генерациях смс запрещено использовать превосходную степень прилагательных.\n"
+        "9. В генерациях смс запрещено использовать страдательный залог.\n"
+        "10. В генерациях смс запрещено использовать порядковые числительные от 10 прописью.\n"
+        "11. В генерациях смс запрещено использовать цепочки с придаточными предложениями.\n"
+        "12. В генерациях смс запрещено использовать разделительные повторяющиеся союзы.\n"
+        "13. В генерациях смс запрещено использовать вводные конструкции.\n"
+        "14. В генерациях смс запрещено использовать усилители.\n"
+        "15. В генерациях смс запрещено использовать паразиты времени.\n"
+        "16. В генерациях смс запрещено использовать несколько существительных подряд, в том числе отг��агольных.\n"
+        "17. В генерациях смс запрещено использовать производные предлоги.\n"
+        "18. В генерациях смс запрещено использовать сложные предложения, в которых нет связи между частями.\n"
+        "19. В генерациях смс запрещено использовать сложноподчинённые предложения.\n"
+        "20. В генерациях смс запрещено использовать даты прописью.\n"
+        "21. В генерациях смс запрещено использовать близкие по смыслу однородные члены.\n"
+        "22. В генерациях смс запрещено использовать шокирующие, экстравагантные, кликбейтные фразы.\n"
+        "23. В генерациях смс запрещено использовать абстрактные заявления без поддержки фактами и отсутствие доказательства пользы для клиента.\n"
+        "24. В генерациях смс запрещено использовать гарантирующие фразы.\n"
+        "25. В генерациях смс запрещено использовать узкоспециализированные термины.\n"
+        "26. В генерациях смс запрещено использовать фразы, способные создать двойственное ощущение, обидеть.\n"
+        "27. В генерациях смс запрещено использовать речевые клише, рекламные штампы, канцеляризмы.\n"
+        "28. В генерациях смс запрещено использовать запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, банкротство, долги, займ, срочно, лучший, главный, номер 1, успех, лидер.\n"
+        "29. Сообщение должно быть написано без орфографических и грамматических ошибок.\n"
+        "30. Запрещены повторы слов.\n"
+        "31. В тексте должны использоваться правильные знаки препинания.\n"
+        "32. Если в тексте используются кавычки, они должны быть в форме «кавычки-ёлочки».\n"
+        "33. В тексте SMS сообщения должны обязательно присутствовать: название продукта, условия использования продукта / Преимущества продукта / Шаги для подключения или начала использования / Условия акции (если предложение по продукту акционное).\n"
+        "Форма ответа: [Ответ должен быть кратким, должен содержать только рекомендации по устранению найденных несоответствий, соответствия каждому пункту правил описывать категорически запрещено]."
+    )
+    return prompt
+
+
+# Функция для выполнения проверки текста с использованием yield
+def check_errors_with_yield(*personalized_messages):
+    if len(personalized_messages) < 4:
+        yield "", "", "", "", "", "Ошибка: недостаточно сообщений для проверки"
+        return
+
+    error_check_prompt = generate_error_check_prompt()
+    yield error_check_prompt, "", "", "", "", "Промпт для проверки текста сгенерирован"
+
+    error_message_gpt4o = perform_personalization(f"{error_check_prompt}\n\n{personalized_messages[0]}", "")
+    yield error_check_prompt, error_message_gpt4o, "", "", "", "Результат проверки GPT-4o сгенерирован"
+
+    error_message_gigachat_pro = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[1]}", "", "gigachat_pro")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, "", "", "Результат проверки GigaChat-Pro сгенерирован"
+    
+    time.sleep(3)
+    error_message_gigachat_lite = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[2]}", "", "gigachat_lite")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, "", "Результат проверки GigaChat-Lite сгенерирован"
+    
+    try:
+        time.sleep(3)
+        error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[3]}", "", "gigachat_plus")
+    except Exception as e:
+        error_message_gigachat_plus = f"Ошибка при обработке GigaChat-Plus: {e}"
+        
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
+
+
+def save_to_github(message, model_name):
+    # Формируем уникальное имя файла на основе текущего времени
+    timestamp = int(time.time())
+    file_name = f"file_{timestamp}.json"
+    
+    # Данные для записи
+    file_content = {
+        "Модель": model_name,
+        "Сообщение": message
+    }
+
+    # Преобразуем контент в JSON-строку и кодируем в base64
+    file_content_encoded = base64.b64encode(json.dumps(file_content).encode()).decode()
+
+    # Параметры для GitHub API
+    repo = "fruitpicker01/Storage_1"
+    path = file_name
+    url = f"https://api.github.com/repos/{repo}/contents/{path}"
+    headers = {
+        "Authorization": f"token {token}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "message": f"Добавлен новый файл {file_name}",
+        "content": file_content_encoded
+    }
+
+    # Отправка POST-запроса на GitHub API для создания файла в репозитории
+    response = requests.put(url, headers=headers, data=json.dumps(data))
+    
+    if response.status_code == 201:
+        print(f"Файл {file_name} успешно добавлен в репозиторий.")
+    else:
+        print(f"Ошибка при добавлении файла: {response.status_code}")
+        print(response.json())
+
+
+# Создание интерфейса Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            description_input = gr.Textbox(
+                label="Описание предложения (предзаполненный пример можно поменять на свой)", 
+                lines=13,
+                value=(
+                    "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
+                    "Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
+                    "Что необходимо сделать, чтобы воспользоваться предложением:\n"
+                    "1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
+                    "2. Забрать карту.\n"
+                    "3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
+                    "4. В течение следующего месяца пользоваться ей бесплатно."
+                )
+            )
+            advantages_input = gr.Textbox(
+                label="Преимущества (предзаполненный пример можно поменять на свой)", 
+                lines=6,
+                value=(
+                    "Предложение по бесплатному обслуживанию — бессрочное.\n"
+                    "Оплата покупок без отчётов и платёжных поручений.\n"
+                    "Платёжные документы без комиссии.\n"
+                    "Лимиты на расходы сотрудников.\n"
+                    "Мгновенные переводы на карты любых банков."
+                )
+            )
+            selections = []
+            for feature in features.keys():
+                selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
+
+            submit_btn = gr.Button("1. Создать неперсонализированное сообщение")  # Оранжевая кнопка по умолчанию
+
+        with gr.Column(scale=2):
+            prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=28, interactive=False)
+            output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3, interactive=False)
+            output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3, interactive=False)
+            output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3, interactive=False)
+            output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3, interactive=False)
+            
+            submit_btn.click(
+                generate_messages, 
+                inputs=[description_input, advantages_input] + selections, 
+                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
+            )
+
+    with gr.Row():
+        personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
+        
+    with gr.Row():
+        personalize_btn.click(
+            personalize_messages_with_yield,
+            inputs=[output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus] + selections,
+            outputs=[
+                gr.Textbox(label="Промпт для персонализации", lines=6, interactive=False),
+                personalized_output_text_gpt4o := gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6, interactive=False),
+                personalized_output_text_gigachat_pro := gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=6, interactive=False),
+                personalized_output_text_gigachat_lite := gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=6, interactive=False),
+                personalized_output_text_gigachat_plus := gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=6, interactive=False)
+            ]
+        )
+    
+    # Отдельная строка для кнопок с использованием пустой колонки
+    with gr.Row():
+        gr.Button("Жми 👍 для сохранения удачного SMS в базу =>")
+        save_gpt4o_btn = gr.Button("👍")
+        save_gigachat_pro_btn = gr.Button("👍")
+        save_gigachat_lite_btn = gr.Button("👍")
+        save_gigachat_plus_btn = gr.Button("👍")
+
+        # Привязка кнопок к функциям сохранения
+        save_gpt4o_btn.click(fn=lambda x: save_to_github(x, "GPT-4o"), inputs=personalized_output_text_gpt4o, outputs=None)
+        save_gigachat_pro_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Pro"), inputs=personalized_output_text_gigachat_pro, outputs=None)
+        save_gigachat_lite_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite"), inputs=personalized_output_text_gigachat_lite, outputs=None)
+        save_gigachat_plus_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite+"), inputs=personalized_output_text_gigachat_plus, outputs=None)
+
+
+    # Использование сохраненных переменных в следующем блоке
+    with gr.Row():
+        check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2)", elem_id="check_errors_button")
+        
+    with gr.Row():
+        check_errors_btn.click(
+            check_errors_with_yield,
+            inputs=[personalized_output_text_gpt4o, personalized_output_text_gigachat_pro, personalized_output_text_gigachat_lite, personalized_output_text_gigachat_plus],
+            outputs=[
+                gr.Textbox(label="Промпт для проверки текста", lines=6, interactive=False),
+                gr.Textbox(label="Результат проверки GPT-4o", lines=6),
+                gr.Textbox(label="Результат проверки GigaChat-Pro", lines=6),
+                gr.Textbox(label="Результат проверки GigaChat-Lite", lines=6),
+                gr.Textbox(label="Результат проверки GigaChat-Lite+", lines=6)
+            ]
+        )
+
+
+    demo.launch()