Update app.py

app.py

@@ -49,6 +49,74 @@ for sheet_name, df in data.items():
         print(f"Ошибка при обработке данных листа {sheet_name}: {e}")
         features[sheet_name] = {}
 
+
+def save_user_request_to_github(description, advantages, personalization_params):
+    # Собираем все данные в один словарь
+    data_to_save = {
+        "description": description,
+        "advantages": advantages,
+        "personalization_params": personalization_params,
+        "timestamp": time.time()
+    }
+
+    # Преобразуем контент в JSON-строку и кодируем в base64
+    file_content_encoded = base64.b64encode(json.dumps(data_to_save).encode()).decode()
+
+    # Параметры для GitHub API
+    repo = "fruitpicker01/Storage_dev"
+    path = f"user_request_{int(time.time())}.json"
+    url = f"https://api.github.com/repos/{repo}/contents/{path}"
+    headers = {
+        "Authorization": f"token {token}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "message": f"Добавлен новый файл {path}",
+        "content": file_content_encoded
+    }
+
+    # Отправка POST-запроса на GitHub API для создания файла в репозитории
+    response = requests.put(url, headers=headers, data=json.dumps(data))
+    if response.status_code == 201:
+        print("Данные успешно сохранены на GitHub")
+    else:
+        print(f"Ошибка при сохранении данных на GitHub: {response.status_code} {response.text}")
+
+def load_last_user_request_from_github():
+    repo = "fruitpicker01/Storage_dev"
+    url = f"https://api.github.com/repos/{repo}/contents"
+    headers = {
+        "Authorization": f"token {token}",
+        "Content-Type": "application/json"
+    }
+
+    # Получаем список файлов в репозитории
+    response = requests.get(url, headers=headers)
+    if response.status_code == 200:
+        files = response.json()
+        json_files = [file for file in files if file['name'].startswith("user_request_")]
+
+        if not json_files:
+            print("Нет сохраненных запросов.")
+            return None
+
+        # Находим последний файл (по времени создания)
+        last_file = max(json_files, key=lambda x: x['name'])
+        file_url = last_file['download_url']
+
+        # Загружаем и декодируем содержимое файла
+        file_response = requests.get(file_url)
+        if file_response.status_code == 200:
+            data = json.loads(file_response.text)
+            return data
+        else:
+            print(f"Ошибка при загрузке файла: {file_response.status_code}")
+            return None
+    else:
+        print(f"Ошибка при обращении к GitHub: {response.status_code}")
+        return None
+
+
 # Функция для генерации стандартного промпта
 def generate_standard_prompt(description, advantages, *selected_values):
     prompt = (
@@ -298,6 +366,9 @@ def generate_message_mistral_with_retry(prompt):
 
 
 def generate_messages(description, advantages, *selected_values):
+    
+    save_user_request_to_github(description, advantages, selected_values)
+    
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
@@ -499,6 +570,9 @@ def personalize_messages_with_yield(
     mistral_message,
     *selected_values
 ):
+
+    save_user_request_to_github(gpt4o_message, gigachat_pro_message, selected_values)
+    
     personalization_prompt = generate_personalization_prompt(*selected_values)
     yield personalization_prompt, "", "", "", "", "", "", "", "", "", ""
 
@@ -720,6 +794,9 @@ with gr.Blocks() as demo:
         submit_btn = gr.Button("1. Создать неперсонализированное сообщение")
         personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
 
+    with gr.Row():
+        load_btn = gr.Button("Вернуть параметры последнего запроса")
+
     # Первый ряд
     with gr.Row():
         output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3, interactive=False)
@@ -849,6 +926,12 @@ with gr.Blocks() as demo:
         ]
     )
 
+    # Обработка клика по кнопке восстановления
+    load_btn.click(
+        fn=lambda: load_last_user_request_from_github(),
+        inputs=[],
+        outputs=[description_input, advantages_input, *selections]
+    )
 
     # Привязка кнопок к функциям сохранения
     save_gpt4o_btn.click(