Update app.py

app.py

@@ -127,25 +127,26 @@ def generate_messages(description, advantages, *selected_values):
         "gigachat_plus": None
     }
 
-    yield results["prompt"], "", "", "", ""
+    yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
     results["gpt4o"] = generate_message_gpt4o(standard_prompt)
-    yield results["prompt"], results["gpt4o"], "", "", ""
+    yield results["prompt"], results["gpt4o"], "", "", "", "Сообщение GPT-4o сгенерировано"
 
     results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", ""
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", "", "Сообщение GigaChat-Pro сгенерировано"
 
     results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], ""
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], "", "Сообщение 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"]
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"], "Все сообщения сгенерированы"
 
     # Возвращаем результаты для дальнейшей персонализации
     return results
 
 # Функция для выполнения персонализации
 def personalize_messages(results, *selected_values):
+    status_text = ""
     # Промпты для персонализации
     personalization_prompt_gpt4o = f"Пожалуйста, адаптируйте следующее сообщение: '{results['gpt4o']}' под следующие требования:\n"
     personalization_prompt_gigachat_pro = f"Пожалуйста, адаптируйте следующее сообщение: '{results['gigachat_pro']}' под следующие требования:\n"
@@ -161,37 +162,47 @@ def personalize_messages(results, *selected_values):
             personalization_prompt_gigachat_lite += requirement
             personalization_prompt_gigachat_plus += requirement
     
-    print("Промпты для персонализации созданы успешно.")
-    print(f"GPT-4o: {personalization_prompt_gpt4o}")
-    print(f"GigaChat-Pro: {personalization_prompt_gigachat_pro}")
-    print(f"GigaChat-Lite: {personalization_prompt_gigachat_lite}")
-    print(f"GigaChat-Plus: {personalization_prompt_gigachat_plus}")
+    status_text += "Промпты для персонализации созданы успешно.\n"
     
     # Персонализированные сообщения
     try:
         personalized_message_gpt4o = generate_message_gpt4o(personalization_prompt_gpt4o)
+        status_text += "Сообщение GPT-4o персонализировано успешно.\n"
     except Exception as e:
         personalized_message_gpt4o = f"Ошибка при персонализации GPT-4o: {e}"
+        status_text += f"{personalized_message_gpt4o}\n"
     
     try:
         time.sleep(2)
         personalized_message_gigachat_pro = generate_message_gigachat_pro(personalization_prompt_gigachat_pro)
+        status_text += "Сообщение GigaChat-Pro персонализировано успешно.\n"
     except Exception as e:
         personalized_message_gigachat_pro = f"Ошибка при персонализации GigaChat-Pro: {e}"
+        status_text += f"{personalized_message_gigachat_pro}\n"
 
     try:
         time.sleep(2)
         personalized_message_gigachat_lite = generate_message_gigachat_lite(personalization_prompt_gigachat_lite)
+        status_text += "Сообщение GigaChat-Lite персонализировано успешно.\n"
     except Exception as e:
         personalized_message_gigachat_lite = f"Ошибка при персонализации GigaChat-Lite: {e}"
+        status_text += f"{personalized_message_gigachat_lite}\n"
 
     try:
         time.sleep(2)
         personalized_message_gigachat_plus = generate_message_gigachat_plus(personalization_prompt_gigachat_plus)
+        status_text += "Сообщение GigaChat-Plus персонализировано успешно.\n"
     except Exception as e:
-        personalized_message_gigachat_plus = f"Ошибка при персонализации GigaChat-Lite+: {e}"
+        personalized_message_gigachat_plus = f"Ошибка при персонализации GigaChat-Plus: {e}"
+        status_text += f"{personalized_message_gigachat_plus}\n"
     
-    return personalization_prompt_gpt4o, personalized_message_gpt4o, personalization_prompt_gigachat_pro, personalized_message_gigachat_pro, personalization_prompt_gigachat_lite, personalized_message_gigachat_lite, personalization_prompt_gigachat_plus, personalized_message_gigachat_plus
+    return (
+        personalization_prompt_gpt4o, personalized_message_gpt4o,
+        personalization_prompt_gigachat_pro, personalized_message_gigachat_pro,
+        personalization_prompt_gigachat_lite, personalized_message_gigachat_lite,
+        personalization_prompt_gigachat_plus, personalized_message_gigachat_plus,
+        status_text
+    )
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -235,30 +246,32 @@ with gr.Blocks() as demo:
             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)
+            status_display = gr.Textbox(label="Статус выполнения", lines=4, 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]
+                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
             )
 
-        with gr.Row():
-            gr.Markdown("### Выполнить персонализацию")
-
-            personalize_btn = gr.Button("Выполнить персонализацию")
-            personalize_btn.click(
-                personalize_messages,
-                inputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus] + selections,
-                outputs=[
-                    gr.Textbox(label="Промпт для персонализации GPT-4o", lines=6, interactive=False),
-                    gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=3),
-                    gr.Textbox(label="Промпт для персонализации GigaChat-Pro", lines=6, interactive=False),
-                    gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=3),
-                    gr.Textbox(label="Промпт для персонализации GigaChat-Lite", lines=6, interactive=False),
-                    gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=3),
-                    gr.Textbox(label="Промпт для персонализации GigaChat-Lite+", lines=6, interactive=False),
-                    gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=3)
-                ]
-            )
+    with gr.Row():
+        personalize_btn = gr.Button("Выполнить персонализацию", elem_id="personalize_button")
+        
+    with gr.Row():
+        personalize_btn.click(
+            personalize_messages,
+            inputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus] + selections,
+            outputs=[
+                gr.Textbox(label="Промпт для персонализации GPT-4o", lines=6, interactive=False),
+                gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=3),
+                gr.Textbox(label="Промпт для персонализации GigaChat-Pro", lines=6, interactive=False),
+                gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=3),
+                gr.Textbox(label="Промпт для персонализации GigaChat-Lite", lines=6, interactive=False),
+                gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=3),
+                gr.Textbox(label="Промпт для персонализации GigaChat-Lite+", lines=6, interactive=False),
+                gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=3),
+                status_display
+            ]
+        )
 
     demo.launch()