Update app.py

app.py

@@ -118,14 +118,44 @@ def download_current_message_database():
         df.drop_duplicates(inplace=True)
         # Сортируем по 'Дата сохранения' по убыванию
         df.sort_values(by='Дата сохранения', ascending=False, inplace=True)
-        # Сохраняем Excel-файл во временное местоположение
-        temp_dir = tempfile.gettempdir()
-        output_file_path = os.path.join(temp_dir, 'messages.xlsx')
-        df.to_excel(output_file_path, index=False)
-        # Возвращаем путь к файлу
-        return output_file_path
+        # Сохраняем Excel-файл в памяти
+        output = io.BytesIO()
+        df.to_excel(output, index=False)
+        output.seek(0)  # Сбрасываем указатель
+
+        # Кодируем содержимое файла в base64 для загрузки на GitHub
+        content = base64.b64encode(output.read()).decode('utf-8')
+
+        # Параметры для загрузки файла на GitHub
+        repo = "fruitpicker01/Storage_dev"
+        path = "messages.xlsx"
+        url = f"https://api.github.com/repos/{repo}/contents/{path}"
+
+        # Проверяем, существует ли файл уже
+        get_response = requests.get(url, headers=headers)
+        if get_response.status_code == 200:
+            sha = get_response.json()['sha']
+        else:
+            sha = None
+
+        data = {
+            "message": f"Обновлен файл messages.xlsx",
+            "content": content,
+        }
+        if sha:
+            data["sha"] = sha  # Необходимо для обновления существующего файла
+
+        # Загружаем (или обновляем) файл на GitHub
+        put_response = requests.put(url, headers=headers, data=json.dumps(data))
+        if put_response.status_code in [200, 201]:
+            # Получаем ссылку для скачивания файла
+            download_url = f"https://raw.githubusercontent.com/{repo}/main/{path}"
+            return download_url  # Возвращаем URL файла
+        else:
+            print(f"Ошибка при загрузке файла на GitHub: {put_response.status_code}, {put_response.text}")
+            return "Ошибка при загрузке файла на GitHub."
     else:
-        return None
+        return "Нет данных для формирования файла."
 
 
 def correct_dash_usage(text):
@@ -1852,7 +1882,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         download_btn = gr.Button("Выгрузить актуальную базу сообщений")
-        download_file = gr.File(label="Скачать базу сообщений")
+        download_link = gr.Markdown(value="", label="Ссылка для скачивания")
 
     # Очистка всех полей кроме prompt_display
     description_input.change(
@@ -2515,10 +2545,18 @@ with gr.Blocks() as demo:
         outputs=[save_meta_llama_405b_btn]
     )
 
+    # Свяжите кнопку с функцией
+    def update_download_link():
+        link = download_current_message_database()
+        if link.startswith("http"):
+            return f"[Скачать базу сообщений]({link})"
+        else:
+            return link  # Возвращаем сообщение об ошибке
+    
     download_btn.click(
-        fn=download_current_message_database,
+        fn=update_download_link,
         inputs=[],
-        outputs=download_file
+        outputs=download_link
     )
 
     demo.launch()