Update app.py

app.py

@@ -15,9 +15,9 @@ gc_key = os.getenv('GC_KEY')
 token = os.getenv('GITHUB_TOKEN')
 
 # Авторизация в сервисе GigaChat
-chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
-chat_lite = GigaChat(credentials=gc_key, model='GigaChat', verify_ssl_certs=False)
-chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', verify_ssl_certs=False)
+chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, verify_ssl_certs=False)
+chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, verify_ssl_certs=False)
+chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, verify_ssl_certs=False)
 
 # Загрузка данных из Excel-файла
 try:
@@ -39,7 +39,6 @@ for sheet_name, df in data.items():
 def generate_standard_prompt(description, advantages, *selected_values):
     prompt = (
         "Сгенерируй смс-сообщение для клиента.\n"
-        "Объем готового текста: до 250 знаков с пробелами.\n"
         "Начни сообщение с призыва к действию с продуктом.\n"
         f"Описание предложения: {description}\n"
         f"Преимущества: {advantages}\n"
@@ -86,7 +85,7 @@ def generate_message_gpt4o(prompt):
         data = {
             "model": "chatgpt-4o-latest",
             "messages": [{"role": "system", "content": prompt}],
-            "max_tokens": 700
+            "max_tokens": 101
         }
         response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
         response_data = response.json()
@@ -94,11 +93,21 @@ def generate_message_gpt4o(prompt):
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
+def clean_message(message):
+    # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
+    if not message.endswith(('.', '!', '?')):
+        last_period = message.rfind('.')
+        if last_period != -1:
+            message = message[:last_period + 1]
+    return message
+
+# Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
 def generate_message_gigachat_pro(prompt):
     try:
         messages = [SystemMessage(content=prompt)]
         res = chat_pro(messages)
-        return res.content.strip()
+        cleaned_message = clean_message(res.content.strip())
+        return cleaned_message
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
@@ -107,7 +116,8 @@ def generate_message_gigachat_lite(prompt):
         time.sleep(2)
         messages = [SystemMessage(content=prompt)]
         res = chat_lite(messages)
-        return res.content.strip()
+        cleaned_message = clean_message(res.content.strip())
+        return cleaned_message
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite: {e}"
 
@@ -116,11 +126,12 @@ def generate_message_gigachat_plus(prompt):
         time.sleep(2)
         messages = [SystemMessage(content=prompt)]
         res = chat_plus(messages)
-        return res.content.strip()
+        cleaned_message = clean_message(res.content.strip())
+        return cleaned_message
     except Exception as e:
-        return f"Ошибка при обращении к GigaChat-Lite+: {e}"
+        return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
-# Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
+# Обновляем генерацию сообщений для отображения в интерфейсе
 def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
@@ -135,21 +146,32 @@ def generate_messages(description, advantages, *selected_values):
     yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
     results["gpt4o"] = generate_message_gpt4o(standard_prompt)
-    yield results["prompt"], results["gpt4o"], "", "", "", "Сообщение GPT-4o сгенерировано"
+    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(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", "", "Сообщение GigaChat-Pro сгенерировано"
+    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(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], "", "Сообщение GigaChat-Lite сгенерировано"
+    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(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"], "Все сообщения сгенерированы"
+    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"
@@ -169,31 +191,42 @@ 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":
-        return generate_message_gigachat_pro(full_prompt)
+        result = generate_message_gigachat_pro(full_prompt)
     elif model == "gigachat_lite":
-        return generate_message_gigachat_lite(full_prompt)
+        result = generate_message_gigachat_lite(full_prompt)
     elif model == "gigachat_plus":
-        return generate_message_gigachat_plus(full_prompt)
+        result = generate_message_gigachat_plus(full_prompt)
+    return clean_message(result)
 
-# Функция для выполнения персонализации с использованием yield
+# Обновляем блок персонализации
 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(gpt4o_message, personalization_prompt)
-    yield personalization_prompt, personalized_message_gpt4o, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
+    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(gigachat_pro_message, personalization_prompt, "gigachat_pro")
-    yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, "", "", "Персонализированное сообщение 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(gigachat_lite_message, personalization_prompt, "gigachat_lite")
-    yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, "", "Персонализированное сообщение 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(gigachat_plus_message, personalization_prompt, "gigachat_plus")
-    yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_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():
@@ -268,41 +301,43 @@ def check_errors_with_yield(*personalized_messages):
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
 
 
-def save_to_github(message, model_name):
-    # Формируем уникальное имя файла на основе текущего времени
-    timestamp = int(time.time())
-    file_name = f"file_{timestamp}.json"
-    
-    # Данные для записи
-    file_content = {
+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):
+    # Собираем все данные в один словарь
+    data_to_save = {
         "Модель": model_name,
-        "Сообщение": message
+        "Персонализированное сообщение": 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(file_content).encode()).decode()
+    file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
 
     # Параметры для GitHub API
     repo = "fruitpicker01/Storage_1"
-    path = file_name
+    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"Добавлен новый файл {file_name}",
+        "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(f"Файл {file_name} успешно добавлен в репозиторий.")
-    else:
-        print(f"Ошибка при добавлении файла: {response.status_code}")
-        print(response.json())
 
 
 # Создание интерфейса Gradio
@@ -342,7 +377,7 @@ with gr.Blocks() as demo:
             submit_btn = gr.Button("1. Создать неперсонализированное сообщение")  # Оранжевая кнопка по умолчанию
 
         with gr.Column(scale=2):
-            prompt_display = gr.Textbox(label="Неперсонализированный промпт", lines=28, interactive=False)
+            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)
@@ -369,6 +404,21 @@ with gr.Blocks() as demo:
                 personalized_output_text_gigachat_plus := gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=6, interactive=False)
             ]
         )
+
+    with gr.Row():
+        gr.Markdown("*Комментарий (опционально):*")
+        comment_gpt4o = gr.Textbox(label="", lines=3)
+        comment_gigachat_pro = gr.Textbox(label="", lines=3)
+        comment_gigachat_lite = gr.Textbox(label="", lines=3)
+        comment_gigachat_plus = gr.Textbox(label="", lines=3)
+
+    with gr.Row():
+        gr.Markdown("*Откорректированное сообщение (опционально):*")
+        corrected_gpt4o = gr.Textbox(label="", lines=3)
+        corrected_gigachat_pro = gr.Textbox(label="", lines=3)
+        corrected_gigachat_lite = gr.Textbox(label="", lines=3)
+        corrected_gigachat_plus = gr.Textbox(label="", lines=3)
+
     
     # Отдельная строка для кнопок с использованием пустой колонки
     with gr.Row():
@@ -379,15 +429,93 @@ with gr.Blocks() as demo:
         save_gigachat_plus_btn = gr.Button("👍")
 
         # Привязка кнопок к функциям сохранения
-        save_gpt4o_btn.click(fn=lambda x: save_to_github(x, "GPT-4o"), inputs=personalized_output_text_gpt4o, outputs=None)
-        save_gigachat_pro_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Pro"), inputs=personalized_output_text_gigachat_pro, outputs=None)
-        save_gigachat_lite_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite"), inputs=personalized_output_text_gigachat_lite, outputs=None)
-        save_gigachat_plus_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite+"), inputs=personalized_output_text_gigachat_plus, 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
+        )
 
+        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
+        )
 
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():
-        check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2)", elem_id="check_errors_button")
+        check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2) - экспериментальная фича, качество пока крайне низкое", elem_id="check_errors_button")
         
     with gr.Row():
         check_errors_btn.click(