Update app.py

app.py

@@ -76,7 +76,7 @@ def generate_standard_prompt(description, advantages, *selected_values):
     return prompt
 
 # Функции для генерации сообщений
-def generate_message_gpt4o(prompt, temperature=1):
+def generate_message_gpt4o(prompt):
     try:
         headers = {
             "Content-Type": "application/json",
@@ -85,16 +85,14 @@ def generate_message_gpt4o(prompt, temperature=1):
         data = {
             "model": "chatgpt-4o-latest",
             "messages": [{"role": "system", "content": prompt}],
-            "max_tokens": 101,
-            "temperature": temperature  # Передача температуры
+            "max_tokens": 101
         }
         response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
         response_data = response.json()
-        return clean_message(response_data["choices"][0]["message"]["content"].strip())
+        return response_data["choices"][0]["message"]["content"].strip()
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
-
 def clean_message(message):
     # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
     if not message.endswith(('.', '!', '?')):
@@ -104,7 +102,7 @@ def clean_message(message):
     return message
 
 # Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
-def generate_message_gigachat_pro(prompt, temperature):
+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)
@@ -114,7 +112,7 @@ def generate_message_gigachat_pro(prompt, temperature):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-def generate_message_gigachat_lite(prompt, temperature):
+def generate_message_gigachat_lite(prompt, temperature=0.87):
     try:
         time.sleep(2)
         messages = [SystemMessage(content=prompt)]
@@ -125,7 +123,7 @@ def generate_message_gigachat_lite(prompt, temperature):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite: {e}"
 
-def generate_message_gigachat_plus(prompt, temperature):
+def generate_message_gigachat_plus(prompt, temperature=0.87):
     try:
         time.sleep(2)
         messages = [SystemMessage(content=prompt)]
@@ -136,37 +134,37 @@ def generate_message_gigachat_plus(prompt, temperature):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
-def generate_message_gpt4o_with_retry(prompt, temperature):
+def generate_message_gpt4o_with_retry(prompt):
     for _ in range(10):  # Максимум 10 попыток
-        message = generate_message_gpt4o(prompt, temperature)  # Передаем температуру
+        message = generate_message_gpt4o(prompt)
         if len(message) <= 250:
             return message
     return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
-def generate_message_gigachat_pro_with_retry(prompt, temperature):
+def generate_message_gigachat_pro_with_retry(prompt):
     for _ in range(10):
-        message = generate_message_gigachat_pro(prompt, temperature)  # Передаем температуру
+        message = generate_message_gigachat_pro(prompt)
         if len(message) <= 250:
             return message
     return message
 
-def generate_message_gigachat_lite_with_retry(prompt, temperature):
+def generate_message_gigachat_lite_with_retry(prompt):
     for _ in range(10):
-        message = generate_message_gigachat_lite(prompt, temperature)  # Передаем температуру
+        message = generate_message_gigachat_lite(prompt)
         if len(message) <= 250:
             return message
     return message
 
-def generate_message_gigachat_plus_with_retry(prompt, temperature):
+def generate_message_gigachat_plus_with_retry(prompt):
     for _ in range(10):
-        message = generate_message_gigachat_plus(prompt, temperature)  # Передаем температуру
+        message = generate_message_gigachat_plus(prompt)
         if len(message) <= 250:
             return message
     return message
 
 
 # Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values, gpt4o_temperature, gigachat_pro_temperature, gigachat_lite_temperature, gigachat_plus_temperature):
+def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -179,33 +177,34 @@ def generate_messages(description, advantages, *selected_values, gpt4o_temperatu
 
     yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
-    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt, gpt4o_temperature)
+    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt)
     gpt4o_length = len(results["gpt4o"])
-    gpt4o_display = f"{results['gpt4o']}\n\n------\nКоличество знаков: {gpt4o_length}\nТемпература: {gpt4o_temperature}"
+    gpt4o_display = f"{results['gpt4o']}\n\n------\nКоличество знаков: {gpt4o_length}"
     yield results["prompt"], gpt4o_display, "", "", "", "Сообщение GPT-4o сгенерировано"
 
-    results["gigachat_pro"] = generate_message_gigachat_pro_with_retry(standard_prompt, gigachat_pro_temperature)
+    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}\nТемпература: {gigachat_pro_temperature}"
+    gigachat_pro_display = f"{results['gigachat_pro']}\n\n------\nКоличество знаков: {gigachat_pro_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, "", "", "Сообщение GigaChat-Pro сгенерировано"
 
     time.sleep(2)
 
-    results["gigachat_lite"] = generate_message_gigachat_lite_with_retry(standard_prompt, gigachat_lite_temperature)
+    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}\nТемпература: {gigachat_lite_temperature}"
+    gigachat_lite_display = f"{results['gigachat_lite']}\n\n------\nКоличество знаков: {gigachat_lite_length}"
     yield results["prompt"], gpt4o_display, gigachat_pro_display, gigachat_lite_display, "", "Сообщение GigaChat-Lite сгенерировано"
 
     time.sleep(2)
 
-    results["gigachat_plus"] = generate_message_gigachat_plus_with_retry(standard_prompt, gigachat_plus_temperature)
+    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}\nТемпература: {gigachat_plus_temperature}"
+    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"
@@ -389,17 +388,16 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
     # Отправка 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-сообщений по заданным признакам")
-
-    # Добавление элементов управления температурой для каждой модели
-    gpt4o_temperature = gr.Slider(label="GPT-4o: temperature", minimum=0, maximum=1.7, step=0.01, value=1)
-    gigachat_pro_temperature = gr.Slider(label="GigaChat-Pro: temperature", minimum=0, maximum=1.7, step=0.01, value=0.87)
-    gigachat_lite_temperature = gr.Slider(label="GigaChat-Lite: temperature", minimum=0, maximum=1.7, step=0.01, value=0.87)
-    gigachat_plus_temperature = gr.Slider(label="GigaChat-Plus: temperature", minimum=0, maximum=1.7, step=0.01, value=0.87)
-
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -440,19 +438,20 @@ with gr.Blocks() as demo:
             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, 
-                    gpt4o_temperature, 
-                    gigachat_pro_temperature, 
-                    gigachat_lite_temperature, 
-                    gigachat_plus_temperature
-                ], 
-                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
-            )
+        submit_btn.click(
+            generate_messages, 
+            inputs=[
+                description_input, 
+                advantages_input, 
+                *selections, 
+                gpt4o_temperature, 
+                gigachat_pro_temperature, 
+                gigachat_lite_temperature, 
+                gigachat_plus_temperature
+            ], 
+            outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
+        )
+
 
     with gr.Row():
         personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
@@ -601,4 +600,179 @@ with gr.Blocks() as demo:
         )
 
 
+    demo.launch()
+
+Есть скрипт, в котором есть полезная для меня функция:
+
+import gradio as gr
+import requests
+import os
+import time
+from langchain.schema import SystemMessage
+from langchain_community.chat_models.gigachat import GigaChat
+
+# Установка ключа API для OpenAI и GigaChat
+openai_api_key = os.getenv('GPT_KEY')
+gc_key = os.getenv('GC_KEY')
+
+# Авторизация в сервисе 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)
+
+# Функция для генерации персонализированного промпта
+def generate_personalization_prompt(text, opf):
+    if opf == "ИП":
+        prompt = (
+            "Текст должен быть адаптирован для индивидуальных предпринимателей, которые часто ведут бизнес самостоятельно или с минимальным количеством сотрудников. "
+            "Подчеркните простоту и эффективность использования продукта в условиях ограниченных ресурсов. Особое внимание уделите возможности быстрого и удобного управления финансами, "
+            "снижению административной нагрузки и оптимизации затрат. Важно, чтобы текст отражал преимущество в виде простоты интеграции продукта в повседневную деятельность ИП, "
+            "включая оперативность расчетов и гибкость условий. Если имеются льготные условия или специальные предложения для ИП, такие как сниженные комиссии или удобные условия оплаты, "
+            "обязательно упомяните их.\n"
+            f"Текст: \"{text}\""
+        )
+    elif opf == "ООО":
+        prompt = (
+            "Текст должен быть адаптирован для обществ с ограниченной ответственностью (ООО), которые управляют более структурированными и масштабными бизнес-процессами. "
+            "Уделите внимание возможностям продукта в контексте работы с большим количеством сотрудников, управлением бюджетами и финансовыми потоками компании. "
+            "Подчеркните преимущества использования продукта для оптимизации бизнес-процессов, включая автоматизацию расчетов, настройку прав доступа и интеграцию с корпоративными системами учета. "
+            "Важно указать на возможности улучшения операционной эффективности и соответствия юридическим и налоговым требованиям. Если продукт предлагает специальные условия ��ля корпоративных клиентов, "
+            "такие как комплексное обслуживание или индивидуальные тарифы, обязательно включите эту информацию в текст.\n"
+            f"Текст: \"{text}\""
+        )
+    return prompt
+
+# Функция для очистки сообщений
+def clean_message(message):
+    if not message.endswith(('.', '!', '?')):
+        last_period = max(message.rfind('.'), message.rfind('!'), message.rfind('?'))
+        if last_period != -1:
+            message = message[:last_period + 1]
+    return message
+
+# Функция для генерации сообщений GPT-4o
+def generate_message_gpt4o(prompt, temperature=1):
+    try:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {openai_api_key}"
+        }
+        data = {
+            "model": "chatgpt-4o-latest",
+            "messages": [{"role": "system", "content": prompt}],
+            "max_tokens": 101,
+            "temperature": temperature  # Передача температуры
+        }
+        response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
+        response_data = response.json()
+        return clean_message(response_data["choices"][0]["message"]["content"].strip())
+    except Exception as e:
+        return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
+
+
+# Функции для генерации сообщений GigaChat с учетом обрезки
+def generate_message_gigachat_pro(prompt, max_tokens, temperature):
+    try:
+        messages = [SystemMessage(content=prompt)]
+        chat_pro = authenticate_gigachat('GigaChat-Pro', max_tokens, temperature)
+        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, max_tokens, temperature):
+    try:
+        time.sleep(2)
+        messages = [SystemMessage(content=prompt)]
+        chat_lite = authenticate_gigachat('GigaChat', max_tokens, temperature)
+        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, max_tokens, temperature):
+    try:
+        time.sleep(2)
+        messages = [SystemMessage(content=prompt)]
+        chat_plus = authenticate_gigachat('GigaChat-Plus', max_tokens, temperature)
+        res = chat_plus(messages)
+        cleaned_message = clean_message(res.content.strip())
+        return cleaned_message
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Plus: {e}"
+
+# Обновление промпта при изменении ОПФ
+def update_prompt(text, opf):
+    return generate_personalization_prompt(text, opf)
+
+# Функция для постепенной генерации результатов
+def personalize_message(text, opf, prompt, gpt4o_max_tokens, gpt4o_temperature, gigachat_max_tokens, gigachat_temperature):
+    yield prompt, "", "", "", ""
+
+    result_gpt4o = generate_message_gpt4o(prompt, gpt4o_max_tokens, gpt4o_temperature)
+    yield prompt, result_gpt4o, "", "", ""
+
+    result_pro = generate_message_gigachat_pro(prompt, gigachat_max_tokens, gigachat_temperature)
+    yield prompt, result_gpt4o, result_pro, "", ""
+
+    time.sleep(2)
+    
+    result_lite = generate_message_gigachat_lite(prompt, gigachat_max_tokens, gigachat_temperature)
+    yield prompt, result_gpt4o, result_pro, result_lite, ""
+
+    time.sleep(2)
+    
+    result_plus = generate_message_gigachat_plus(prompt, gigachat_max_tokens, gigachat_temperature)
+    yield prompt, result_gpt4o, result_pro, result_lite, result_plus
+
+# Создание интерфейса Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# Персонализация под ОПФ")
+
+    # Добавление элементов управления температурой для каждой модели
+    gpt4o_temperature = gr.Slider(label="GPT-4o: temperature", minimum=0, maximum=2, step=0.01, value=1)
+    gigachat_pro_temperature = gr.Slider(label="GigaChat-Pro: temperature", minimum=0, maximum=1.7, step=0.01, value=0.87)
+    gigachat_lite_temperature = gr.Slider(label="GigaChat-Lite: temperature", minimum=0, maximum=1.7, step=0.01, value=0.87)
+    gigachat_plus_temperature = gr.Slider(label="GigaChat-Plus: temperature", minimum=0, maximum=1.7, step=0.01, value=0.87)
+
+    with gr.Row():
+        with gr.Column():
+            # Параметры для GPT-4o
+            gpt4o_max_tokens = gr.Slider(label="GPT-4o: max_tokens", minimum=0, maximum=10000, step=1, value=103)
+            gpt4o_temperature = gr.Slider(label="GPT-4o: temperature", minimum=0, maximum=2, step=0.01, value=1)
+
+            # Параметры для всех моделей GigaChat
+            gigachat_max_tokens = gr.Slider(label="GigaChat: max_tokens", minimum=0, maximum=10000, step=1, value=68)
+            gigachat_temperature = gr.Slider(label="GigaChat: temperature", minimum=0, maximum=1.7, step=0.01, value=0.87)
+    
+    # Предзаполненное значение текста
+    text_input = gr.Textbox(
+        label="Введите ваш текст",
+        lines=5,
+        value="Перечислять зарплату вашим сотрудникам удобно с зарплатным проектом от Сбера! Любые виды выплат в едином договоре, зачисление происходит от 1 минуты, в любое время 24/7. Оставить заявку ->"
+    )
+    
+    opf_input = gr.Dropdown(choices=["ООО", "ИП"], label="Выберите ОПФ")
+    
+    # Окно для отображения и редактирования сгенерированного промпта
+    prompt_display = gr.Textbox(label="Сгенерированный промпт", lines=5, interactive=True)
+    
+    # Обновляем промпт при изменении ОПФ или текста
+    opf_input.change(fn=update_prompt, inputs=[text_input, opf_input], outputs=prompt_display)
+    text_input.change(fn=update_prompt, inputs=[text_input, opf_input], outputs=prompt_display)
+    
+    personalize_button = gr.Button("Персонализировать")
+    
+    output_gpt4o = gr.Textbox(label="Результат GPT-4o", lines=3, interactive=False)
+    output_gigachat_pro = gr.Textbox(label="Результат GigaChat-Pro", lines=3, interactive=False)
+    output_gigachat_lite = gr.Textbox(label="Результат GigaChat-Lite", lines=3, interactive=False)
+    output_gigachat_plus = gr.Textbox(label="Результат GigaChat-Lite+", lines=3, interactive=False)
+    
+    personalize_button.click(
+        personalize_message, 
+        inputs=[text_input, opf_input, prompt_display, gpt4o_max_tokens, gpt4o_temperature, gigachat_max_tokens, gigachat_temperature], 
+        outputs=[prompt_display, output_gpt4o, output_gigachat_pro, output_gigachat_lite, output_gigachat_plus]
+    )
+
     demo.launch()