Update app.py

app.py

@@ -755,6 +755,17 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
     response = requests.put(url, headers=headers, data=json.dumps(data))
 
 
+def personalize_and_save(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, meta_llama_405b_message, meta_llama_70b_message, meta_llama_8b_message, gemma_27b_message, gemma_9b_message, mistral_message, description, advantages, *selected_values):
+    personalization_results = list(personalize_messages_with_yield(
+        gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, meta_llama_405b_message, meta_llama_70b_message, meta_llama_8b_message, gemma_27b_message, gemma_9b_message, mistral_message, *selected_values
+    ))
+
+    # Save to GitHub after the personalization
+    save_user_request_to_github(description, advantages, selected_values)
+
+    return personalization_results[-1]  # Return the last set of outputs from personalization
+
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -901,10 +912,7 @@ with gr.Blocks() as demo:
     )
     
     personalize_btn.click(
-        fn=lambda gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, meta_llama_405b_message, meta_llama_70b_message, meta_llama_8b_message, gemma_27b_message, gemma_9b_message, mistral_message, description, advantages, *selected_values: (
-            personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, meta_llama_405b_message, meta_llama_70b_message, meta_llama_8b_message, gemma_27b_message, gemma_9b_message, mistral_message, *selected_values),
-            save_user_request_to_github(description, advantages, selected_values)
-        ),
+        fn=personalize_and_save,
         inputs=[
             output_text_gpt4o, 
             output_text_gigachat_pro,