Update app.py

app.py

@@ -136,38 +136,38 @@ def generate_message_gigachat_plus(prompt, temperature=0.87):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Plus: {e}"
 
-def generate_message_gpt4o_with_retry(prompt):
+def generate_message_gpt4o_with_retry(prompt, gpt4o_temperature):
     for _ in range(10):  # Максимум 10 попыток
-        message = generate_message_gpt4o(prompt)
+        message = generate_message_gpt4o(prompt, gpt4o_temperature)
         if len(message) <= 250:
             return message
     return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
-def generate_message_gigachat_pro_with_retry(prompt):
+def generate_message_gigachat_pro_with_retry(prompt, gigachat_pro_temperature):
     for _ in range(10):
-        message = generate_message_gigachat_pro(prompt)
+        message = generate_message_gigachat_pro(prompt, gigachat_pro_temperature)
         if len(message) <= 250:
             return message
     return message
 
-def generate_message_gigachat_lite_with_retry(prompt):
+def generate_message_gigachat_lite_with_retry(prompt, gigachat_lite_temperature):
     for _ in range(10):
-        message = generate_message_gigachat_lite(prompt)
+        message = generate_message_gigachat_lite(prompt, gigachat_lite_temperature)
         if len(message) <= 250:
             return message
     return message
 
-def generate_message_gigachat_plus_with_retry(prompt):
+def generate_message_gigachat_plus_with_retry(prompt, gigachat_plus_temperature):
     for _ in range(10):
-        message = generate_message_gigachat_plus(prompt)
+        message = generate_message_gigachat_plus(prompt, gigachat_plus_temperature)
         if len(message) <= 250:
             return message
     return message
 
-
+                
 # Обновляем генерацию сообщений для отображения в интерфейсе
 def generate_messages(description, advantages, *selected_values):
-    standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
+    standard_prompt = generate_standard_prompt(description, advantages, *selected_values, gpt4o_temperature, gigachat_pro_temperature, gigachat_lite_temperature, gigachat_plus_temperature)
 
     results = {
         "prompt": standard_prompt,
@@ -179,26 +179,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, 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, 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, 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, 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, "Все сообщения сгенерированы"