Update app.py

app.py

@@ -14,10 +14,22 @@ openai_api_key = os.getenv('GPT_KEY')
 gc_key = os.getenv('GC_KEY')
 token = os.getenv('GITHUB_TOKEN')
 
-# Авторизация в сервисе GigaChat
-chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, verify_ssl_certs=False)
-chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, verify_ssl_certs=False)
-chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, verify_ssl_certs=False)
+# Функция аутентификации в сервисе GigaChat с заданной температурой
+def authenticate_gigachat(model, max_tokens, temperature):
+    return GigaChat(credentials=gc_key, model=model, max_tokens=max_tokens, temperature=temperature, verify_ssl_certs=False)
+
+# Функция аутентификации для GPT-4o
+def authenticate_gpt4o(max_tokens, temperature):
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {openai_api_key}"
+    }
+    return {
+        "model": "chatgpt-4o-latest",
+        "max_tokens": max_tokens,
+        "temperature": temperature,
+        "headers": headers
+    }
 
 # Загрузка данных из Excel-файла
 try:
@@ -75,98 +87,56 @@ def generate_standard_prompt(description, advantages, *selected_values):
     )
     return prompt
 
-# Функции для генерации сообщений
-def generate_message_gpt4o(prompt, temperature=1):
+# Функция для генерации сообщений GPT-4o
+def generate_message_gpt4o(auth_params, prompt):
     try:
-        headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {openai_api_key}"
-        }
         data = {
-            "model": "chatgpt-4o-latest",
+            "model": auth_params["model"],
             "messages": [{"role": "system", "content": prompt}],
-            "max_tokens": 101,
-            "temperature": temperature  # Передача температуры
+            "max_tokens": auth_params["max_tokens"],
+            "temperature": auth_params["temperature"]
         }
-        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
+        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=auth_params["headers"])
         response_data = response.json()
-        return clean_message(response_data["choices"][0]["message"]["content"].strip()) + f" {temperature}"
+        return clean_message(response_data["choices"][0]["message"]["content"].strip())
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
+# Функции для генерации сообщений GigaChat
+def generate_message_gigachat(chat_instance, prompt):
+    try:
+        messages = [SystemMessage(content=prompt)]
+        res = chat_instance(messages)
+        cleaned_message = clean_message(res.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat: {e}"
 
 def clean_message(message):
     # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
     if not message.endswith(('.', '!', '?')):
         last_period = message.rfind('.')
-        if last_period != -1:
+        if (last_period != -1):
             message = message[:last_period + 1]
     return message
 
-# Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
-def generate_message_gigachat_pro(prompt, temperature=0.87):
-    try:
-        messages = [SystemMessage(content=prompt)]
-        chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, temperature=temperature, verify_ssl_certs=False)
-        res = chat_pro(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Pro: {e}"
-
-def generate_message_gigachat_lite(prompt, temperature=0.87):
-    try:
-        time.sleep(2)
-        messages = [SystemMessage(content=prompt)]
-        chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=temperature, verify_ssl_certs=False)
-        res = chat_lite(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Lite: {e}"
-
-def generate_message_gigachat_plus(prompt, temperature=0.87):
-    try:
-        time.sleep(2)
-        messages = [SystemMessage(content=prompt)]
-        chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, temperature=temperature, verify_ssl_certs=False)
-        res = chat_plus(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
-    except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Plus: {e}"
-
-def generate_message_gpt4o_with_retry(prompt):
+# Функции для генерации сообщений с повторными попытками
+def generate_message_gpt4o_with_retry(auth_params, prompt):
     for _ in range(10):  # Максимум 10 попыток
-        message = generate_message_gpt4o(prompt)
+        message = generate_message_gpt4o(auth_params, prompt)
         if len(message) <= 250:
             return message
     return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
-def generate_message_gigachat_pro_with_retry(prompt):
-    for _ in range(10):
-        message = generate_message_gigachat_pro(prompt)
-        if len(message) <= 250:
-            return message
-    return message
-
-def generate_message_gigachat_lite_with_retry(prompt):
-    for _ in range(10):
-        message = generate_message_gigachat_lite(prompt)
-        if len(message) <= 250:
-            return message
-    return message
-
-def generate_message_gigachat_plus_with_retry(prompt):
+def generate_message_gigachat_with_retry(chat_instance, prompt):
     for _ in range(10):
-        message = generate_message_gigachat_plus(prompt)
+        message = generate_message_gigachat(chat_instance, prompt)
         if len(message) <= 250:
             return message
     return message
 
-
 # Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values):
+def generate_messages(description, advantages, *selected_values, gpt4o_auth_params, chat_pro, chat_lite, chat_plus):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -179,224 +149,32 @@ def generate_messages(description, advantages, *selected_values):
 
     yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
-    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt)
+    results["gpt4o"] = generate_message_gpt4o_with_retry(gpt4o_auth_params, standard_prompt)
     gpt4o_length = len(results["gpt4o"])
     gpt4o_display = f"{results['gpt4o']}\n\n------\nКоличество знаков: {gpt4o_length}"
     yield results["prompt"], gpt4o_display, "", "", "", "Сообщение GPT-4o сгенерировано"
 
-    results["gigachat_pro"] = generate_message_gigachat_pro_with_retry(standard_prompt)
+    results["gigachat_pro"] = generate_message_gigachat_with_retry(chat_pro, standard_prompt)
     gigachat_pro_length = len(results["gigachat_pro"])
     gigachat_pro_display = f"{results['gigachat_pro']}\n\n------\nКоличество знаков: {gigachat_pro_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "Сообщение GigaChat-Pro сгенерировано"
 
     time.sleep(2)
 
-    results["gigachat_lite"] = generate_message_gigachat_lite_with_retry(standard_prompt)
+    results["gigachat_lite"] = generate_message_gigachat_with_retry(chat_lite, standard_prompt)
     gigachat_lite_length = len(results["gigachat_lite"])
     gigachat_lite_display = f"{results['gigachat_lite']}\n\n------\nКоличество знаков: {gigachat_lite_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Сообщение GigaChat-Lite сгенерировано"
 
     time.sleep(2)
 
-    results["gigachat_plus"] = generate_message_gigachat_plus_with_retry(standard_prompt)
+    results["gigachat_plus"] = generate_message_gigachat_with_retry(chat_plus, standard_prompt)
     gigachat_plus_length = len(results["gigachat_plus"])
     gigachat_plus_display = f"{results['gigachat_plus']}\n\n------\nКоличество знаков: {gigachat_plus_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все сообщения сгенерированы"
 
     return results
 
-
-
-# Функция для генерации персонализированного промпта
-def generate_personalization_prompt(*selected_values):
-    prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
-    for i, feature in enumerate(features.keys()):
-        if selected_values[i]:
-            try:
-                prompt += f"{features[feature][selected_values[i]]}\n"
-            except KeyError:
-                return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
-                
-    prompt += "Убедись, что в готовом тексте до 250 знаков с пробелами."
-    
-    return prompt.strip()
-
-# Функция для выполнения персонализации на основе сгенерированного промпта и сообщения
-def perform_personalization(standard_message, personalization_prompt):
-    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
-    return generate_message_gpt4o(full_prompt)
-
-# Также обновляем функции персонализации
-def perform_personalization_gigachat(standard_message, personalization_prompt, model):
-    full_prompt = f"{personalization_prompt}\n\nТекст для адаптации:\n{standard_message}"
-    if model == "gigachat_pro":
-        result = generate_message_gigachat_pro(full_prompt)
-    elif model == "gigachat_lite":
-        result = generate_message_gigachat_lite(full_prompt)
-    elif model == "gigachat_plus":
-        result = generate_message_gigachat_plus(full_prompt)
-    return clean_message(result)
-
-def perform_personalization_with_retry(standard_message, personalization_prompt):
-    for _ in range(10):  # Максимум 10 попыток
-        message = perform_personalization(standard_message, personalization_prompt)
-        if len(message) <= 250:
-            return message
-    return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
-
-def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model):
-    for _ in range(10):
-        message = perform_personalization_gigachat(standard_message, personalization_prompt, model)
-        if len(message) <= 250:
-            return message
-    return message
-
-
-# Обновляем блок персонализации
-def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, *selected_values):
-    personalization_prompt = generate_personalization_prompt(*selected_values)
-    yield personalization_prompt, "", "", "", "", "Промпт для персонализации сгенерирован"
-
-    personalized_message_gpt4o = perform_personalization_with_retry(gpt4o_message, personalization_prompt)
-    gpt4o_length = len(personalized_message_gpt4o)
-    gpt4o_display = f"{personalized_message_gpt4o}\n\n------\nКоличество знаков: {gpt4o_length}"
-    yield personalization_prompt, gpt4o_display, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
-
-    personalized_message_gigachat_pro = perform_personalization_gigachat_with_retry(gigachat_pro_message, personalization_prompt, "gigachat_pro")
-    gigachat_pro_length = len(personalized_message_gigachat_pro)
-    gigachat_pro_display = f"{personalized_message_gigachat_pro}\n\n------\nКоличество знаков: {gigachat_pro_length}"
-    yield personalization_prompt, gpt4o_display, gigachat_pro_display, "", "", "Персонализированное сообщение GigaChat-Pro сгенерировано"
-
-    personalized_message_gigachat_lite = perform_personalization_gigachat_with_retry(gigachat_lite_message, personalization_prompt, "gigachat_lite")
-    gigachat_lite_length = len(personalized_message_gigachat_lite)
-    gigachat_lite_display = f"{personalized_message_gigachat_lite}\n\n------\nКоличество знаков: {gigachat_lite_length}"
-    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Персонализированное сообщение GigaChat-Lite сгенерировано"
-
-    personalized_message_gigachat_plus = perform_personalization_gigachat_with_retry(gigachat_plus_message, personalization_prompt, "gigachat_plus")
-    gigachat_plus_length = len(personalized_message_gigachat_plus)
-    gigachat_plus_display = f"{personalized_message_gigachat_plus}\n\n------\nКоличество знаков: {gigachat_plus_length}"
-    yield personalization_prompt, gpt4o_display, gigachat_pro_display, gigachat_lite_display, gigachat_plus_display, "Все персонализированные сообщения сгенерированы"
-
-
-# Функция для генерации промпта проверки текста
-def generate_error_check_prompt():
-    prompt = (
-        "Проверь текст SMS-сообщения на соответствие установленным правилам и ограничениям, касающимся его формирования. На основе выявленных несоответствий предоставь рекомендации по исправлению текста. "
-        "Особое внимание удели проверке: количества символов в тексте SMS-сообщения, орфографическим и пунктуационным ошибкам, определению частей речи (причастия, деепричастия, причастный оборот, деепричастный оборот). "
-        "Анализируй только текст SMS-сообщения, ничего не придумывай и не добавляй лишнего. "
-        "Правила и ограничения, которым должен соответствовать текст SMS-сообщения:\n"
-        "1. Количество символов в SMS-сообщении должно быть до 250 знаков с учетом пробелов.\n"
-        "2. В тексте должен быть призыв к действию с использованием глагола в повелительном наклонении (например: оформите, получите, разместите, размещайте, откройте, подключите, подайте заявку).\n"
-        "3. Должно соблюдаться соответствие фактов о продукте.\n"
-        "4. В генерациях смс запрещено использовать обещания и гарантии.\n"
-        "5. В генерациях смс запрещено использовать составные конструкции из двух глаголов.\n"
-        "6. В генерациях смс запрещено использовать причастия и причастные обороты.\n"
-        "7. В генерациях смс запрещено использовать деепричастия и деепричастные обороты.\n"
-        "8. В генерациях смс запрещено использовать превосходную степень прилагательных.\n"
-        "9. В генерациях смс запрещено использовать страдательный залог.\n"
-        "10. В генерациях смс запрещено использовать порядковые числительные от 10 прописью.\n"
-        "11. В генерациях смс запрещено использовать цепочки с придаточными предложениями.\n"
-        "12. В генерациях смс запрещено использовать разделительные повторяющиеся союзы.\n"
-        "13. В генерациях смс запрещено использовать вводные конструкции.\n"
-        "14. В генерациях смс запрещено использовать усилители.\n"
-        "15. В генерациях смс запрещено использовать паразиты времени.\n"
-        "16. В генерациях смс запрещено использовать несколько существительных подряд, в том числе отглагольных.\n"
-        "17. В генерациях смс запрещено использовать производные предлоги.\n"
-        "18. В генерациях смс запрещено использовать сложные предложения, в которых нет связи между частями.\n"
-        "19. В генерациях смс запрещено использовать сложноподчинённые предложения.\n"
-        "20. В генерациях смс запрещено использовать даты прописью.\n"
-        "21. В генерациях смс запрещено использовать близкие по смыслу однородные члены.\n"
-        "22. В генерациях смс запрещено использовать шокирующие, экстравагантные, кликбейтные фразы.\n"
-        "23. В генерациях смс запрещено использовать абстрактные заявления без поддержки фактами и отсутствие доказательства пользы для клиента.\n"
-        "24. В генерациях смс запрещено использовать гарантирующие фразы.\n"
-        "25. В генерациях смс запрещено использовать узкоспециализированные термины.\n"
-        "26. В генерациях смс запрещено использовать фразы, способные создать двойственное ощущение, обидеть.\n"
-        "27. В генерациях смс запрещено использовать речевые клише, рекламные штампы, канцеляризмы.\n"
-        "28. В генерациях смс запрещено использовать запрещенные слова: № один, номер один, № 1, вкусный, дешёвый, продукт, спам, банкротство, долги, займ, срочно, лучший, главный, номер 1, успех, лидер.\n"
-        "29. Сообщение должно быть написано без орфографических и грамматических ошибок.\n"
-        "30. Запрещены повторы слов.\n"
-        "31. В тексте должны использоваться правильные знаки препинания.\n"
-        "32. Если в тексте используются кавычки, они должны быть в форме «кавычки-ёлочки».\n"
-        "33. В тексте SMS сообщения должны обязательно присутствовать: название продукта, условия использования продукта / Преимущества продукта / Шаги для подключения или начала использования / Условия акции (если предложение по продукту акционное).\n"
-        "Форма ответа: [Ответ должен быть кратким, должен содержать только рекомендации по устранению найденных несоответствий, соответствия каждому пункту правил описывать категорически запрещено]."
-    )
-    return prompt
-
-
-# Функция для выполнения проверки текста с использованием yield
-def check_errors_with_yield(*personalized_messages):
-    if len(personalized_messages) < 4:
-        yield "", "", "", "", "", "Ошибка: недостаточно сообщений для проверки"
-        return
-
-    error_check_prompt = generate_error_check_prompt()
-    yield error_check_prompt, "", "", "", "", "Промпт для проверки текста сгенерирован"
-
-    error_message_gpt4o = perform_personalization(f"{error_check_prompt}\n\n{personalized_messages[0]}", "")
-    yield error_check_prompt, error_message_gpt4o, "", "", "", "Результат проверки GPT-4o сгенерирован"
-
-    error_message_gigachat_pro = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[1]}", "", "gigachat_pro")
-    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, "", "", "Результат проверки GigaChat-Pro сгенерирован"
-    
-    time.sleep(3)
-    error_message_gigachat_lite = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[2]}", "", "gigachat_lite")
-    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, "", "Результат проверки GigaChat-Lite сгенерирован"
-    
-    try:
-        time.sleep(3)
-        error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[3]}", "", "gigachat_plus")
-    except Exception as e:
-        error_message_gigachat_plus = f"Ошибка при обработке GigaChat-Plus: {e}"
-        
-    yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
-
-
-def save_to_github(personalized_message, model_name, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature):
-    # Собираем все данные в один словарь
-    data_to_save = {
-        "Модель": model_name,
-        "Температура": temperature,  # Добавляем температуру
-        "Персонализированное сообщение": personalized_message,
-        "Комментарий": comment,
-        "Откорректированное сообщение": corrected_message,
-        "Описание предложения": description,
-        "Преимущества": advantages,
-        "Неперсонализированный промпт": non_personalized_prompt,
-        "Неперсонализированное сообщение": non_personalized_message,
-        "Пол": gender,
-        "Поколение": generation,
-        "Психотип": psychotype,
-        "Стадия бизнеса": business_stage,
-        "Отрасль": industry,
-        "ОПФ": legal_form
-    }
-
-    # Преобразуем контент в JSON-строку и кодируем в base64
-    file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
-
-    # Параметры для GitHub API
-    repo = "fruitpicker01/Storage_1"
-    path = f"file_{int(time.time())}.json"
-    url = f"https://api.github.com/repos/{repo}/contents/{path}"
-    headers = {
-        "Authorization": f"token {token}",
-        "Content-Type": "application/json"
-    }
-    data = {
-        "message": f"Добавлен новый файл {path}",
-        "content": file_content_encoded
-    }
-
-    # Отправка POST-запроса на GitHub API для создания файла в репозитории
-    response = requests.put(url, headers=headers, data=json.dumps(data))
-
-    if response.status_code == 201:
-        print("Файл успешно загружен на GitHub.")
-    else:
-        print(f"Ошибка при загрузке файла на GitHub: {response.status_code}")
-        print(f"Ответ сервера: {response.json()}")
-
-
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -444,18 +222,18 @@ with gr.Blocks() as demo:
             output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3, interactive=False)
             output_text_gigachat_pro = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Pro", lines=3, interactive=False)
             output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3, interactive=False)
-            output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3, interactive=False)
+            output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Plus", lines=3, interactive=False)
             
         submit_btn.click(
             generate_messages, 
             inputs=[
                 description_input, 
                 advantages_input, 
-                *selections, 
-                gpt4o_temperature, 
-                gigachat_pro_temperature, 
-                gigachat_lite_temperature, 
-                gigachat_plus_temperature
+                *selections,
+                authenticate_gpt4o(101, gpt4o_temperature),
+                authenticate_gigachat('GigaChat-Pro', 68, gigachat_pro_temperature),
+                authenticate_gigachat('GigaChat', 68, gigachat_lite_temperature),
+                authenticate_gigachat('GigaChat-Plus', 68, gigachat_plus_temperature)
             ], 
             outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
         )