Update app.py

app.py

@@ -812,6 +812,21 @@ def personalize_and_save(
         save_user_request_to_github(description, advantages, selected_values)
 
 
+def clear_fields():
+    return (
+        "", "", "", "", "", "", "", "", "", "",  # personalized outputs and prompts
+        "", "", "", "", "", "", "", "", "", "",  # comment fields
+        "", "", "", "", "", "", "", "", "", "",  # corrected message fields
+    )
+
+def clear_personalization_fields():
+    return (
+        "", "", "", "", "", "", "", "", "", "",  # personalized outputs
+        "", "", "", "", "", "", "", "", "", "",  # comment fields
+        "", "", "", "", "", "", "", "", "", ""   # corrected message fields
+    )
+
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -940,32 +955,21 @@ with gr.Blocks() as demo:
    
     # Добавление функционала для кнопок
     submit_btn.click(
-        generate_messages, 
-        inputs=[description_input, advantages_input] + selections, 
+        clear_fields,
+        inputs=[],
         outputs=[
-            prompt_display, 
-            output_text_gpt4o, 
-            output_text_gigachat_pro, 
-            output_text_gigachat_lite, 
-            output_text_gigachat_plus, 
-            output_text_meta_llama_405b,
-            output_text_meta_llama_70b,
-            output_text_meta_llama_8b,
-            output_text_gemma_27b,
-            output_text_gemma_9b,
-            output_text_mistral,
-            personalization_prompt,  # Очищаем персонализированный промпт
-            personalized_output_text_gpt4o, 
-            personalized_output_text_gigachat_pro, 
-            personalized_output_text_gigachat_lite, 
-            personalized_output_text_gigachat_plus, 
+            personalization_prompt,
+            personalized_output_text_gpt4o,
+            personalized_output_text_gigachat_pro,
+            personalized_output_text_gigachat_lite,
+            personalized_output_text_gigachat_plus,
             personalized_output_text_meta_llama_405b,
             personalized_output_text_meta_llama_70b,
             personalized_output_text_meta_llama_8b,
             personalized_output_text_gemma_27b,
             personalized_output_text_gemma_9b,
             personalized_output_text_mistral,
-            comment_gpt4o,  # Очищаем комментарии
+            comment_gpt4o,  
             corrected_gpt4o,
             comment_gigachat_pro,
             corrected_gigachat_pro,
@@ -987,11 +991,12 @@ with gr.Blocks() as demo:
             corrected_mistral
         ]
     )
-
     
-    personalize_btn.click(
-        fn=personalize_and_save,
-        inputs=[
+    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, 
@@ -1001,12 +1006,15 @@ with gr.Blocks() as demo:
             output_text_meta_llama_8b,
             output_text_gemma_27b,
             output_text_gemma_9b,
-            output_text_mistral,
-            description_input,
-            advantages_input
-        ] + selections,
+            output_text_mistral
+        ]
+    )
+
+ 
+    personalize_btn.click(
+        fn=clear_personalization_fields,
+        inputs=[],
         outputs=[
-            personalization_prompt,
             personalized_output_text_gpt4o,
             personalized_output_text_gigachat_pro,
             personalized_output_text_gigachat_lite,
@@ -1016,8 +1024,8 @@ with gr.Blocks() as demo:
             personalized_output_text_meta_llama_8b,
             personalized_output_text_gemma_27b,
             personalized_output_text_gemma_9b,
-            personalized_output_text_mistral,
-            comment_gpt4o,  # Очищаем комментарии
+            personalized_output_text_mistral,  # Поля для персонализированных сообщений
+            comment_gpt4o,  
             corrected_gpt4o,
             comment_gigachat_pro,
             corrected_gigachat_pro,
@@ -1036,7 +1044,38 @@ with gr.Blocks() as demo:
             comment_gemma_9b,
             corrected_gemma_9b,
             comment_mistral,
-            corrected_mistral
+            corrected_mistral  # Поля для комментариев и откорректированных сообщений
+        ]
+    )
+    
+    personalize_btn.click(
+        fn=personalize_and_save,
+        inputs=[
+            output_text_gpt4o, 
+            output_text_gigachat_pro, 
+            output_text_gigachat_lite, 
+            output_text_gigachat_plus, 
+            output_text_meta_llama_405b,
+            output_text_meta_llama_70b,
+            output_text_meta_llama_8b,
+            output_text_gemma_27b,
+            output_text_gemma_9b,
+            output_text_mistral,  # Входные данные для персонализации
+            description_input,
+            advantages_input
+        ] + selections,
+        outputs=[
+            personalization_prompt,
+            personalized_output_text_gpt4o,
+            personalized_output_text_gigachat_pro,
+            personalized_output_text_gigachat_lite,
+            personalized_output_text_gigachat_plus,
+            personalized_output_text_meta_llama_405b,
+            personalized_output_text_meta_llama_70b,
+            personalized_output_text_meta_llama_8b,
+            personalized_output_text_gemma_27b,
+            personalized_output_text_gemma_9b,
+            personalized_output_text_mistral  # Выходные данные для персонализации
         ]
     )