Update app.py

app.py

@@ -234,30 +234,28 @@ def generate_error_check_prompt():
 
 
 # Функция для выполнения проверки текста с использованием yield
-def check_errors_with_yield(*personalized_messages):
-    if len(personalized_messages) < 4:
+def check_errors_with_yield(personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus):
+    if not personalized_message_gpt4o or not personalized_message_gigachat_pro or not personalized_message_gigachat_lite or not personalized_message_gigachat_plus:
         yield "", "", "", "", "", "Ошибка: недостаточно сообщений для проверки"
         return
 
     error_check_prompt = generate_error_check_prompt()
     yield error_check_prompt, "", "", "", "", "Промпт для проверки текста сгенерирован"
 
-    error_message_gpt4o = perform_personalization(f"{error_check_prompt}\n\n{personalized_messages[0]}", "")
+    error_message_gpt4o = perform_personalization(f"{error_check_prompt}\n\n{personalized_message_gpt4o}", "")
     yield error_check_prompt, error_message_gpt4o, "", "", "", "Результат проверки GPT-4o сгенерирован"
 
-    error_message_gigachat_pro = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[1]}", "", "gigachat_pro")
+    error_message_gigachat_pro = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_message_gigachat_pro}", "", "gigachat_pro")
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, "", "", "Результат проверки GigaChat-Pro сгенерирован"
-    
-    time.sleep(3)
-    error_message_gigachat_lite = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[2]}", "", "gigachat_lite")
+
+    error_message_gigachat_lite = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_message_gigachat_lite}", "", "gigachat_lite")
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, "", "Результат проверки GigaChat-Lite сгенерирован"
-    
+
     try:
-        time.sleep(3)
-        error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_messages[3]}", "", "gigachat_plus")
+        error_message_gigachat_plus = perform_personalization_gigachat(f"{error_check_prompt}\n\n{personalized_message_gigachat_plus}", "", "gigachat_plus")
     except Exception as e:
         error_message_gigachat_plus = f"Ошибка при обработке GigaChat-Plus: {e}"
-        
+
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
 
 
@@ -303,10 +301,6 @@ 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)
-            personalized_output_text_gpt4o = gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6, interactive=False)
-            personalized_output_text_gigachat_pro = gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=6, interactive=False)
-            personalized_output_text_gigachat_lite = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=6, interactive=False)
-            personalized_output_text_gigachat_plus = gr.Textbox(label="Персонализированное сообщение GigaChat-Plus", lines=6, interactive=False)
             gr.Markdown("**Статус выполнения**")
             status_display = gr.Textbox(label="", lines=4, interactive=False, elem_classes="status-box")  # Новый блок для отображения статуса
             
@@ -324,15 +318,16 @@ with gr.Blocks() as demo:
             personalize_messages_with_yield,
             inputs=[output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus] + selections,
             outputs=[
-                personalized_output_text_gpt4o,  # Сохранение персонализированного текста GPT-4o
-                personalized_output_text_gigachat_pro,  # Сохранение персонализированного текста GigaChat-Pro
-                personalized_output_text_gigachat_lite,  # Сохранение персонализированного текста GigaChat-Lite
-                personalized_output_text_gigachat_plus,  # Сохра��ение персонализированного текста GigaChat-Plus
+                gr.Textbox(label="Промпт для персонализации", lines=6, interactive=False),
+                gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6),
+                gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=6),
+                gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=6),
+                gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=6),
                 status_display
             ]
         )
 
-   
+    
     with gr.Row():
         check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2)", elem_id="check_errors_button")
         
@@ -345,7 +340,7 @@ with gr.Blocks() as demo:
                 gr.Textbox(label="Результат проверки GPT-4o", lines=6),
                 gr.Textbox(label="Результат проверки GigaChat-Pro", lines=6),
                 gr.Textbox(label="Результат проверки GigaChat-Lite", lines=6),
-                gr.Textbox(label="Результат проверки GigaChat-Plus", lines=6),
+                gr.Textbox(label="Результат проверки GigaChat-Lite+", lines=6),
                 status_display
             ]
         )