Update app.py

app.py

@@ -14,6 +14,27 @@ openai_api_key = os.getenv('GPT_KEY')
 gc_key = os.getenv('GC_KEY')
 token = os.getenv('GITHUB_TOKEN')
 
+gpt4o_temp = 1.0
+gigachat_pro_temp = 0.87
+gigachat_lite_temp = 0.87
+gigachat_plus_temp = 0.87
+
+def set_gpt4o_temp(value):
+    global gpt4o_temp
+    gpt4o_temp = value
+
+def set_gigachat_pro_temp(value):
+    global gigachat_pro_temp
+    gigachat_pro_temp = value
+
+def set_gigachat_lite_temp(value):
+    global gigachat_lite_temp
+    gigachat_lite_temp = value
+
+def set_gigachat_plus_temp(value):
+    global gigachat_plus_temp
+    gigachat_plus_temp = value
+
 # Авторизация в сервисе 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)
@@ -86,7 +107,7 @@ def generate_message_gpt4o(prompt, temperature=1):
             "model": "chatgpt-4o-latest",
             "messages": [{"role": "system", "content": prompt}],
             "max_tokens": 101,
-            "temperature": temperature  # Передача температуры
+            "temperature": gpt4o_temp  # Передача температуры
         }
         response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
         response_data = response.json()
@@ -104,10 +125,10 @@ def clean_message(message):
     return message
 
 # Обновленные функции генерации сообщений с учетом обрезки незаконченных предложений
-def generate_message_gigachat_pro(prompt, temperature=0.87):
+def generate_message_gigachat_pro(prompt):
     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 = GigaChat(credentials=gc_key, model='GigaChat-Pro', max_tokens=68, temperature=gigachat_pro_temp, verify_ssl_certs=False)
         res = chat_pro(messages)
         cleaned_message = clean_message(res.content.strip())
         return cleaned_message
@@ -118,7 +139,7 @@ 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 = GigaChat(credentials=gc_key, model='GigaChat', max_tokens=68, temperature=gigachat_lite_temp, verify_ssl_certs=False)
         res = chat_lite(messages)
         cleaned_message = clean_message(res.content.strip())
         return cleaned_message
@@ -129,7 +150,7 @@ 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 = GigaChat(credentials=gc_key, model='GigaChat-Plus', max_tokens=68, temperature=gigachat_plus_temp, verify_ssl_certs=False)
         res = chat_plus(messages)
         cleaned_message = clean_message(res.content.strip())
         return cleaned_message
@@ -402,10 +423,10 @@ 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)
+    gpt4o_slider = gr.Slider(label="GPT-4o: temperature", minimum=0, maximum=2, step=0.01, value=1, change=set_gpt4o_temp)
+    gigachat_pro_slider = gr.Slider(label="GigaChat-Pro: temperature", minimum=0, maximum=2, step=0.01, value=0.87, change=set_gigachat_pro_temp)
+    gigachat_lite_slider = gr.Slider(label="GigaChat-Lite: temperature", minimum=0, maximum=2, step=0.01, value=0.87, change=set_gigachat_lite_temp)
+    gigachat_plus_slider = gr.Slider(label="GigaChat-Plus: temperature", minimum=0, maximum=2, step=0.01, value=0.87, change=set_gigachat_plus_temp)
 
     with gr.Row():
         with gr.Column(scale=1):