Update app.py

app.py

@@ -30,18 +30,48 @@ for sheet_name, df in data.items():
         print(f"Ошибка при обработке данных листа {sheet_name}: {e}")
         features[sheet_name] = {}
 
-# Функция для генерации промпта
-def generate_prompt(*selected_values):
-    prompt = "Напиши типичное рекламное SMS-сообщение от ПАО Сбербанк, но с учетом следующих правил (длина сообщения не должна превышать 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]} не найдено в данных."
-    return prompt.strip()
+# Функция для генерации стандартного промпта
+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
 
-# Функция для генерации сообщения с помощью GPT-4o
+# Функция для генерации сообщения с помощью chatgpt-4o-latest
 def generate_message_gpt4o(prompt):
     try:
         headers = {
@@ -49,7 +79,7 @@ def generate_message_gpt4o(prompt):
             "Authorization": f"Bearer {openai_api_key}"
         }
         data = {
-            "model": "gpt-4o",
+            "model": "chatgpt-4o-latest",
             "messages": [{"role": "system", "content": prompt}],
             "max_tokens": 700
         }
@@ -57,7 +87,7 @@ def generate_message_gpt4o(prompt):
         response_data = response.json()
         return response_data["choices"][0]["message"]["content"].strip()
     except Exception as e:
-        return f"Ошибка при обращении к GPT-4o: {e}"
+        return f"Ошибка при обращении к chatgpt-4o-latest: {e}"
 
 # Функция для генерации сообщения с помощью GigaChat-Pro
 def generate_message_gigachat_pro(prompt):
@@ -79,14 +109,12 @@ def generate_message_gigachat_lite(prompt):
         return f"Ошибка при обращении к GigaChat-Lite: {e}"
 
 # Функция для обработки нажатия кнопки Submit
-def generate_messages(*selected_values):
-    prompt = generate_prompt(*selected_values)
-    if "Ошибка" in prompt:
-        return prompt, "", "", ""
-    message_gpt4o = generate_message_gpt4o(prompt)
-    message_gigachat_pro = generate_message_gigachat_pro(prompt)
-    message_gigachat_lite = generate_message_gigachat_lite(prompt)
-    return prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite
+def generate_messages(description, advantages, *selected_values):
+    standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
+    message_gpt4o = generate_message_gpt4o(standard_prompt)
+    message_gigachat_pro = generate_message_gigachat_pro(standard_prompt)
+    message_gigachat_lite = generate_message_gigachat_lite(standard_prompt)
+    return standard_prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -94,6 +122,8 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column(scale=1):
+            description_input = gr.Textbox(label="Описание предложения", lines=3)
+            advantages_input = gr.Textbox(label="Преимущества", lines=3)
             selections = []
             for feature in features.keys():
                 selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
@@ -101,11 +131,15 @@ with gr.Blocks() as demo:
             submit_btn = gr.Button("Submit")  # Оранжевая кнопка по умолчанию
 
         with gr.Column(scale=2):
-            prompt_display = gr.Textbox(label="Сгенерированный промпт", lines=4, interactive=False)
+            prompt_display = gr.Textbox(label="Стандартный промпт", lines=15, interactive=False)
             output_text_gpt4o = gr.Textbox(label="Создание сообщения GPT-4o", lines=3)
             output_text_gigachat_pro = gr.Textbox(label="Создание сообщения GigaChat-Pro", lines=3)
             output_text_gigachat_lite = gr.Textbox(label="Создание сообщения GigaChat-Lite", lines=3)
             
-            submit_btn.click(generate_messages, selections, [prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite])
+            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]
+            )
 
     demo.launch()