Update app.py

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import time
 from langchain.schema import SystemMessage
 from langchain_community.chat_models.gigachat import GigaChat
+from openpyxl import load_workbook
 
 # Установка ключа API для OpenAI и GigaChat
 openai_api_key = os.getenv('GPT_KEY')
@@ -270,17 +271,21 @@ def save_to_excel(message, model_name):
     
     # Проверяем, существует ли файл
     if os.path.exists(file_path):
-        df = pd.read_excel(file_path)
+        workbook = load_workbook(file_path)
+        sheet = workbook.active
     else:
-        # Если файл не существует, создаем новый DataFrame
-        df = pd.DataFrame(columns=["Модель", "Сообщение"])
+        # Если файл не существует, создаем новый Workbook и лист
+        workbook = load_workbook()
+        sheet = workbook.active
+        # Добавляем заголовки колонок
+        sheet.append(["Модель", "Сообщение"])
 
-    # Добавляем новую запись
-    new_row = {"Модель": model_name, "Сообщение": message}
-    df = df.append(new_row, ignore_index=True)
+    # Находим первую пустую строку
+    new_row = [model_name, message]
+    sheet.append(new_row)
 
-    # Сохраняем DataFrame в Excel
-    df.to_excel(file_path, index=False)
+    # Сохраняем изменения
+    workbook.save(file_path)
 
 
 # Создание интерфейса Gradio
@@ -357,10 +362,10 @@ 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)
-        save_gigachat_lite_btn.click(fn=save_to_excel, inputs=personalized_output_text_gigachat_lite)
-        save_gigachat_plus_btn.click(fn=save_to_excel, inputs=personalized_output_text_gigachat_plus)
+        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)
 
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():