Update app.py

app.py

@@ -9,31 +9,63 @@ from langchain_community.chat_models.gigachat import GigaChat
 from openpyxl import load_workbook
 import base64
 
-# Установка ключа API для OpenAI и GigaChat
+# Установка ключей API для OpenAI, GigaChat и GitHub
 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', max_tokens=68, verify_ssl_certs=False)
-chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, verify_ssl_certs=False)
-chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, verify_ssl_certs=False)
+# Класс конфигурации модели
+class ModelConfig:
+    def __init__(self, temperature):
+        self.max_tokens = 68  # фиксированное значение
+        self.temperature = temperature
+
+# Класс для генерации сообщений
+class MessageGenerator:
+    def __init__(self, config: ModelConfig):
+        self.config = config
+        self.chat_pro = self.authenticate_gigachat('GigaChat-Pro')
+        self.chat_lite = self.authenticate_gigachat('GigaChat')
+        self.chat_plus = self.authenticate_gigachat('GigaChat-Plus')
+
+    def authenticate_gigachat(self, model):
+        return GigaChat(credentials=gc_key, model=model, max_tokens=self.config.max_tokens, temperature=self.config.temperature, verify_ssl_certs=False)
+
+    def clean_message(self, message):
+        if not message.endswith(('.', '!', '?')):
+            last_period = message.rfind('.')
+            if last_period != -1:
+                message = message[:last_period + 1]
+        return message
+
+    def generate_message_gpt4o(self, prompt):
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {openai_api_key}"
+        }
+        data = {
+            "model": "chatgpt-4o-latest",
+            "messages": [{"role": "system", "content": prompt}],
+            "max_tokens": self.config.max_tokens,
+            "temperature": self.config.temperature
+        }
+        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
+        response_data = response.json()
+        return self.clean_message(response_data["choices"][0]["message"]["content"].strip())
+
+    def generate_message_gigachat(self, prompt, model):
+        messages = [SystemMessage(content=prompt)]
+        if model == 'GigaChat-Pro':
+            res = self.chat_pro(messages)
+        elif model == 'GigaChat':
+            res = self.chat_lite(messages)
+        elif model == 'GigaChat-Plus':
+            res = self.chat_plus(messages)
+        return self.clean_message(res.content.strip())
 
 # Загрузка данных из 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] = {}
+data = pd.read_excel('Признаки.xlsx', sheet_name=None)
+features = {sheet_name: df.set_index(df.columns[0]).to_dict()[df.columns[1]] for sheet_name, df in data.items()}
 
 # Функция для генерации стандартного промпта
 def generate_standard_prompt(description, advantages, *selected_values):
@@ -75,279 +107,65 @@ def generate_standard_prompt(description, advantages, *selected_values):
     )
     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": 101
-        }
-        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 clean_message(message):
-    # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
-    if not message.endswith(('.', '!', '?')):
-        last_period = message.rfind('.')
-        if last_period != -1:
-            message = message[:last_period + 1]
-    return message
-
-# Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
-def generate_message_gigachat_pro(prompt):
-    try:
-        messages = [SystemMessage(content=prompt)]
-        res = chat_pro(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    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)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    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)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Plus: {e}"
-
-def generate_message_gpt4o_with_retry(prompt):
-    for _ in range(10):  # Максимум 10 попыток
-        message = generate_message_gpt4o(prompt)
-        if len(message) <= 250:
-            return message
-    return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
-
-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_message_gigachat_lite_with_retry(prompt):
+# Функция для генерации сообщений с учетом повторных попыток
+def generate_message_with_retry(generator, prompt, model):
     for _ in range(10):
-        message = generate_message_gigachat_lite(prompt)
+        message = generator.generate_message_gigachat(prompt, model)
         if len(message) <= 250:
             return message
-    return message
-
-def generate_message_gigachat_plus_with_retry(prompt):
-    for _ in range(10):
-        message = generate_message_gigachat_plus(prompt)
-        if len(message) <= 250:
-            return message
-    return message
-
+    return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
-# Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values):
+# Обновление интерфейса для отображения сообщений
+def generate_messages(description, advantages, config, *selected_values):
+    generator = MessageGenerator(config)
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
         "prompt": standard_prompt,
-        "gpt4o": None,
-        "gigachat_pro": None,
-        "gigachat_lite": None,
-        "gigachat_plus": None
+        "gpt4o": generator.generate_message_gpt4o(standard_prompt),
+        "gigachat_pro": generate_message_with_retry(generator, standard_prompt, 'GigaChat-Pro'),
+        "gigachat_lite": generate_message_with_retry(generator, standard_prompt, 'GigaChat'),
+        "gigachat_plus": generate_message_with_retry(generator, standard_prompt, 'GigaChat-Plus')
     }
 
-    yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
-
-    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt)
-    gpt4o_length = len(results["gpt4o"])
-    gpt4o_display = f"{results['gpt4o']}\n\n------\nКоличество знаков: {gpt4o_length}"
-    yield results["prompt"], gpt4o_display, "", "", "", "Сообщение GPT-4o сгенерировано"
-
-    results["gigachat_pro"] = generate_message_gigachat_pro_with_retry(standard_prompt)
-    gigachat_pro_length = len(results["gigachat_pro"])
-    gigachat_pro_display = f"{results['gigachat_pro']}\n\n------\nКоличество знаков: {gigachat_pro_length}"
-    yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "Сообщение GigaChat-Pro сгенерировано"
-
-    time.sleep(2)
-
-    results["gigachat_lite"] = generate_message_gigachat_lite_with_retry(standard_prompt)
-    gigachat_lite_length = len(results["gigachat_lite"])
-    gigachat_lite_display = f"{results['gigachat_lite']}\n\n------\nКоличество знаков: {gigachat_lite_length}"
-    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Сообщение GigaChat-Lite сгенерировано"
-
-    time.sleep(2)
-
-    results["gigachat_plus"] = generate_message_gigachat_plus_with_retry(standard_prompt)
-    gigachat_plus_length = len(results["gigachat_plus"])
-    gigachat_plus_display = f"{results['gigachat_plus']}\n\n------\nКоличество знаков: {gigachat_plus_length}"
-    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все сообщения сгенерированы"
-
-    return results
-
-
+    return results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"]
 
 # Функция для генерации персонализированного промпта
 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 += f"{features[feature][selected_values[i]]}\n"
     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):
+def perform_personalization(standard_message, personalization_prompt, generator):
     full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
-    if model == "gigachat_pro":
-        result = generate_message_gigachat_pro(full_prompt)
-    elif model == "gigachat_lite":
-        result = generate_message_gigachat_lite(full_prompt)
-    elif model == "gigachat_plus":
-        result = generate_message_gigachat_plus(full_prompt)
-    return clean_message(result)
-
-def perform_personalization_with_retry(standard_message, personalization_prompt):
-    for _ in range(10):  # Максимум 10 попыток
-        message = perform_personalization(standard_message, personalization_prompt)
-        if len(message) <= 250:
-            return message
-    return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
+    return generator.generate_message_gpt4o(full_prompt)
 
-def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model):
+# Функция для выполнения персонализации с учетом повторных попыток
+def perform_personalization_with_retry(standard_message, personalization_prompt, generator, model):
     for _ in range(10):
-        message = perform_personalization_gigachat(standard_message, personalization_prompt, model)
+        message = perform_personalization(standard_message, personalization_prompt, generator)
         if len(message) <= 250:
             return message
     return message
 
-
-# Обновляем блок персонализации
-def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, *selected_values):
+# Обновление интерфейса для персонализации сообщений
+def personalize_messages(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, config, *selected_values):
+    generator = MessageGenerator(config)
     personalization_prompt = generate_personalization_prompt(*selected_values)
-    yield personalization_prompt, "", "", "", "", "Промпт для персонализации сгенерирован"
-
-    personalized_message_gpt4o = perform_personalization_with_retry(gpt4o_message, personalization_prompt)
-    gpt4o_length = len(personalized_message_gpt4o)
-    gpt4o_display = f"{personalized_message_gpt4o}\n\n------\nКоличество знаков: {gpt4o_length}"
-    yield personalization_prompt, gpt4o_display, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
-
-    personalized_message_gigachat_pro = perform_personalization_gigachat_with_retry(gigachat_pro_message, personalization_prompt, "gigachat_pro")
-    gigachat_pro_length = len(personalized_message_gigachat_pro)
-    gigachat_pro_display = f"{personalized_message_gigachat_pro}\n\n------\n��оличество знаков: {gigachat_pro_length}"
-    yield personalization_prompt, gpt4o_display, gigachat_pro_display, "", "", "Персонализированное сообщение GigaChat-Pro сгенерировано"
-
-    personalized_message_gigachat_lite = perform_personalization_gigachat_with_retry(gigachat_lite_message, personalization_prompt, "gigachat_lite")
-    gigachat_lite_length = len(personalized_message_gigachat_lite)
-    gigachat_lite_display = f"{personalized_message_gigachat_lite}\n\n------\nКоличество знаков: {gigachat_lite_length}"
-    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Персонализированное сообщение GigaChat-Lite сгенерировано"
-
-    personalized_message_gigachat_plus = perform_personalization_gigachat_with_retry(gigachat_plus_message, personalization_prompt, "gigachat_plus")
-    gigachat_plus_length = len(personalized_message_gigachat_plus)
-    gigachat_plus_display = f"{personalized_message_gigachat_plus}\n\n------\nКоличество знаков: {gigachat_plus_length}"
-    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все персонализированные сообщения сгенерированы"
-
-
-# Функция для генерации промпта проверки текста
-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, "Все результаты проверки сгенерированы"
+    personalized_gpt4o = perform_personalization_with_retry(gpt4o_message, personalization_prompt, generator, "GPT-4o")
+    personalized_gigachat_pro = perform_personalization_with_retry(gigachat_pro_message, personalization_prompt, generator, "GigaChat-Pro")
+    personalized_gigachat_lite = perform_personalization_with_retry(gigachat_lite_message, personalization_prompt, generator, "GigaChat")
+    personalized_gigachat_plus = perform_personalization_with_retry(gigachat_plus_message, personalization_prompt, generator, "GigaChat-Plus")
 
+    return personalization_prompt, personalized_gpt4o, personalized_gigachat_pro, personalized_gigachat_lite, personalized_gigachat_plus
 
+# Функция для сохранения данных на GitHub
 def save_to_github(personalized_message, model_name, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form):
-    # Собираем все данные в один словарь
     data_to_save = {
         "Модель": model_name,
         "Персонализированное сообщение": personalized_message,
@@ -364,11 +182,7 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
         "Отрасль": industry,
         "ОПФ": legal_form
     }
-
-    # Преобразуем контент в JSON-строку и кодируем в base64
     file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
-
-    # Параметры для GitHub API
     repo = "fruitpicker01/Storage_1"
     path = f"file_{int(time.time())}.json"
     url = f"https://api.github.com/repos/{repo}/contents/{path}"
@@ -380,10 +194,7 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
         "message": f"Добавлен новый файл {path}",
         "content": file_content_encoded
     }
-
-    # Отправка POST-запроса на GitHub API для создания файла в репозитории
-    response = requests.put(url, headers=headers, data=json.dumps(data))
-
+    requests.put(url, headers=headers, data=json.dumps(data))
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -391,64 +202,53 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column(scale=1):
+            temperature_slider = gr.Slider(label="Температура", minimum=0, maximum=2, step=0.01, value=1)
+
             description_input = gr.Textbox(
-                label="Описание предложения (предзаполненный пример можно поменять на свой)", 
-                lines=13,
-                value=(
-                    "Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered. "
-                    "Обслуживание карты стоит 700 рублей в месяц, но клиент может пользоваться ей бесплатно. "
-                    "Что необходимо сделать, чтобы воспользоваться предложением:\n"
-                    "1. Оформить премиальную бизнес-карту в офисе банка или онлайн в интернет-банке СберБизнес.\n"
-                    "2. Забрать карту.\n"
-                    "3. В течение календарного месяца совершить по ней покупки на сумму от 100 000 рублей.\n"
-                    "4. В течение следующего месяца пользоваться ей бесплатно."
-                )
+                label="Описание предложения", 
+                lines=5,
+                value="Необходимо предложить клиенту оформить дебетовую премиальную бизнес-карту Mastercard Preffered..."
             )
             advantages_input = gr.Textbox(
-                label="Преимущества (предзаполненный пример можно поменять на свой)", 
-                lines=6,
-                value=(
-                    "Предложение по бесплатному обслуживанию — бессрочное.\n"
-                    "Оплата покупок без отчётов и платёжных поручений.\n"
-                    "Платёжные документы без комиссии.\n"
-                    "Лимиты на расходы сотрудников.\n"
-                    "Мгновенные переводы на карты любых банков."
-                )
+                label="Преимущества", 
+                lines=5,
+                value="Предложение по бесплатному обслуживанию — бессрочное..."
             )
+
             selections = []
             for feature in features.keys():
                 selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
 
-            submit_btn = gr.Button("1. Создать неперсонализированное сообщение")  # Оранжевая кнопка по умолчанию
+            submit_btn = gr.Button("Создать неперсонализированное сообщение")
 
         with gr.Column(scale=2):
-            prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=20, interactive=False)
+            prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=10, 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)
-            
+            output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Plus", lines=3, interactive=False)
+
             submit_btn.click(
-                generate_messages, 
-                inputs=[description_input, advantages_input] + selections, 
+                fn=generate_messages, 
+                inputs=[description_input, advantages_input, gr.State(lambda: ModelConfig(temperature_slider.value))] + 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)
-            ]
-        )
+        personalize_btn = gr.Button("Выполнить персонализацию")
+
+        with gr.Column(scale=2):
+            personalization_prompt_display = 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-Plus", lines=6, interactive=False)
+
+            personalize_btn.click(
+                fn=personalize_messages,
+                inputs=[output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, gr.State(lambda: ModelConfig(temperature_slider.value))] + selections,
+                outputs=[personalization_prompt_display, personalized_output_text_gpt4o, personalized_output_text_gigachat_pro, personalized_output_text_gigachat_lite, personalized_output_text_gigachat_plus]
+            )
 
     with gr.Row():
         gr.Markdown("*Комментарий (опционально):*")
@@ -464,8 +264,6 @@ with gr.Blocks() as demo:
         corrected_gigachat_lite = gr.Textbox(label="", lines=3)
         corrected_gigachat_plus = gr.Textbox(label="", lines=3)
 
-    
-    # Отдельная строка для кнопок с использованием пустой колонки
     with gr.Row():
         gr.Button("Жми 👍 для сохранения удачного SMS в базу =>")
         save_gpt4o_btn = gr.Button("👍")
@@ -473,7 +271,6 @@ with gr.Blocks() as demo:
         save_gigachat_lite_btn = gr.Button("👍")
         save_gigachat_plus_btn = gr.Button("👍")
 
-        # Привязка кнопок к функциям сохранения
         save_gpt4o_btn.click(
             fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
                 save_to_github(personalized_message, "GPT-4o", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
@@ -539,7 +336,7 @@ with gr.Blocks() as demo:
 
         save_gigachat_plus_btn.click(
             fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
-                save_to_github(personalized_message, "GigaChat-Lite+", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+                save_to_github(personalized_message, "GigaChat-Plus", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
             inputs=[
                 personalized_output_text_gigachat_plus,
                 comment_gigachat_plus,
@@ -558,22 +355,4 @@ with gr.Blocks() as demo:
             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()