Update app.py

app.py

@@ -14,10 +14,8 @@ 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)
+def authenticate_gigachat(model, max_tokens, temperature):
+    return GigaChat(credentials=gc_key, model=model, max_tokens=max_tokens, temperature=temperature, verify_ssl_certs=False)
 
 # Загрузка данных из Excel-файла
 try:
@@ -107,10 +105,10 @@ def clean_message(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)
+        chat_pro = authenticate_gigachat('GigaChat-Pro', max_tokens=68, temperature=temperature)
         res = chat_pro(messages)
         cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
+        return cleaned_message + f" {temperature}"
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
@@ -118,10 +116,10 @@ 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)
+        chat_lite = authenticate_gigachat('GigaChat', max_tokens=68, temperature=temperature)
         res = chat_lite(messages)
         cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
+        return cleaned_message + f" {temperature}"
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite: {e}"
 
@@ -129,13 +127,14 @@ 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)
+        chat_plus = authenticate_gigachat('GigaChat-Plus', max_tokens=68, temperature=temperature)
         res = chat_plus(messages)
         cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
+        return cleaned_message + f" {temperature}"
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
+
 def generate_message_gpt4o_with_retry(prompt):
     for _ in range(10):  # Максимум 10 попыток
         message = generate_message_gpt4o(prompt)
@@ -166,7 +165,7 @@ def generate_message_gigachat_plus_with_retry(prompt):
 
 
 # Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values):
+def generate_messages(description, advantages, *selected_values, gpt4o_temperature, gigachat_pro_temperature, gigachat_lite_temperature, gigachat_plus_temperature):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -179,26 +178,26 @@ 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(standard_prompt, temperature=gpt4o_temperature)
     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_pro_with_retry(standard_prompt, temperature=gigachat_pro_temperature)
     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_lite_with_retry(standard_prompt, temperature=gigachat_lite_temperature)
     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_plus_with_retry(standard_prompt, temperature=gigachat_plus_temperature)
     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, "Все сообщения сгенерированы"
@@ -206,7 +205,6 @@ def generate_messages(description, advantages, *selected_values):
     return results
 
 
-
 # Функция для генерации персонализированного промпта
 def generate_personalization_prompt(*selected_values):
     prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"