Update app.py

app.py

@@ -264,6 +264,25 @@ def check_errors_with_yield(*personalized_messages):
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
 
 
+# Функция для сохранения сообщения в файл Excel
+def save_to_excel(message, model_name):
+    file_path = "База удачных SMS.xlsx"
+    
+    # Проверяем, существует ли файл
+    if os.path.exists(file_path):
+        df = pd.read_excel(file_path)
+    else:
+        # Если файл не существует, создаем новый DataFrame
+        df = pd.DataFrame(columns=["Модель", "Сообщение"])
+
+    # Добавляем новую запись
+    new_row = {"Модель": model_name, "Сообщение": message}
+    df = df.append(new_row, ignore_index=True)
+
+    # Сохраняем DataFrame в Excel
+    df.to_excel(file_path, index=False)
+
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -337,6 +356,12 @@ with gr.Blocks() as demo:
         gr.Button("👍")
         gr.Button("👍")
 
+    # Привязка функций сохранения к кнопкам
+    save_btn1.click(lambda: save_to_excel(personalized_output_text_gpt4o.value, "GPT-4o"))
+    save_btn2.click(lambda: save_to_excel(personalized_output_text_gigachat_pro.value, "GigaChat-Pro"))
+    save_btn3.click(lambda: save_to_excel(personalized_output_text_gigachat_lite.value, "GigaChat-Lite"))
+    save_btn4.click(lambda: save_to_excel(personalized_output_text_gigachat_plus.value, "GigaChat-Plus"))
+
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():
         check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2)", elem_id="check_errors_button")