Update app.py

app.py

@@ -7,10 +7,13 @@ import time
 from langchain.schema import SystemMessage
 from langchain_community.chat_models.gigachat import GigaChat
 from openpyxl import load_workbook
+import base64
+import git
 
 # Установка ключа API для OpenAI и GigaChat
 openai_api_key = os.getenv('GPT_KEY')
 gc_key = os.getenv('GC_KEY')
+token = os.getenv('GITHUB_TOKEN')
 
 # Авторизация в сервисе GigaChat
 chat_pro = GigaChat(credentials=gc_key, model='GigaChat-Pro', verify_ssl_certs=False)
@@ -267,9 +270,10 @@ def check_errors_with_yield(*personalized_messages):
 
 
 # Функция для сохранения сообщения в файл
-def save_to_individual_json(message, model_name):
+def save_to_github(message, model_name):
     # Формируем уникальное имя файла на основе текущего времени
-    file_name = f"saved_sms_{int(time.time())}.json"
+    timestamp = int(time.time())
+    file_name = f"file_{timestamp}.json"
     file_path = os.path.join(".", file_name)
 
     # Данные для записи
@@ -279,6 +283,25 @@ def save_to_individual_json(message, model_name):
     with open(file_path, 'w') as file:
         json.dump(data, file, ensure_ascii=False, indent=4)
 
+    # Инициализация репозитория и добавление файла
+    repo = git.Repo(".")
+    repo.index.add([file_path])
+    repo.index.commit(f"Add {file_name} generated by {model_name}")
+
+    # Настройка удаленного репозитория с использованием токена
+    token = os.getenv('GITHUB_TOKEN')
+    repo_url = f"https://{token}@github.com/fruitpicker01/Storage_1.git"
+    
+    # Если origin еще не настроен, настраиваем его
+    if 'origin' not in [remote.name for remote in repo.remotes]:
+        repo.create_remote('origin', repo_url)
+    else:
+        repo.remotes.origin.set_url(repo_url)
+
+    # Пушим изменения в main ветку
+    repo.remotes.origin.push(f"HEAD:main")
+
+
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -354,10 +377,11 @@ with gr.Blocks() as demo:
         save_gigachat_plus_btn = gr.Button("👍")
 
         # Привязка кнопок к функциям сохранения
-        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)
+        save_gpt4o_btn.click(fn=lambda x: save_to_github(x, "GPT-4o"), inputs=personalized_output_text_gpt4o, outputs=None)
+        save_gigachat_pro_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Pro"), inputs=personalized_output_text_gigachat_pro, outputs=None)
+        save_gigachat_lite_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite"), inputs=personalized_output_text_gigachat_lite, outputs=None)
+        save_gigachat_plus_btn.click(fn=lambda x: save_to_github(x, "GigaChat-Lite+"), inputs=personalized_output_text_gigachat_plus, outputs=None)
+
 
     # Использование сохраненных переменных в следующем блоке
     with gr.Row():