Update app.py

app.py

@@ -14,22 +14,10 @@ openai_api_key = os.getenv('GPT_KEY')
 gc_key = os.getenv('GC_KEY')
 token = os.getenv('GITHUB_TOKEN')
 
-# Функция аутентификации в сервисе GigaChat с заданной температурой
-def authenticate_gigachat(model, max_tokens, temperature):
-    return GigaChat(credentials=gc_key, model=model, max_tokens=max_tokens, temperature=float(temperature), verify_ssl_certs=False)
-
-# Функция аутентификации для GPT-4o
-def authenticate_gpt4o(max_tokens, temperature):
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {openai_api_key}"
-    }
-    return {
-        "model": "chatgpt-4o-latest",
-        "max_tokens": max_tokens,
-        "temperature": float(temperature),
-        "headers": headers
-    }
+# Авторизация в сервисе 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)
 
 # Загрузка данных из Excel-файла
 try:
@@ -87,56 +75,98 @@ def generate_standard_prompt(description, advantages, *selected_values):
     )
     return prompt
 
-# Функция для генерации сообщений GPT-4o
-def generate_message_gpt4o(auth_params, prompt):
+# Функции для генерации сообщений
+def generate_message_gpt4o(prompt, temperature=1):
     try:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {openai_api_key}"
+        }
         data = {
-            "model": auth_params["model"],
+            "model": "chatgpt-4o-latest",
             "messages": [{"role": "system", "content": prompt}],
-            "max_tokens": auth_params["max_tokens"],
-            "temperature": auth_params["temperature"]
+            "max_tokens": 101,
+            "temperature": temperature  # Передача температуры
         }
-        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=auth_params["headers"])
+        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
         response_data = response.json()
-        return clean_message(response_data["choices"][0]["message"]["content"].strip())
+        return clean_message(response_data["choices"][0]["message"]["content"].strip()) + f" {temperature}"
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
-# Функции для генерации сообщений GigaChat
-def generate_message_gigachat(chat_instance, prompt):
-    try:
-        messages = [SystemMessage(content=prompt)]
-        res = chat_instance(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat: {e}"
 
 def clean_message(message):
     # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
     if not message.endswith(('.', '!', '?')):
         last_period = message.rfind('.')
-        if (last_period != -1):
+        if last_period != -1:
             message = message[:last_period + 1]
     return message
 
-# Функции для генерации сообщений с повторными попытками
-def generate_message_gpt4o_with_retry(auth_params, prompt):
+# Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
+def generate_message_gigachat_pro(prompt, temperature=0.87):
+    try:
+        messages = [SystemMessage(content=prompt)]
+        chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, temperature=temperature, verify_ssl_certs=False)
+        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, temperature=0.87):
+    try:
+        time.sleep(2)
+        messages = [SystemMessage(content=prompt)]
+        chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=temperature, verify_ssl_certs=False)
+        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, temperature=0.87):
+    try:
+        time.sleep(2)
+        messages = [SystemMessage(content=prompt)]
+        chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, temperature=temperature, verify_ssl_certs=False)
+        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(auth_params, prompt)
+        message = generate_message_gpt4o(prompt)
         if len(message) <= 250:
             return message
     return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
-def generate_message_gigachat_with_retry(chat_instance, prompt):
+def generate_message_gigachat_pro_with_retry(prompt):
     for _ in range(10):
-        message = generate_message_gigachat(chat_instance, prompt)
+        message = generate_message_gigachat_pro(prompt)
         if len(message) <= 250:
             return message
     return message
 
+def generate_message_gigachat_lite_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_gigachat_lite(prompt)
+        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
+
+
 # Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values, gpt4o_auth_params, chat_pro, chat_lite, chat_plus):
+def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -149,32 +179,224 @@ def generate_messages(description, advantages, *selected_values, gpt4o_auth_para
 
     yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
-    results["gpt4o"] = generate_message_gpt4o_with_retry(gpt4o_auth_params, standard_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_with_retry(chat_pro, standard_prompt)
+    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_with_retry(chat_lite, standard_prompt)
+    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_with_retry(chat_plus, standard_prompt)
+    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
 
+
+
+# Функция для генерации персонализированного промпта
+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":
+        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  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
+
+def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model):
+    for _ in range(10):
+        message = perform_personalization_gigachat(standard_message, personalization_prompt, model)
+        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):
+    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, "Все результаты проверки сгенерированы"
+
+
+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, temperature):
+    # Собираем все данные в один словарь
+    data_to_save = {
+        "Модель": model_name,
+        "Температура": temperature,  # Добавляем температуру
+        "Персонализированное сообщение": personalized_message,
+        "Комментарий": comment,
+        "Откорректированное сообщение": corrected_message,
+        "Описание предложения": description,
+        "Преимущества": advantages,
+        "Неперсонализированный промпт": non_personalized_prompt,
+        "Неперсонализированное сообщение": non_personalized_message,
+        "Пол": gender,
+        "Поколение": generation,
+        "Психотип": psychotype,
+        "Стадия бизнеса": business_stage,
+        "Отрасль": 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}"
+    headers = {
+        "Authorization": f"token {token}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "message": f"Добавлен новый файл {path}",
+        "content": file_content_encoded
+    }
+
+    # Отправка POST-запроса на GitHub API для создания файла в репозитории
+    response = requests.put(url, headers=headers, data=json.dumps(data))
+
+    if response.status_code == 201:
+        print("Файл успешно загружен на GitHub.")
+    else:
+        print(f"Ошибка при загрузке файла на GitHub: {response.status_code}")
+        print(f"Ответ сервера: {response.json()}")
+
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -222,40 +444,168 @@ with gr.Blocks() as demo:
             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-Plus", 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,
-                gpt4o_temperature,  # Передаем температуру как компонент
-                gigachat_pro_temperature,  # Передаем температуру как компонент
-                gigachat_lite_temperature,  # Передаем температуру как компонент
-                gigachat_plus_temperature  # Передаем температуру как компонент
+                *selections, 
+                gpt4o_temperature, 
+                gigachat_pro_temperature, 
+                gigachat_lite_temperature, 
+                gigachat_plus_temperature
             ], 
             outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
         )
 
 
-        submit_btn.click(
-            fn=lambda description, advantages, *selected_values, gpt4o_temperature, gigachat_pro_temperature, gigachat_lite_temperature, gigachat_plus_temperature: 
-                generate_messages(description, advantages, *selected_values,
-                                  authenticate_gpt4o(101, gpt4o_temperature),
-                                  authenticate_gigachat('GigaChat-Pro', 68, gigachat_pro_temperature),
-                                  authenticate_gigachat('GigaChat', 68, gigachat_lite_temperature),
-                                  authenticate_gigachat('GigaChat-Plus', 68, gigachat_plus_temperature)),
+    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.Markdown("*Комментарий (опционально):*")
+        comment_gpt4o = gr.Textbox(label="", lines=3)
+        comment_gigachat_pro = gr.Textbox(label="", lines=3)
+        comment_gigachat_lite = gr.Textbox(label="", lines=3)
+        comment_gigachat_plus = gr.Textbox(label="", lines=3)
+
+    with gr.Row():
+        gr.Markdown("*Откорректирован��ое сообщение (опционально):*")
+        corrected_gpt4o = gr.Textbox(label="", lines=3)
+        corrected_gigachat_pro = gr.Textbox(label="", lines=3)
+        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("👍")
+        save_gigachat_pro_btn = gr.Button("👍")
+        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, temperature: 
+                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, temperature),
             inputs=[
-                description_input, 
-                advantages_input, 
-                *selections,
-                gpt4o_temperature,  # Передаем значение температуры
-                gigachat_pro_temperature,  # Передаем значение температуры
-                gigachat_lite_temperature,  # Передаем значение температуры
-                gigachat_plus_temperature  # Передаем значение температуры
-            ], 
-            outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
+                personalized_output_text_gpt4o,
+                comment_gpt4o,
+                corrected_gpt4o,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gpt4o,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5],  # ОПФ
+                gpt4o_temperature  # Передача температуры
+            ],
+            outputs=None
+        )
+        
+        save_gigachat_pro_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, temperature: 
+                save_to_github(personalized_message, "GigaChat-Pro", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature),
+            inputs=[
+                personalized_output_text_gigachat_pro,
+                comment_gigachat_pro,
+                corrected_gigachat_pro,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gigachat_pro,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5],  # ОПФ
+                gigachat_pro_temperature  # Передача температуры
+            ],
+            outputs=None
+        )
+        
+        save_gigachat_lite_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, temperature: 
+                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, temperature),
+            inputs=[
+                personalized_output_text_gigachat_lite,
+                comment_gigachat_lite,
+                corrected_gigachat_lite,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gigachat_lite,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5],  # ОПФ
+                gigachat_lite_temperature  # Передача температуры
+            ],
+            outputs=None
+        )
+        
+        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, temperature: 
+                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, temperature),
+            inputs=[
+                personalized_output_text_gigachat_plus,
+                comment_gigachat_plus,
+                corrected_gigachat_plus,
+                description_input,
+                advantages_input,
+                prompt_display,
+                output_text_gigachat_plus,
+                selections[0],  # Пол
+                selections[1],  # Поколение
+                selections[2],  # Психотип
+                selections[3],  # Стадия бизнеса
+                selections[4],  # Отрасль
+                selections[5],  # ОПФ
+                gigachat_plus_temperature  # Передача температуры
+            ],
+            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()