Update app.py

app.py

@@ -53,7 +53,7 @@ for sheet_name, df in data.items():
 current_request_index = -1  # Изначально указывает на последний запрос
 
 
-def save_user_request_to_github(description, advantages, personalization_params):
+def save_user_request_to_github(description, advantages, key_message, approach, personalization_params):
 
     global current_request_index  # Используем глобальную переменную
     current_request_index = -1  # Сбрасываем позицию к последнему запросу
@@ -62,6 +62,8 @@ def save_user_request_to_github(description, advantages, personalization_params)
     data_to_save = {
         "description": description,
         "advantages": advantages,
+        "key_message": key_message,
+        "approach": approach,
         "personalization_params": personalization_params,
         "timestamp": time.time()
     }
@@ -108,7 +110,7 @@ def load_previous_user_request_from_github():
 
         if not json_files:
             print("Нет сохраненных запросов.")
-            return "", "", None, None, None, None, None, None
+            return "", "", "", "", None, None, None, None, None, None
 
         # Определяем новый индекс для загрузки предыдущего файла
         current_request_index -= 1
@@ -128,20 +130,30 @@ def load_previous_user_request_from_github():
             description = data.get('description', "")
             advantages = data.get('advantages', "")
             personalization_params = data.get('personalization_params', [None] * 6)  # Убедитесь, что размер списка соответствует количеству полей
+            key_message = data.get('key_message', "")
+            approach = data.get('approach', "")
 
             # Возвращаем данные по отдельности для каждого компонента Gradio
-            return description, advantages, *personalization_params
+             return description, advantages, key_message, approach, *personalization_params
         else:
             print(f"Ошибка при загрузке файла: {file_response.status_code}")
-            return "", "", None, None, None, None, None, None
+            return "", "", "", "", None, None, None, None, None, None
     else:
         print(f"Ошибка при обращении к GitHub: {response.status_code}")
-        return "", "", None, None, None, None, None, None
+        return "", "", "", "", None, None, None, None, None, None
 
 
 # Функция для генерации стандартного промпта
-def generate_standard_prompt(description, advantages, *selected_values):
-    prompt = (
+def generate_standard_prompt(description, advantages, key_message, approach, *selected_values):
+
+    if approach == "Призыв к действию":
+        prompt = "Сгенерируй смс-сообщение для клиента. Начни сообщение с призыва к действию с продуктом.\n"
+    elif approach == "Указание на пользу":
+        prompt = "Сгенерируй смс-сообщение для клиента. Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.\n"
+    elif approach == "Вопрос":
+        prompt = "Сгенерируй смс-сообщение для клиента. Начни сообщение с вопроса, который указывает на пользу продукта для клиента.\n"
+    
+    prompt += (
         "Сгенерируй смс-сообщение для клиента.\n"
         "Начни сообщение с призыва к действию с продуктом.\n"
         f"Описание предложения: {description}\n"
@@ -176,6 +188,10 @@ def generate_standard_prompt(description, advantages, *selected_values):
         "- Речевые клише, рекламные штампы, канцеляризмы;\n"
         "Убедись, что в готовом тексте до 250 знаков с пробелами."
     )
+
+    if key_message.strip():
+        prompt += f"Убедись, что в сообщении есть следующая ключевая информация: {key_message.strip()}"
+        
     return prompt
 
 # Функции для генерации сообщений
@@ -881,6 +897,19 @@ with gr.Blocks() as demo:
                     "Мгновенные переводы на карты любых банков."
                 )
             )
+
+            key_message_input = gr.Textbox(
+                label="Ключевое сообщение (предзаполненный пример можно поменять на свой)",
+                lines=3,
+                value="Бесплатное обслуживание при покупках от 100 000 рублей в месяц."
+            )
+            
+            approach_input = gr.Dropdown(
+                label="Подход",
+                choices=["Призыв к действию", "Указание на пользу", "Вопрос"],
+                value="Призыв к действию"  # Default value
+            )
+            
             selections = []
             for feature in features.keys():
                 if feature not in ["Пол Поколение Психотип"]:  # Исключаем этот лист из выбора
@@ -1018,7 +1047,7 @@ with gr.Blocks() as demo:
     
     submit_btn.click(
         generate_messages, 
-        inputs=[description_input, advantages_input] + selections, 
+        inputs=[description_input, advantages_input, key_message_input, approach_input] + selections, 
         outputs=[
             prompt_display, 
             output_text_gpt4o, 
@@ -1086,7 +1115,9 @@ with gr.Blocks() as demo:
             output_text_gemma_9b,
             output_text_mistral,  # Входные данные для персонализации
             description_input,
-            advantages_input
+            advantages_input,
+            key_message_input,  # Add key_message_input
+            approach_input,     # Add approach_input
         ] + selections,
         outputs=[
             personalization_prompt,
@@ -1103,7 +1134,6 @@ with gr.Blocks() as demo:
         ]
     )
 
-
     # Обработка клика по кнопке восстановления
     load_btn.click(
         fn=lambda: load_previous_user_request_from_github(),