Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 import os
+import json
 import pandas as pd
 import time
 from langchain.schema import SystemMessage
@@ -266,26 +267,24 @@ def check_errors_with_yield(*personalized_messages):
 
 
 # Функция для сохранения сообщения в файл Excel
-def save_to_excel(message, model_name):
-    file_path = "База удачных SMS.xlsx"
+def save_to_json(message, model_name):
+    file_path = "results.json"
     
     # Проверяем, существует ли файл
     if os.path.exists(file_path):
-        workbook = load_workbook(file_path)
-        sheet = workbook.active
+        with open(file_path, "r") as file:
+            data = json.load(file)
     else:
-        # Если файл не существует, создаем новый Workbook и лист
-        workbook = load_workbook()
-        sheet = workbook.active
-        # Добавляем заголовки колонок
-        sheet.append(["Модель", "Сообщение"])
-
-    # Находим первую пустую строку
-    new_row = [model_name, message]
-    sheet.append(new_row)
+        data = []
+    
+    # Добавляем новую запись
+    new_entry = {"Модель": model_name, "Сообщение": message}
+    data.append(new_entry)
+    
+    # Сохраняем данные в файл
+    with open(file_path, "w") as file:
+        json.dump(data, file, ensure_ascii=False, indent=4)
 
-    # Сохраняем изменения
-    workbook.save(file_path)
 
 
 # Создание интерфейса Gradio
@@ -362,10 +361,11 @@ with gr.Blocks() as demo:
         save_gigachat_plus_btn = gr.Button("👍")
 
         # Привязка кнопок к функциям сохранения
-        save_gpt4o_btn.click(fn=save_to_excel, inputs=[personalized_output_text_gpt4o], outputs=None)
-        save_gigachat_pro_btn.click(fn=save_to_excel, inputs=[personalized_output_text_gigachat_pro], outputs=None)
-        save_gigachat_lite_btn.click(fn=save_to_excel, inputs=[personalized_output_text_gigachat_lite], outputs=None)
-        save_gigachat_plus_btn.click(fn=save_to_excel, inputs=[personalized_output_text_gigachat_plus], outputs=None)
+        save_gpt4o_btn.click(fn=lambda x: save_to_json(x, "GPT-4o"), inputs=personalized_output_text_gpt4o, outputs=None)
+        save_gigachat_pro_btn.click(fn=lambda x: save_to_json(x, "GigaChat-Pro"), inputs=personalized_output_text_gigachat_pro, outputs=None)
+        save_gigachat_lite_btn.click(fn=lambda x: save_to_json(x, "GigaChat-Lite"), inputs=personalized_output_text_gigachat_lite, outputs=None)
+        save_gigachat_plus_btn.click(fn=lambda x: save_to_json(x, "GigaChat-Lite+"), inputs=personalized_output_text_gigachat_plus, outputs=None)
+
 
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():