Update app.py

app.py

@@ -76,7 +76,7 @@ def generate_standard_prompt(description, advantages, *selected_values):
     return prompt
 
 # Функции для генерации сообщений
-def generate_message_gpt4o(prompt):
+def generate_message_gpt4o(prompt, temperature):
     try:
         headers = {
             "Content-Type": "application/json",
@@ -85,11 +85,12 @@ def generate_message_gpt4o(prompt):
         data = {
             "model": "chatgpt-4o-latest",
             "messages": [{"role": "system", "content": prompt}],
-            "max_tokens": 101
+            "max_tokens": 101,
+            "temperature": temperature
         }
         response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
         response_data = response.json()
-        return response_data["choices"][0]["message"]["content"].strip()
+        return clean_message(response_data["choices"][0]["message"]["content"].strip())
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
@@ -102,29 +103,29 @@ def clean_message(message):
     return message
 
 # Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
-def generate_message_gigachat_pro(prompt):
+def generate_message_gigachat_pro(prompt, temperature):
     try:
         messages = [SystemMessage(content=prompt)]
+        chat_pro = authenticate_gigachat('GigaChat-Pro', 68, temperature)
         res = chat_pro(messages)
-        cleaned_message = clean_message(res.content.strip())
-        return cleaned_message
+        return clean_message(res.content.strip())
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-def generate_message_gigachat_lite(prompt):
+def generate_message_gigachat_lite(prompt, temperature):
     try:
-        time.sleep(2)
         messages = [SystemMessage(content=prompt)]
+        chat_lite = authenticate_gigachat('GigaChat', 68, 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):
+def generate_message_gigachat_plus(prompt, temperature):
     try:
-        time.sleep(2)
         messages = [SystemMessage(content=prompt)]
+        chat_plus = authenticate_gigachat('GigaChat-Plus', 68, temperature)
         res = chat_plus(messages)
         cleaned_message = clean_message(res.content.strip())
         return cleaned_message
@@ -231,16 +232,16 @@ def perform_personalization_gigachat(standard_message, personalization_prompt, m
     return clean_message(result)
 
 
-def perform_personalization_with_retry(standard_message, personalization_prompt):
+def perform_personalization_with_retry(standard_message, personalization_prompt, temperature):
     for _ in range(10):  # Максимум 10 попыток
-        message = perform_personalization(standard_message, personalization_prompt)
+        message = perform_personalization(standard_message, personalization_prompt, temperature)
         if len(message) <= 250:
             return message
     return message  # Возвращаем последнее сгенерированное сообщение, если все попытки не удались
 
-def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model):
+def perform_personalization_gigachat_with_retry(standard_message, personalization_prompt, model, temperature):
     for _ in range(10):
-        message = perform_personalization_gigachat(standard_message, personalization_prompt, model)
+        message = perform_personalization_gigachat(standard_message, personalization_prompt, model, temperature)
         if len(message) <= 250:
             return message
     return message
@@ -345,7 +346,7 @@ 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(personalized_message, model_name, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form):
+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, temperature):
     # Собираем все данные в один словарь
     data_to_save = {
         "Модель": model_name,
@@ -361,7 +362,8 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
         "Психотип": psychotype,
         "Стадия бизнеса": business_stage,
         "Отрасль": industry,
-        "ОПФ": legal_form
+        "ОПФ": legal_form,
+        "Температура": temperature  # Добавляем температуру в сохраненные данные
     }
 
     # Преобразуем контент в JSON-строку и кодируем в base64
@@ -388,6 +390,13 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
     
+    with gr.Column(scale=1):
+        # Добавляем слайдеры для установки параметра температуры для каждой модели
+        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(scale=1):
             description_input = gr.Textbox(
@@ -474,8 +483,8 @@ with gr.Blocks() as demo:
 
         # Привязка кнопок к функциям сохранения
         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),
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature: 
+                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, temperature),
             inputs=[
                 personalized_output_text_gpt4o,
                 comment_gpt4o,
@@ -489,14 +498,17 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gpt4o_temperature  # Температура
             ],
             outputs=None
         )
 
+
+        # Привязка кнопок к функциям сохранения для GigaChat-Pro
         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),
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature: 
+                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, temperature),
             inputs=[
                 personalized_output_text_gigachat_pro,
                 comment_gigachat_pro,
@@ -510,14 +522,16 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gigachat_pro_temperature  # Температура
             ],
             outputs=None
         )
-
+        
+        # Привязка кнопок к функциям сохранения для GigaChat-Lite
         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),
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature: 
+                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, temperature),
             inputs=[
                 personalized_output_text_gigachat_lite,
                 comment_gigachat_lite,
@@ -531,14 +545,16 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gigachat_lite_temperature  # Температура
             ],
             outputs=None
         )
-
+        
+        # Привязка кнопок к функциям сохранения для GigaChat-Plus
         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),
+            fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature: 
+                save_to_github(personalized_message, "GigaChat-Plus", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, temperature),
             inputs=[
                 personalized_output_text_gigachat_plus,
                 comment_gigachat_plus,
@@ -552,11 +568,13 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gigachat_plus_temperature  # Температура
             ],
             outputs=None
         )
 
+
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():
         check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2) - экспериментальная фича, качество пока крайне низкое", elem_id="check_errors_button")