Update app.py

app.py

@@ -266,25 +266,18 @@ 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_json(message, model_name):
-    file_path = "results.json"
-    
-    # Проверяем, существует ли файл
-    if os.path.exists(file_path):
-        with open(file_path, "r") as file:
-            data = json.load(file)
-    else:
-        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)
+# Функция для сохранения сообщения в файл
+def save_to_individual_json(message, model_name):
+    # Формируем уникальное имя файла на основе текущего времени
+    file_name = f"saved_sms_{int(time.time())}.json"
+    file_path = os.path.join(".", file_name)
 
+    # Данные для записи
+    data = {"Модель": model_name, "Сообщение": message}
+
+    # Сохранение данных в новый JSON файл
+    with open(file_path, 'w') as file:
+        json.dump(data, file, ensure_ascii=False, indent=4)
 
 
 # Создание интерфейса Gradio
@@ -361,11 +354,10 @@ with gr.Blocks() as demo:
         save_gigachat_plus_btn = gr.Button("👍")
 
         # Привязка кнопок к функциям сохранения
-        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)
-
+        save_gpt4o_btn.click(fn=lambda x: save_to_individual_json(x, "GPT-4o"), inputs=personalized_output_text_gpt4o, outputs=None)
+        save_gigachat_pro_btn.click(fn=lambda x: save_to_individual_json(x, "GigaChat-Pro"), inputs=personalized_output_text_gigachat_pro, outputs=None)
+        save_gigachat_lite_btn.click(fn=lambda x: save_to_individual_json(x, "GigaChat-Lite"), inputs=personalized_output_text_gigachat_lite, outputs=None)
+        save_gigachat_plus_btn.click(fn=lambda x: save_to_individual_json(x, "GigaChat-Lite+"), inputs=personalized_output_text_gigachat_plus, outputs=None)
 
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():