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', max_tokens=512, verify_ssl_certs=False)
-chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=512, verify_ssl_certs=False)
-chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=512, verify_ssl_certs=False)
+chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=70, verify_ssl_certs=False)
+chat_lite = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=70, verify_ssl_certs=False)
+chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=70, verify_ssl_certs=False)
 
 # Загрузка данных из Excel-файла
 try:
@@ -93,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}"
 
@@ -106,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}"
 
@@ -115,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)
 
@@ -179,16 +191,18 @@ 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, "", "", "", "", "Промпт для персонализации сгенерирован"