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,14 +85,16 @@ 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}"
 
+
 def clean_message(message):
     # Если сообщение не заканчивается на точку или восклицательный знак, обрезаем его до последней точки
     if not message.endswith(('.', '!', '?')):
@@ -102,29 +104,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
@@ -346,10 +351,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,
@@ -385,9 +391,17 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
     response = requests.put(url, headers=headers, data=json.dumps(data))
 
 
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
+
+    # Добавление элементов управления температурой для каждой модели
+    gpt4o_temperature = gr.Slider(label="GPT-4o: temperature", minimum=0, maximum=1.7, 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):
@@ -430,7 +444,15 @@ with gr.Blocks() as demo:
             
             submit_btn.click(
                 generate_messages, 
-                inputs=[description_input, advantages_input] + selections, 
+                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]
             )