Update app.py

app.py

@@ -8,11 +8,21 @@ from langchain.schema import SystemMessage
 from langchain_community.chat_models.gigachat import GigaChat
 from openpyxl import load_workbook
 import base64
+from together import Together
+from mistralai import Mistral
 
-# Установка ключа API для OpenAI и GigaChat
+# Установка ключа API для OpenAI, GigaChat и Mistral
 openai_api_key = os.getenv('GPT_KEY')
 gc_key = os.getenv('GC_KEY')
 token = os.getenv('GITHUB_TOKEN')
+TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
+MISTRAL_API_KEY = os.getenv('MISTRAL_API_KEY')
+
+# Инициализация клиента для Together
+client = Together(api_key=TOGETHER_API_KEY)
+
+# Инициализация клиента для Mistral
+client_mistral = Mistral(api_key=MISTRAL_API_KEY)
 
 # Авторизация в сервисе GigaChat
 chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, verify_ssl_certs=False)
@@ -30,7 +40,11 @@ except Exception as e:
 features = {}
 for sheet_name, df in data.items():
     try:
-        features[sheet_name] = df.set_index(df.columns[0]).to_dict()[df.columns[1]]
+        if sheet_name == "Пол Поколение Психотип":
+            # Создаем словарь, где ключи — это кортежи (Пол, Поколение, Психотип), а значения — инструкции
+            features[sheet_name] = df.set_index(['Пол', 'Поколение', 'Психотип'])['Инструкция'].to_dict()
+        else:
+            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] = {}
@@ -42,7 +56,6 @@ def generate_standard_prompt(description, advantages, *selected_values):
         "Начни сообщение с призыва к действию с продуктом.\n"
         f"Описание предложения: {description}\n"
         f"Преимущества: {advantages}\n"
-        "Вклад на короткий срок.\n"
         "В тексте смс запрещено использование:\n"
         "- Запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, доступный, банкротство, долги, займ, срочно, сейчас, лучший, главный, номер 1, гарантия, успех, лидер;\n"
         "- Обращение к клиенту;\n"
@@ -131,6 +144,88 @@ def generate_message_gigachat_plus(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
+def generate_message_meta_llama_3_1_405b(prompt):
+    try:
+        response = client.chat.completions.create(
+            model="meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=74,
+            temperature=0.7
+        )
+        cleaned_message = clean_message(response.choices[0].message.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к Meta-Llama-3.1-405B: {e}"
+
+def generate_message_meta_llama_3_1_70b(prompt):
+    try:
+        response = client.chat.completions.create(
+            model="meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=74,
+            temperature=0.7
+        )
+        cleaned_message = clean_message(response.choices[0].message.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к Meta-Llama-3.1-70B: {e}"
+
+def generate_message_meta_llama_3_1_8b(prompt):
+    try:
+        response = client.chat.completions.create(
+            model="meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=74,
+            temperature=0.7
+        )
+        cleaned_message = clean_message(response.choices[0].message.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к Meta-Llama-3.1-8B: {e}"
+
+def generate_message_gemma_2_27b_it(prompt):
+    try:
+        response = client.chat.completions.create(
+            model="google/gemma-2-27b-it",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=74,
+            temperature=0.7
+        )
+        cleaned_message = clean_message(response.choices[0].message.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к Gemma-2-27b-it: {e}"
+
+def generate_message_gemma_2_9b_it(prompt):
+    try:
+        response = client.chat.completions.create(
+            model="google/gemma-2-9b-it",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=74,
+            temperature=0.7
+        )
+        cleaned_message = clean_message(response.choices[0].message.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к Gemma-2-9b-it: {e}"
+
+def generate_message_mistral(prompt):
+    try:
+        chat_response = client_mistral.chat.complete(
+            model="mistral-large-latest",
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                    "max_tokens": 74
+                },
+            ]
+        )
+        cleaned_message = clean_message(chat_response.choices[0].message.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к Mistral: {e}"
+
 def generate_message_gpt4o_with_retry(prompt):
     for _ in range(10):  # Максимум 10 попыток
         message = generate_message_gpt4o(prompt)
@@ -159,8 +254,49 @@ def generate_message_gigachat_plus_with_retry(prompt):
             return message
     return message
 
+def generate_message_meta_llama_3_1_405b_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_meta_llama_3_1_405b(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
+def generate_message_meta_llama_3_1_70b_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_meta_llama_3_1_70b(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
+def generate_message_meta_llama_3_1_8b_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_meta_llama_3_1_8b(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
+def generate_message_gemma_2_27b_it_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_gemma_2_27b_it(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
+def generate_message_gemma_2_9b_it_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_gemma_2_9b_it(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
+def generate_message_mistral_with_retry(prompt):
+    for _ in range(10):
+        message = generate_message_mistral(prompt)
+        if len(message) <= 250:
+            return message
+    return message
+
 
-# Обновляем генерацию сообщений для отображения в интерфейсе
 def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
@@ -169,108 +305,252 @@ def generate_messages(description, advantages, *selected_values):
         "gpt4o": None,
         "gigachat_pro": None,
         "gigachat_lite": None,
-        "gigachat_plus": None
+        "gigachat_plus": None,
+        "meta_llama_3_1_405b": None,
+        "meta_llama_3_1_70b": None,
+        "meta_llama_3_1_8b": None,
+        "gemma_2_27b_it": None,
+        "gemma_2_9b_it": None,
+        "mistral": None  # Добавляем Mistral
     }
 
-    yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
+    yield results["prompt"], "", "", "", "", "", "", "", "", "", ""
 
+    # Generating messages using existing models (as before)
     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 сгенерировано"
+    yield results["prompt"], gpt4o_display, "", "", "", "", "", "", "", "", ""
 
     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 сгенерировано"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "", "", "", "", "", ""
 
     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 сгенерировано"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "", "", "", "", "", ""
 
     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, "Все сообщения сгенерированы"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "", "", "", "", "", ""
 
-    return results
+    time.sleep(2)
+
+    results["meta_llama_3_1_405b"] = generate_message_meta_llama_3_1_405b_with_retry(standard_prompt)
+    meta_llama_405b_length = len(results["meta_llama_3_1_405b"])
+    meta_llama_405b_display = f"{results['meta_llama_3_1_405b']}\n\n------\nКоличество знаков: {meta_llama_405b_length}"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, "", "", "", "", ""
+
+    time.sleep(4)
+
+    results["meta_llama_3_1_70b"] = generate_message_meta_llama_3_1_70b_with_retry(standard_prompt)
+    meta_llama_70b_length = len(results["meta_llama_3_1_70b"])
+    meta_llama_70b_display = f"{results['meta_llama_3_1_70b']}\n\n------\nКоличество знаков: {meta_llama_70b_length}"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, "", "", "", ""
+
+    time.sleep(4)
 
+    results["meta_llama_3_1_8b"] = generate_message_meta_llama_3_1_8b_with_retry(standard_prompt)
+    meta_llama_8b_length = len(results["meta_llama_3_1_8b"])
+    meta_llama_8b_display = f"{results['meta_llama_3_1_8b']}\n\n------\nКоличество знаков: {meta_llama_8b_length}"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, "", "", ""
+
+    time.sleep(4)
+
+    results["gemma_2_27b_it"] = generate_message_gemma_2_27b_it_with_retry(standard_prompt)
+    gemma_27b_length = len(results["gemma_2_27b_it"])
+    gemma_27b_display = f"{results['gemma_2_27b_it']}\n\n------\nКоличество знаков: {gemma_27b_length}"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, gemma_27b_display, "", ""
+
+    time.sleep(4)
+
+    results["gemma_2_9b_it"] = generate_message_gemma_2_9b_it_with_retry(standard_prompt)
+    gemma_9b_length = len(results["gemma_2_9b_it"])
+    gemma_9b_display = f"{results['gemma_2_9b_it']}\n\n------\nКоличество знаков: {gemma_9b_length}"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, gemma_27b_display, gemma_9b_display, ""
+
+    time.sleep(4)
+
+    # Добавляем Mistral
+    results["mistral"] = generate_message_mistral_with_retry(standard_prompt)
+    mistral_length = len(results["mistral"])
+    mistral_display = f"{results['mistral']}\n\n------\nКоличество знаков: {mistral_length}"
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, gemma_27b_display, gemma_9b_display, mistral_display
+
+    time.sleep(4)
+
+    return results
 
 
 # Функция для генерации персонализированного промпта
 def generate_personalization_prompt(*selected_values):
     prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
+    gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
+    combined_instruction = ""
+    additional_instructions = ""
+
+    print(f"Выбранные значения: Пол={gender}, Поколение={generation}, Психотип={psychotype}")
+
+    # Проверяем, выбраны ли все три параметра: Пол, Поколение, Психотип
+    if gender and generation and psychotype:
+        # Получаем данные с листа "Пол Поколение Психотип"
+        sheet = features.get("Пол Поколение Психотип", {})
+
+        # Ищем ключ, соответствующий комбинации "Пол", "Поколение", "Психотип"
+        key = (gender, generation, psychotype)
+        if key in sheet:
+            combined_instruction = sheet[key]
+            print(f"Найдена комбинированная инструкция: {combined_instruction}")
+        else:
+            print(f"Комбинированная инструкция для ключа {key} не найдена.")
+
+    # Если не найдена комбинированная инструкция, добавляем индивидуальные инструкции
+    if not combined_instruction:
+        print("Добавляем индивидуальные инструкции для Пол, Поколение, Психотип.")
+        for i, feature in enumerate(["Пол", "Поколение", "Психотип"]):
+            if selected_values[i]:
+                try:
+                    instruction = features[feature][selected_values[i]]
+                    additional_instructions += f"{instruction}\n"
+                    print(f"Добавлена инструкция из {feature}: {instruction}")
+                except KeyError:
+                    return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
+
+    # Добавляем инструкции для остальных параметров (например, Отрасль)
     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]} не найдено в данных."
-                
+        if feature not in ["Пол", "Поколение", "Психотип", "Пол Поколение Психотип"]:
+            if i < len(selected_values) and selected_values[i]:
+                try:
+                    instruction = features[feature][selected_values[i]]
+                    additional_instructions += f"{instruction}\n"
+                    print(f"Добавлена инструкция из {feature}: {instruction}")
+                except KeyError:
+                    return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
+
+    # Формируем итоговый промпт
+    if combined_instruction:
+        prompt += combined_instruction  # Добавляем комбинированную инструкцию, если она есть
+    if additional_instructions:
+        prompt += additional_instructions  # Добавляем остальные инструкции
+
     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)
+    return generate_message_gpt4o_with_retry(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)
+        result = generate_message_gigachat_pro_with_retry(full_prompt)
     elif model == "gigachat_lite":
-        result = generate_message_gigachat_lite(full_prompt)
+        result = generate_message_gigachat_lite_with_retry(full_prompt)
     elif model == "gigachat_plus":
-        result = generate_message_gigachat_plus(full_prompt)
+        result = generate_message_gigachat_plus_with_retry(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_meta_llama_405b(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_meta_llama_3_1_405b_with_retry(full_prompt)
 
-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 perform_personalization_meta_llama_70b(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_meta_llama_3_1_70b_with_retry(full_prompt)
+
+def perform_personalization_meta_llama_8b(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_meta_llama_3_1_8b_with_retry(full_prompt)
+
+def perform_personalization_gemma_27b_it(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_gemma_2_27b_it_with_retry(full_prompt)
 
+def perform_personalization_gemma_9b_it(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_gemma_2_9b_it_with_retry(full_prompt)
 
-# Обновляем блок персонализации
-def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, *selected_values):
+def perform_personalization_mistral(standard_message, personalization_prompt):
+    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
+    return generate_message_mistral_with_retry(full_prompt)
+
+# Updated function to include additional models in personalization
+def personalize_messages_with_yield(
+    gpt4o_message, 
+    gigachat_pro_message, 
+    gigachat_lite_message, 
+    gigachat_plus_message, 
+    meta_llama_405b_message, 
+    meta_llama_70b_message, 
+    meta_llama_8b_message, 
+    gemma_27b_message, 
+    gemma_9b_message, 
+    mistral_message,
+    *selected_values
+):
     personalization_prompt = generate_personalization_prompt(*selected_values)
-    yield personalization_prompt, "", "", "", "", "Промпт для персонализации сгенерирован"
+    yield personalization_prompt, "", "", "", "", "", "", "", "", "", ""
 
-    personalized_message_gpt4o = perform_personalization_with_retry(gpt4o_message, personalization_prompt)
+    personalized_message_gpt4o = perform_personalization(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 сгенерировано"
+    yield personalization_prompt, gpt4o_display, "", "", "", "", "", "", "", "", ""
 
-    personalized_message_gigachat_pro = perform_personalization_gigachat_with_retry(gigachat_pro_message, personalization_prompt, "gigachat_pro")
+    personalized_message_gigachat_pro = perform_personalization_gigachat(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 сгенерировано"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, "", "", "", "", "", "", "", ""
 
-    personalized_message_gigachat_lite = perform_personalization_gigachat_with_retry(gigachat_lite_message, personalization_prompt, "gigachat_lite")
+    personalized_message_gigachat_lite = perform_personalization_gigachat(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 сгенерировано"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "", "", "", "", "", ""
 
-    personalized_message_gigachat_plus = perform_personalization_gigachat_with_retry(gigachat_plus_message, personalization_prompt, "gigachat_plus")
+    personalized_message_gigachat_plus = perform_personalization_gigachat(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, "Все персонализированные сообщения сгенерированы"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "", "", "", "", "", ""
+
+    personalized_message_meta_llama_405b = perform_personalization_meta_llama_405b(meta_llama_405b_message, personalization_prompt)
+    meta_llama_405b_length = len(personalized_message_meta_llama_405b)
+    meta_llama_405b_display = f"{personalized_message_meta_llama_405b}\n\n------\nКоличество знаков: {meta_llama_405b_length}"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, "", "", "", "", ""
+
+    personalized_message_meta_llama_70b = perform_personalization_meta_llama_70b(meta_llama_70b_message, personalization_prompt)
+    meta_llama_70b_length = len(personalized_message_meta_llama_70b)
+    meta_llama_70b_display = f"{personalized_message_meta_llama_70b}\n\n------\nКоличество знаков: {meta_llama_70b_length}"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, "", "", "", ""
+
+    personalized_message_meta_llama_8b = perform_personalization_meta_llama_8b(meta_llama_8b_message, personalization_prompt)
+    meta_llama_8b_length = len(personalized_message_meta_llama_8b)
+    meta_llama_8b_display = f"{personalized_message_meta_llama_8b}\n\n------\nКоличество знаков: {meta_llama_8b_length}"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, "", "", ""
+
+    personalized_message_gemma_27b_it = perform_personalization_gemma_27b_it(gemma_27b_message, personalization_prompt)
+    gemma_27b_length = len(personalized_message_gemma_27b_it)
+    gemma_27b_display = f"{personalized_message_gemma_27b_it}\n\n------\nКоличество знаков: {gemma_27b_length}"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, gemma_27b_display, "", ""
+
+    personalized_message_gemma_9b_it = perform_personalization_gemma_9b_it(gemma_9b_message, personalization_prompt)
+    gemma_9b_length = len(personalized_message_gemma_9b_it)
+    gemma_9b_display = f"{personalized_message_gemma_9b_it}\n\n------\nКоличество знаков: {gemma_9b_length}"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, gemma_27b_display, gemma_9b_display, ""
+
+    personalized_message_mistral = perform_personalization_mistral(mistral_message, personalization_prompt)
+    mistral_length = len(personalized_message_mistral)
+    mistral_display = f"{personalized_message_mistral}\n\n------\nКоличество знаков: {mistral_length}"
+    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_display, meta_llama_70b_display, meta_llama_8b_display, gemma_27b_display, gemma_9b_display, mistral_display
 
 
 # Функция для генерации промпта проверки текста
@@ -320,30 +600,42 @@ def generate_error_check_prompt():
 
 # Функция для выполнения проверки текста с использованием yield
 def check_errors_with_yield(*personalized_messages):
-    if len(personalized_messages) < 4:
-        yield "", "", "", "", "", "Ошибка: недостаточно сообщений для проверки"
+    if len(personalized_messages) < 10:  # Adjusted for the inclusion of Mistral
+        yield "", "", "", "", "", "", "", "", "", "", "Ошибка: недостаточно сообщений для проверки"
         return
 
     error_check_prompt = generate_error_check_prompt()
-    yield 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 сгенерирован"
+    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 сгенерирован"
+    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 сгенерирован"
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, "", "", "", "", "", "", "Результат проверки GigaChat-Lite сгенерирован"
+
+    error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[3]}", "", "gigachat_plus")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "", "", "", "", "", "Результат проверки GigaChat-Plus сгенерирован"
     
-    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, "Все результаты проверки сгенерированы"
+    error_message_meta_llama_405b = perform_personalization(f"{error_check_prompt}\n\n{personalized_messages[4]}", "")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, error_message_meta_llama_405b, "", "", "", "", "Результат проверки Meta-Llama-3.1-405B сгенерирован"
+
+    error_message_meta_llama_70b = perform_personalization_meta_llama_70b(f"{error_check_prompt}\n\n{personalized_messages[5]}", "")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, error_message_meta_llama_405b, error_message_meta_llama_70b, "", "", "", "Результат проверки Meta-Llama-3.1-70B сгенерирован"
+
+    error_message_meta_llama_8b = perform_personalization_meta_llama_8b(f"{error_check_prompt}\n\n{personalized_messages[6]}", "")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, error_message_meta_llama_405b, error_message_meta_llama_70b, error_message_meta_llama_8b, "", "", "Результат проверки Meta-Llama-3.1-8B сгенерирован"
+
+    error_message_gemma_27b_it = perform_personalization_gemma_27b_it(f"{error_check_prompt}\n\n{personalized_messages[7]}", "")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, error_message_meta_llama_405b, error_message_meta_llama_70b, error_message_meta_llama_8b, error_message_gemma_27b_it, "", "Результат проверки Gemma-2-27B-IT сгенерирован"
+
+    error_message_gemma_9b_it = perform_personalization_gemma_9b_it(f"{error_check_prompt}\n\n{personalized_messages[8]}", "")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, error_message_meta_llama_405b, error_message_meta_llama_70b, error_message_meta_llama_8b, error_message_gemma_27b_it, error_message_gemma_9b_it, "", "Результат проверки Gemma-2-9B-IT сгенерирован"
+
+    error_message_mistral = perform_personalization_mistral(f"{error_check_prompt}\n\n{personalized_messages[9]}", "")
+    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, error_message_meta_llama_405b, error_message_meta_llama_70b, error_message_meta_llama_8b, error_message_gemma_27b_it, error_message_gemma_9b_it, error_message_mistral, "Все результаты проверки сгенерированы"
 
 
 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):
@@ -369,7 +661,7 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
     file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
 
     # Параметры для GitHub API
-    repo = "fruitpicker01/Storage_1"
+    repo = "fruitpicker01/Storage_Kate"
     path = f"file_{int(time.time())}.json"
     url = f"https://api.github.com/repos/{repo}/contents/{path}"
     headers = {
@@ -388,7 +680,7 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
-    
+
     with gr.Row():
         with gr.Column(scale=1):
             description_input = gr.Textbox(
@@ -417,163 +709,357 @@ with gr.Blocks() as demo:
             )
             selections = []
             for feature in features.keys():
-                selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
-
-            submit_btn = gr.Button("1. Создать неперсонализированное сообщение")  # Оранжевая кнопка по умолчанию
+                if feature not in ["Пол Поколение Психотип"]:  # Исключаем этот лист из выбора
+                    selections.append(gr.Dropdown(choices=[None] + list(features[feature].keys()), label=f"Выберите {feature}"))
 
         with gr.Column(scale=2):
-            prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=20, 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]
-            )
+            prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=25, interactive=False)
+            personalization_prompt = gr.Textbox(label="Персонализированный промпт", lines=24, interactive=False)
 
     with gr.Row():
+        submit_btn = gr.Button("1. Создать неперсонализированное сообщение")
         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)
-            ]
-        )
+        output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3, interactive=False)
+        personalized_output_text_gpt4o = gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=3, interactive=False)
+        comment_gpt4o = gr.Textbox(label="Комментарий к сообщению GPT-4o", lines=3)
+        corrected_gpt4o = gr.Textbox(label="Откорректированное сообщение GPT-4o", lines=3)
+        save_gpt4o_btn = gr.Button("👍 GPT-4o")
 
+    # Второй ряд
     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)
+        output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3, interactive=False)
+        personalized_output_text_gigachat_pro = gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=3, interactive=False)
+        comment_gigachat_pro = gr.Textbox(label="Комментарий к сообщению GigaChat-Pro", lines=3)
+        corrected_gigachat_pro = gr.Textbox(label="Откорректированное сообщение GigaChat-Pro", lines=3)
+        save_gigachat_pro_btn = gr.Button("👍 GigaChat-Pro")
 
+    # Третий ряд
     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)
+        output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3, interactive=False)
+        personalized_output_text_gigachat_lite = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=3, interactive=False)
+        comment_gigachat_lite = gr.Textbox(label="Комментарий к сообщению GigaChat-Lite", lines=3)
+        corrected_gigachat_lite = gr.Textbox(label="Откорректированное сообщение GigaChat-Lite", lines=3)
+        save_gigachat_lite_btn = gr.Button("👍 GigaChat-Lite")
 
+    # Четвертый ряд
+    with gr.Row():
+        output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3, interactive=False)
+        personalized_output_text_gigachat_plus = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=3, interactive=False)
+        comment_gigachat_plus = gr.Textbox(label="Комментарий к сообщению GigaChat-Lite+", lines=3)
+        corrected_gigachat_plus = gr.Textbox(label="Откорректированное сообщение GigaChat-Lite+", lines=3)
+        save_gigachat_plus_btn = gr.Button("👍 GigaChat-Lite+")
+        
+    # Пятый ряд
+    with gr.Row():
+        output_text_meta_llama_405b = gr.Textbox(label="Неперсонализированное сообщение Meta-Llama-3.1-405B", lines=3, interactive=False)
+        personalized_output_text_meta_llama_405b = gr.Textbox(label="Персонализированное сообщение Meta-Llama-3.1-405B", lines=3, interactive=False)
+        comment_meta_llama_405b = gr.Textbox(label="Комментарий к сообщению Meta-Llama-3.1-405B", lines=3)
+        corrected_meta_llama_405b = gr.Textbox(label="Откорректированное сообщение Meta-Llama-3.1-405B", lines=3)
+        save_meta_llama_405b_btn = gr.Button("👍 Meta-Llama-3.1-405B")
     
-    # Отдельная строка для кнопок с использованием пустой колонки
+    # Шестой ряд
     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: 
-                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),
-            inputs=[
-                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]   # ОПФ
-            ],
-            outputs=None
-        )
+        output_text_meta_llama_70b = gr.Textbox(label="Неперсонализированное сообщение Meta-Llama-3.1-70B", lines=3, interactive=False)
+        personalized_output_text_meta_llama_70b = gr.Textbox(label="Персонализированное сообщение Meta-Llama-3.1-70B", lines=3, interactive=False)
+        comment_meta_llama_70b = gr.Textbox(label="Комментарий к сообщению Meta-Llama-3.1-70B", lines=3)
+        corrected_meta_llama_70b = gr.Textbox(label="Откорректированное сообщение Meta-Llama-3.1-70B", lines=3)
+        save_meta_llama_70b_btn = gr.Button("👍 Meta-Llama-3.1-70B")
+    
+    # Седьмой ряд
+    with gr.Row():
+        output_text_meta_llama_8b = gr.Textbox(label="Неперсонализ��рованное сообщение Meta-Llama-3.1-8B", lines=3, interactive=False)
+        personalized_output_text_meta_llama_8b = gr.Textbox(label="Персонализированное сообщение Meta-Llama-3.1-8B", lines=3, interactive=False)
+        comment_meta_llama_8b = gr.Textbox(label="Комментарий к сообщению Meta-Llama-3.1-8B", lines=3)
+        corrected_meta_llama_8b = gr.Textbox(label="Откорректированное сообщение Meta-Llama-3.1-8B", lines=3)
+        save_meta_llama_8b_btn = gr.Button("👍 Meta-Llama-3.1-8B")
+    
+    # Восьмой ряд
+    with gr.Row():
+        output_text_gemma_27b = gr.Textbox(label="Неперсонализированное сообщение Gemma-2-27B-IT", lines=3, interactive=False)
+        personalized_output_text_gemma_27b = gr.Textbox(label="Персонализированное сообщение Gemma-2-27B-IT", lines=3, interactive=False)
+        comment_gemma_27b = gr.Textbox(label="Комментарий к сообщению Gemma-2-27B-IT", lines=3)
+        corrected_gemma_27b = gr.Textbox(label="Откорректированное сообщение Gemma-2-27B-IT", lines=3)
+        save_gemma_27b_btn = gr.Button("👍 Gemma-2-27B-IT")
+    
+    # Девятый ряд
+    with gr.Row():
+        output_text_gemma_9b = gr.Textbox(label="Неперсонализированное сообщение Gemma-2-9B-IT", lines=3, interactive=False)
+        personalized_output_text_gemma_9b = gr.Textbox(label="Персонализированное сообщение Gemma-2-9B-IT", lines=3, interactive=False)
+        comment_gemma_9b = gr.Textbox(label="Комментарий к сообщению Gemma-2-9B-IT", lines=3)
+        corrected_gemma_9b = gr.Textbox(label="Откорректированное сообщение Gemma-2-9B-IT", lines=3)
+        save_gemma_9b_btn = gr.Button("👍 Gemma-2-9B-IT")
 
-        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: 
-                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),
-            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]   # ОПФ
-            ],
-            outputs=None
-        )
+    # Десятый ряд
+    with gr.Row():
+        output_text_mistral = gr.Textbox(label="Неперсонализированное сообщение Mistral-Large-2407", lines=3, interactive=False)
+        personalized_output_text_mistral = gr.Textbox(label="Персонализированное сообщение Mistral-Large-2407", lines=3, interactive=False)
+        comment_mistral = gr.Textbox(label="Комментарий к сообщению Mistral-Large-2407", lines=3)
+        corrected_mistral = gr.Textbox(label="Откорректированное сообщение Mistral-Large-2407", lines=3)
+        save_mistral_btn = gr.Button("👍 Mistral-Large-2407")
+
+   
+    # Добавление функционала для кнопок
+    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, 
+            output_text_meta_llama_405b,
+            output_text_meta_llama_70b,
+            output_text_meta_llama_8b,
+            output_text_gemma_27b,
+            output_text_gemma_9b,
+            output_text_mistral  # Добавляем Mistral
+        ]
+    )
+    
+    personalize_btn.click(
+        personalize_messages_with_yield,
+        inputs=[
+            output_text_gpt4o, 
+            output_text_gigachat_pro, 
+            output_text_gigachat_lite, 
+            output_text_gigachat_plus, 
+            output_text_meta_llama_405b,
+            output_text_meta_llama_70b,
+            output_text_meta_llama_8b,
+            output_text_gemma_27b,
+            output_text_gemma_9b,
+            output_text_mistral  # Добавляем Mistral
+        ] + selections,
+        outputs=[
+            personalization_prompt,
+            personalized_output_text_gpt4o,
+            personalized_output_text_gigachat_pro,
+            personalized_output_text_gigachat_lite,
+            personalized_output_text_gigachat_plus,
+            personalized_output_text_meta_llama_405b,
+            personalized_output_text_meta_llama_70b,
+            personalized_output_text_meta_llama_8b,
+            personalized_output_text_gemma_27b,
+            personalized_output_text_gemma_9b,
+            personalized_output_text_mistral  # Добавляем Mistral
+        ]
+    )
 
-        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: 
-                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),
-            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]   # ОПФ
-            ],
-            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: 
-                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),
-            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]   # ОПФ
-            ],
-            outputs=None
-        )
+    # Привязка кнопок к функциям сохранения
+    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),
+        inputs=[
+            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]   # ОПФ
+        ],
+        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)
-            ]
-        )
+    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: 
+            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),
+        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]   # ОПФ
+        ],
+        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: 
+            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),
+        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]   # ОПФ
+        ],
+        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: 
+            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),
+        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]   # ОПФ
+        ],
+        outputs=None
+    )
+
+    save_meta_llama_405b_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, "Meta-Llama-3.1-405B", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+        inputs=[
+            personalized_output_text_meta_llama_405b,
+            comment_meta_llama_405b,
+            corrected_meta_llama_405b,
+            description_input,
+            advantages_input,
+            prompt_display,
+            output_text_meta_llama_405b,
+            selections[0],  # Пол
+            selections[1],  # Поколение
+            selections[2],  # Психотип
+            selections[3],  # Стадия бизнеса
+            selections[4],  # Отрасль
+            selections[5]   # ОПФ
+        ],
+        outputs=None
+    )
+    
+    save_meta_llama_70b_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, "Meta-Llama-3.1-70B", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+        inputs=[
+            personalized_output_text_meta_llama_70b,
+            comment_meta_llama_70b,
+            corrected_meta_llama_70b,
+            description_input,
+            advantages_input,
+            prompt_display,
+            output_text_meta_llama_70b,
+            selections[0],  # Пол
+            selections[1],  # Поколение
+            selections[2],  # Психотип
+            selections[3],  # Стадия бизнеса
+            selections[4],  # Отрасль
+            selections[5]   # ОПФ
+        ],
+        outputs=None
+    )
+    
+    save_meta_llama_8b_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, "Meta-Llama-3.1-8B", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+        inputs=[
+            personalized_output_text_meta_llama_8b,
+            comment_meta_llama_8b,
+            corrected_meta_llama_8b,
+            description_input,
+            advantages_input,
+            prompt_display,
+            output_text_meta_llama_8b,
+            selections[0],  # Пол
+            selections[1],  # Поколение
+            selections[2],  # Психотип
+            selections[3],  # Стадия бизнеса
+            selections[4],  # Отрасль
+            selections[5]   # ОПФ
+        ],
+        outputs=None
+    )
+    
+    save_gemma_27b_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, "Gemma-2-27B-IT", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+        inputs=[
+            personalized_output_text_gemma_27b,
+            comment_gemma_27b,
+            corrected_gemma_27b,
+            description_input,
+            advantages_input,
+            prompt_display,
+            output_text_gemma_27b,
+            selections[0],  # Пол
+            selections[1],  # Поколение
+            selections[2],  # Психотип
+            selections[3],  # Стадия бизнеса
+            selections[4],  # Отрасль
+            selections[5]   # ОПФ
+        ],
+        outputs=None
+    )
+    
+    save_gemma_9b_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, "Gemma-2-9B-IT", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+        inputs=[
+            personalized_output_text_gemma_9b,
+            comment_gemma_9b,
+            corrected_gemma_9b,
+            description_input,
+            advantages_input,
+            prompt_display,
+            output_text_gemma_9b,
+            selections[0],  # Пол
+            selections[1],  # Поколение
+            selections[2],  # Психотип
+            selections[3],  # Стадия бизнеса
+            selections[4],  # Отрасль
+            selections[5]   # ОПФ
+        ],
+        outputs=None
+    )
+
+    save_mistral_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, "Mistral", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
+        inputs=[
+            personalized_output_text_mistral,
+            comment_mistral,
+            corrected_mistral,
+            description_input,
+            advantages_input,
+            prompt_display,
+            output_text_mistral,
+            selections[0],  # Пол
+            selections[1],  # Поколение
+            selections[2],  # Психотип
+            selections[3],  # Стадия бизнеса
+            selections[4],  # Отрасль
+            selections[5]   # ОПФ
+        ],
+        outputs=None
+    )
 
 
     demo.launch()