Update app.py

app.py

@@ -16,14 +16,10 @@ 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, temperature=1, verify_ssl_certs=False)
 chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=1, verify_ssl_certs=False)
@@ -295,50 +291,6 @@ def generate_message_meta_llama_3_1_8b(prompt):
     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.8
-        )
-        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.8
-        )
-        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,
-                    "temperature": 0.8
-                },
-            ]
-        )
-        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)
@@ -374,41 +326,6 @@ def generate_message_meta_llama_3_1_405b_with_retry(prompt):
             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, key_message, approach, *selected_values):
     
@@ -422,85 +339,44 @@ def generate_messages(description, advantages, key_message, approach, *selected_
         "gigachat_pro": None,
         "gigachat_lite": 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
+        "meta_llama_3_1_405b": None
     }
 
-    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, "", "", "", "", "", "", "", "", ""
+    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, "", "", "", "", "", "", "", ""
+    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, "", "", "", "", "", "", ""
+    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, ""
 
     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, "", "", ""
+    yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, meta_llama_405b_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)
+    time.sleep(2)
 
     return results
 
@@ -590,38 +466,13 @@ def perform_personalization_meta_llama_405b(standard_message, personalization_pr
     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_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 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,
+    meta_llama_405b_message,
     key_message,
     approach,
     *selected_values
@@ -655,31 +506,6 @@ def personalize_messages_with_yield(
     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
-
 
 # Функция для генерации промпта проверки текста
 def generate_error_check_prompt():
@@ -726,46 +552,6 @@ def generate_error_check_prompt():
     return prompt
 
 
-# Функция для выполнения проверки текста с использованием yield
-def check_errors_with_yield(*personalized_messages):
-    if len(personalized_messages) < 10:  # Adjusted for the inclusion of Mistral
-        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 сгенерирован"
-    
-    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 сгенерирован"
-
-    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 сгенерирован"
-    
-    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, key_message, approach):
     # Собираем все данные в один словарь
     data_to_save = {
@@ -813,11 +599,6 @@ def personalize_and_save(
     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,
     description,
     advantages,
     key_message,  # Обратите внимание, что key_message здесь не передается в selected_values
@@ -831,11 +612,6 @@ def personalize_and_save(
         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,
         key_message,
         approach,
         *selected_values  # Только признаки для персонализации
@@ -881,10 +657,10 @@ def reset_button_text():
 
 def clear_unnecessary_fields():
     return (
-        "", "", "", "", "", "", "", "", "", "",  # personalized outputs and prompts
-        "", "", "", "", "", "", "", "", "", "",  # comment fields
-        "", "", "", "", "", "", "", "", "", "",  # corrected message fields
-        "", "", "", "", "", "", "", "", "", "",  # оставшиеся поля
+        "", "", "", "", "",  # personalized outputs and prompts
+        "", "", "", "", "",  # comment fields
+        "", "", "", "", "",  # corrected message fields
+        "", "", "", "", "",  # оставшиеся поля
         "", ""  # Дополнительное пустое значение
     )
 
@@ -1001,11 +777,6 @@ with gr.Row():
             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,
             comment_gpt4o,  
             corrected_gpt4o,
             comment_gigachat_pro,
@@ -1015,17 +786,7 @@ with gr.Row():
             comment_gigachat_plus,
             corrected_gigachat_plus,
             comment_meta_llama_405b,
-            corrected_meta_llama_405b,
-            comment_meta_llama_70b,
-            corrected_meta_llama_70b,
-            comment_meta_llama_8b,
-            corrected_meta_llama_8b,
-            comment_gemma_27b,
-            corrected_gemma_27b,
-            comment_gemma_9b,
-            corrected_gemma_9b,
-            comment_mistral,
-            corrected_mistral
+            corrected_meta_llama_405b
         ]
     )
     
@@ -1038,12 +799,7 @@ with gr.Row():
             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
+            output_text_meta_llama_405b
         ]
     )
 
@@ -1057,11 +813,6 @@ with gr.Row():
             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,  # Поля для персонализированных сообщений
             comment_gpt4o,  
             corrected_gpt4o,
             comment_gigachat_pro,
@@ -1071,17 +822,7 @@ with gr.Row():
             comment_gigachat_plus,
             corrected_gigachat_plus,
             comment_meta_llama_405b,
-            corrected_meta_llama_405b,
-            comment_meta_llama_70b,
-            corrected_meta_llama_70b,
-            comment_meta_llama_8b,
-            corrected_meta_llama_8b,
-            comment_gemma_27b,
-            corrected_gemma_27b,
-            comment_gemma_9b,
-            corrected_gemma_9b,
-            comment_mistral,
-            corrected_mistral  # Поля для комментариев и откорректированных сообщений
+            corrected_meta_llama_405b
         ]
     )
     
@@ -1092,12 +833,7 @@ with gr.Row():
             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,  # Входные данные для персонализации
+            output_text_meta_llama_405b
             description_input,
             advantages_input,
             key_message_input,  # Add key_message_input
@@ -1109,12 +845,7 @@ with gr.Row():
             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  # Выходные данные для персонализации
+            personalized_output_text_meta_llama_405b
         ]
     )
 
@@ -1140,21 +871,11 @@ with gr.Row():
             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,
             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,
             comment_gpt4o,  
             corrected_gpt4o,
             comment_gigachat_pro,
@@ -1164,17 +885,7 @@ with gr.Row():
             comment_gigachat_plus,
             corrected_gigachat_plus,
             comment_meta_llama_405b,
-            corrected_meta_llama_405b,
-            comment_meta_llama_70b,
-            corrected_meta_llama_70b,
-            comment_meta_llama_8b,
-            corrected_meta_llama_8b,
-            comment_gemma_27b,
-            corrected_gemma_27b,
-            comment_gemma_9b,
-            corrected_gemma_9b,
-            comment_mistral,
-            corrected_mistral
+            corrected_meta_llama_405b
         ]
     )
 
@@ -1349,170 +1060,5 @@ with gr.Row():
         outputs=[save_meta_llama_405b_btn]
     )
     
-    save_meta_llama_70b_btn.click(
-        fn=prepare_button_text,
-        inputs=[],
-        outputs=[save_meta_llama_70b_btn]
-    ).then(
-        fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, key_message, approach:
-            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, key_message, approach),
-        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],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
-        ],
-        outputs=None
-    ).then(
-        fn=update_button_text,
-        outputs=[save_meta_llama_70b_btn]
-    ).then(
-        fn=reset_button_text,
-        outputs=[save_meta_llama_70b_btn]
-    )
-    
-    save_meta_llama_8b_btn.click(
-        fn=prepare_button_text,
-        inputs=[],
-        outputs=[save_meta_llama_8b_btn]
-    ).then(
-        fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, key_message, approach:
-            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, key_message, approach),
-        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],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
-        ],
-        outputs=None
-    ).then(
-        fn=update_button_text,
-        outputs=[save_meta_llama_8b_btn]
-    ).then(
-        fn=reset_button_text,
-        outputs=[save_meta_llama_8b_btn]
-    )
-    
-    save_gemma_27b_btn.click(
-        fn=prepare_button_text,
-        inputs=[],
-        outputs=[save_gemma_27b_btn]
-    ).then(
-        fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, key_message, approach:
-            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, key_message, approach),
-        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],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
-        ],
-        outputs=None
-    ).then(
-        fn=update_button_text,
-        outputs=[save_gemma_27b_btn]
-    ).then(
-        fn=reset_button_text,
-        outputs=[save_gemma_27b_btn]
-    )
-    
-    save_gemma_9b_btn.click(
-        fn=prepare_button_text,
-        inputs=[],
-        outputs=[save_gemma_9b_btn]
-    ).then(
-        fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, key_message, approach:
-            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, key_message, approach),
-        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],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
-        ],
-        outputs=None
-    ).then(
-        fn=update_button_text,
-        outputs=[save_gemma_9b_btn]
-    ).then(
-        fn=reset_button_text,
-        outputs=[save_gemma_9b_btn]
-    )
-
-    save_mistral_btn.click(
-        fn=prepare_button_text,
-        inputs=[],
-        outputs=[save_mistral_btn]
-    ).then(
-        fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, key_message, approach:
-            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, key_message, approach),
-        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],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
-        ],
-        outputs=None
-    ).then(
-        fn=update_button_text,
-        outputs=[save_mistral_btn]
-    ).then(
-        fn=reset_button_text,
-        outputs=[save_mistral_btn]
-    )
-
 
     demo.launch()