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=1):
     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()) + f" {temperature}"
     except Exception as e:
         return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
@@ -102,29 +103,32 @@ def clean_message(message):
     return message
 
 # Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
-def generate_message_gigachat_pro(prompt):
+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)
         res = chat_pro(messages)
         cleaned_message = clean_message(res.content.strip())
         return cleaned_message
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-def generate_message_gigachat_lite(prompt):
+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)
         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=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)
         res = chat_plus(messages)
         cleaned_message = clean_message(res.content.strip())
         return cleaned_message
@@ -159,9 +163,43 @@ def generate_message_gigachat_plus_with_retry(prompt):
             return message
     return message
 
+# Создание классов для каждой модели
+class GPT4oModel:
+    def __init__(self, temperature=1):
+        self.temperature = temperature
+
+    def generate_message(self, prompt):
+        return generate_message_gpt4o(prompt, temperature=self.temperature)
+
+class GigaChatProModel:
+    def __init__(self, temperature=0.87):
+        self.temperature = temperature
+
+    def generate_message(self, prompt):
+        return generate_message_gigachat_pro(prompt, temperature=self.temperature)
+
+class GigaChatLiteModel:
+    def __init__(self, temperature=0.87):
+        self.temperature = temperature
+
+    def generate_message(self, prompt):
+        return generate_message_gigachat_lite(prompt, temperature=self.temperature)
+
+class GigaChatPlusModel:
+    def __init__(self, temperature=0.87):
+        self.temperature = temperature
+
+    def generate_message(self, prompt):
+        return generate_message_gigachat_plus(prompt, temperature=self.temperature)
 
 # Обновляем генерацию сообщений для отображения в интерфейсе
-def generate_messages(description, advantages, *selected_values):
+def generate_messages(description, advantages, *selected_values, gpt4o_temp, gigachat_pro_temp, gigachat_lite_temp, gigachat_plus_temp):
+    # Создаем экземпляры классов с переданными значениями температур
+    gpt4o_model = GPT4oModel(temperature=gpt4o_temp)
+    gigachat_pro_model = GigaChatProModel(temperature=gigachat_pro_temp)
+    gigachat_lite_model = GigaChatLiteModel(temperature=gigachat_lite_temp)
+    gigachat_plus_model = GigaChatPlusModel(temperature=gigachat_plus_temp)
+
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -174,34 +212,32 @@ def generate_messages(description, advantages, *selected_values):
 
     yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
-    results["gpt4o"] = generate_message_gpt4o_with_retry(standard_prompt)
+    results["gpt4o"] = gpt4o_model.generate_message(standard_prompt)
     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"] = gigachat_pro_model.generate_message(standard_prompt)
     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"] = gigachat_lite_model.generate_message(standard_prompt)
     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"] = gigachat_plus_model.generate_message(standard_prompt)
     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"
@@ -246,7 +282,6 @@ def perform_personalization_gigachat_with_retry(standard_message, personalizatio
             return message
     return message
 
-
 # Обновляем блок персонализации
 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)
@@ -272,7 +307,6 @@ def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigacha
     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():
     prompt = (
@@ -317,7 +351,6 @@ def generate_error_check_prompt():
     )
     return prompt
 
-
 # Функция для выполнения проверки текста с использованием yield
 def check_errors_with_yield(*personalized_messages):
     if len(personalized_messages) < 4:
@@ -345,11 +378,11 @@ 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,
+        "Температура": temperature,  # Добавляем температуру
         "Персонализированное сообщение": personalized_message,
         "Комментарий": comment,
         "Откорректированное сообщение": corrected_message,
@@ -384,11 +417,22 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
     # Отправка POST-запроса на GitHub API для создания файла в репозитории
     response = requests.put(url, headers=headers, data=json.dumps(data))
 
+    if response.status_code == 201:
+        print("Файл успешно загружен на GitHub.")
+    else:
+        print(f"Ошибка при загрузке файла на GitHub: {response.status_code}")
+        print(f"Ответ сервера: {response.json()}")
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
     
+    # Добавление элементов управления температурой для каждой модели
+    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=2, step=0.01, value=0.87)
+    gigachat_lite_temperature = gr.Slider(label="GigaChat-Lite: temperature", minimum=0, maximum=2, step=0.01, value=0.87)
+    gigachat_plus_temperature = gr.Slider(label="GigaChat-Plus: temperature", minimum=0, maximum=2, step=0.01, value=0.87)
+
     with gr.Row():
         with gr.Column(scale=1):
             description_input = gr.Textbox(
@@ -428,11 +472,19 @@ with gr.Blocks() as demo:
             output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3, interactive=False)
             output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite+", lines=3, interactive=False)
             
-            submit_btn.click(
-                generate_messages, 
-                inputs=[description_input, advantages_input] + selections, 
-                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
-            )
+        submit_btn.click(
+            generate_messages, 
+            inputs=[
+                description_input, 
+                advantages_input, 
+                *selections, 
+                gpt4o_temperature, 
+                gigachat_pro_temperature, 
+                gigachat_lite_temperature, 
+                gigachat_plus_temperature
+            ], 
+            outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
+        )
 
     with gr.Row():
         personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
@@ -464,7 +516,6 @@ with gr.Blocks() as demo:
         corrected_gigachat_lite = gr.Textbox(label="", lines=3)
         corrected_gigachat_plus = gr.Textbox(label="", lines=3)
 
-    
     # Отдельная строка для кнопок с использованием пустой колонки
     with gr.Row():
         gr.Button("Жми 👍 для сохранения удачного SMS в базу =>")
@@ -475,8 +526,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,
@@ -490,14 +541,15 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gpt4o_temperature  # Передача температуры
             ],
             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),
+            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,
@@ -511,14 +563,15 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gigachat_pro_temperature  # Передача температуры
             ],
             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),
+            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,
@@ -532,14 +585,15 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gigachat_lite_temperature  # Передача температуры
             ],
             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),
+            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,
@@ -553,7 +607,8 @@ with gr.Blocks() as demo:
                 selections[2],  # Психотип
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
-                selections[5]   # ОПФ
+                selections[5],  # ОПФ
+                gigachat_plus_temperature  # Передача температуры
             ],
             outputs=None
         )
@@ -575,5 +630,4 @@ with gr.Blocks() as demo:
             ]
         )
 
-
     demo.launch()