Update app.py

app.py

@@ -13,6 +13,7 @@ gc_key = os.getenv('GC_KEY')
 # Авторизация в сервисе GigaChat
 chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
 chat_lite = GigaChat(credentials=gc_key, model='GigaChat', verify_ssl_certs=False)
+chat_plus = GigaChat(credentials=gc_key, model='GigaChat-Plus', verify_ssl_certs=False)
 
 # Загрузка данных из Excel-файла
 try:
@@ -71,7 +72,7 @@ def generate_standard_prompt(description, advantages, *selected_values):
     )
     return prompt
 
-# Функция для генерации сообщения с помощью chatgpt-4o-latest
+# Функция для генерации сообщения с помощью ChatGPT-4o-Latest
 def generate_message_gpt4o(prompt):
     try:
         headers = {
@@ -87,7 +88,7 @@ def generate_message_gpt4o(prompt):
         response_data = response.json()
         return response_data["choices"][0]["message"]["content"].strip()
     except Exception as e:
-        return f"Ошибка при обращении к chatgpt-4o-latest: {e}"
+        return f"Ошибка при обращении к ChatGPT-4o-Latest: {e}"
 
 # Функция для генерации сообщения с помощью GigaChat-Pro
 def generate_message_gigachat_pro(prompt):
@@ -98,7 +99,7 @@ def generate_message_gigachat_pro(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Pro: {e}"
 
-# Функция для генерации сообщения с помощью GigaChat
+# Функция для генерации сообщения с помощью GigaChat-Lite
 def generate_message_gigachat_lite(prompt):
     try:
         time.sleep(2)  # Задержка 2 секунды перед запросом к GigaChat
@@ -108,13 +109,41 @@ def generate_message_gigachat_lite(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite: {e}"
 
-# Функция для обработки нажатия кнопки Submit
+# Функция для генерации сообщения с помощью GigaChat-Lite+
+def generate_message_gigachat_plus(prompt):
+    try:
+        time.sleep(2)  # Задержка 2 секунды перед запросом к GigaChat
+        messages = [SystemMessage(content=prompt)]
+        res = chat_plus(messages)
+        return res.content.strip()
+    except Exception as e:
+        return f"Ошибка при обращении к GigaChat-Lite+: {e}"
+
+# Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
 def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
-    message_gpt4o = generate_message_gpt4o(standard_prompt)
-    message_gigachat_pro = generate_message_gigachat_pro(standard_prompt)
-    message_gigachat_lite = generate_message_gigachat_lite(standard_prompt)
-    return standard_prompt, message_gpt4o, message_gigachat_pro, message_gigachat_lite
+
+    results = {
+        "prompt": standard_prompt,
+        "gpt4o": None,
+        "gigachat_pro": None,
+        "gigachat_lite": None,
+        "gigachat_plus": None
+    }
+
+    yield results["prompt"], "", "", "", ""
+
+    results["gpt4o"] = generate_message_gpt4o(standard_prompt)
+    yield results["prompt"], results["gpt4o"], "", "", ""
+
+    results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", ""
+
+    results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], ""
+
+    results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"]
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -135,11 +164,12 @@ with gr.Blocks() as demo:
             output_text_gpt4o = gr.Textbox(label="Создание сообщения GPT-4o", lines=3)
             output_text_gigachat_pro = gr.Textbox(label="Создание сообщения GigaChat-Pro", lines=3)
             output_text_gigachat_lite = gr.Textbox(label="Создание сообщения GigaChat-Lite", lines=3)
+            output_text_gigachat_plus = gr.Textbox(label="Создание сообщения GigaChat-Lite+", lines=3)
             
             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]
+                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus]
             )
 
     demo.launch()