Update app.py

app.py

@@ -14,27 +14,6 @@ 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)
@@ -107,7 +86,7 @@ def generate_message_gpt4o(prompt, temperature=1):
             "model": "chatgpt-4o-latest",
             "messages": [{"role": "system", "content": prompt}],
             "max_tokens": 101,
-            "temperature": gpt4o_temp  # Передача температуры
+            "temperature": temperature  # Передача температуры
         }
         response = requests.post("https://api.openai.com/v1/chat/completions", json=data, headers=headers)
         response_data = response.json()
@@ -125,10 +104,10 @@ 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=gigachat_pro_temp, verify_ssl_certs=False)
+        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
@@ -139,7 +118,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=gigachat_lite_temp, verify_ssl_certs=False)
+        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
@@ -150,7 +129,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=gigachat_plus_temp, verify_ssl_certs=False)
+        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
@@ -423,17 +402,10 @@ with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
     
     # Добавление элементов управления температурой для каждой модели
-    gpt4o_slider = gr.Slider(label="GPT-4o: temperature", minimum=0, maximum=2, step=0.01, value=gpt4o_temp)
-    gpt4o_slider.change(set_gpt4o_temp, inputs=gpt4o_slider, outputs=None)
-
-    gigachat_pro_slider = gr.Slider(label="GigaChat-Pro: temperature", minimum=0, maximum=2, step=0.01, value=gigachat_pro_temp)
-    gigachat_pro_slider.change(set_gigachat_pro_temp, inputs=gigachat_pro_slider, outputs=None)
-
-    gigachat_lite_slider = gr.Slider(label="GigaChat-Lite: temperature", minimum=0, maximum=2, step=0.01, value=gigachat_lite_temp)
-    gigachat_lite_slider.change(set_gigachat_lite_temp, inputs=gigachat_lite_slider, outputs=None)
-
-    gigachat_plus_slider = gr.Slider(label="GigaChat-Plus: temperature", minimum=0, maximum=2, step=0.01, value=gigachat_plus_temp)
-    gigachat_plus_slider.change(set_gigachat_plus_temp, inputs=gigachat_plus_slider, outputs=None)
+    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):
@@ -480,10 +452,10 @@ with gr.Blocks() as demo:
                 description_input, 
                 advantages_input, 
                 *selections, 
-                gpt4o_temp, 
-                gigachat_pro_temp, 
-                gigachat_lite_temp, 
-                gigachat_plus_temp
+                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]
         )
@@ -546,7 +518,7 @@ with gr.Blocks() as demo:
                 selections[3],  # Стадия бизнеса
                 selections[4],  # Отрасль
                 selections[5],  # ОПФ
-                gpt4o_temp  # Передача температуры
+                gpt4o_temperature  # Передача температуры
             ],
             outputs=None
         )