Update app.py

app.py

@@ -8,13 +8,16 @@ from aiogram.utils.keyboard import ReplyKeyboardBuilder, InlineKeyboardBuilder
 from flask import Flask, request, jsonify, render_template_string
 import logging
 import threading
+from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub.utils import RepositoryNotFoundError
+from werkzeug.utils import secure_filename
 
 # Настройка логирования
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Инициализация бота и Flask
-BOT_TOKEN = '7734802681:AAGKHGG8O9uNk64JWTHH5yqXzvSxCcoLUdA'
+BOT_TOKEN = '7734802681:AAGKHGG8O9uNk64JWTHH5yqXzvSxCcoLUdA'  # Замените на токен вашего бота
 bot = Bot(token=BOT_TOKEN)
 dp = Dispatcher()
 app = Flask(__name__)
@@ -22,24 +25,75 @@ app = Flask(__name__)
 # Путь для хранения данных (товары и заказы)
 DATA_FILE = 'data.json'
 
-# Загрузка или создание данных
+# Настройки Hugging Face
+REPO_ID = "flpolprojects/Clients"  # Замените на ваш репозиторий
+HF_TOKEN_WRITE = os.getenv("HF_TOKEN")  # Установите переменную окружения HF_TOKEN
+HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")  # Установите переменную окружения HF_TOKEN_READ
+
+# Функция для загрузки данных из Hugging Face Hub
 def load_data():
     try:
-        if os.path.exists(DATA_FILE):
-            with open(DATA_FILE, 'r', encoding='utf-8') as f:
-                return json.load(f)
+        download_db_from_hf()
+        with open(DATA_FILE, 'r', encoding='utf-8') as f:
+            return json.load(f)
+    except FileNotFoundError:
+        logger.warning("Локальный файл базы данных не найден после скачивания.")
+        return {'products': [], 'orders': []}
+    except json.JSONDecodeError:
+        logger.error("Ошибка при загрузке данных: Невозможно декодировать JSON файл.")
+        return {'products': [], 'orders': []}
+    except RepositoryNotFoundError:
+        logger.error("Репозиторий не найден. Создание локальной базы данных.")
         return {'products': [], 'orders': []}
     except Exception as e:
         logger.error(f"Ошибка при загрузке данных: {e}")
         return {'products': [], 'orders': []}
 
+# Функция для сохранения данных в Hugging Face Hub
 def save_data(data):
     try:
         with open(DATA_FILE, 'w', encoding='utf-8') as f:
             json.dump(data, f, ensure_ascii=False, indent=4)
+        upload_db_to_hf()
     except Exception as e:
         logger.error(f"Ошибка при сохранении данных: {e}")
 
+# Функция для загрузки базы данных из Hugging Face Hub
+def upload_db_to_hf():
+    try:
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=DATA_FILE,
+            path_in_repo=DATA_FILE,
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            token=HF_TOKEN_WRITE,
+            commit_message=f"Автоматическое резервное копирование базы данных {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+        )
+        logger.info("Резервная копия JSON базы успешно загружена на Hugging Face.")
+    except Exception as e:
+        logger.error(f"Ошибка при загрузке резервной копии: {e}")
+
+# Функция для скачивания базы данных из Hugging Face Hub
+def download_db_from_hf():
+    try:
+        hf_hub_download(
+            repo_id=REPO_ID,
+            filename=DATA_FILE,
+            repo_type="dataset",
+            token=HF_TOKEN_READ,
+            local_dir=".",
+            local_dir_use_symlinks=False
+        )
+        logger.info("JSON база успешно скачана из Hugging Face.")
+    except RepositoryNotFoundError as e:
+        logger.error(f"Репозиторий не найден: {e}")
+        raise
+    except Exception as e:
+        logger.error(f"Ошибка при скачивании JSON базы: {e}")
+        raise
+
+# Загрузка данных
 data = load_data()
 
 # Клавиатуры
@@ -67,8 +121,17 @@ async def show_products(message: types.Message):
         await message.answer("Нет доступных товаров.")
         return
     for product in data['products']:
-        await message.answer(f"🏷 {product['name']} - {product['price']} руб.\nОписание: {product['description']}\n/id: {product['id']}",
-                            reply_markup=get_product_keyboard(product['id']))
+        photo_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/photos/{product['photo']}" if 'photo' in product else None
+        caption = f"🏷 {product['name']} - {product['price']} руб.\nОписание: {product['description']}\n/id: {product['id']}"
+
+        if photo_url:
+            try:
+                await bot.send_photo(chat_id=message.chat.id, photo=photo_url, caption=caption, reply_markup=get_product_keyboard(product['id']))
+            except Exception as e:
+                logger.error(f"Ошибка при отправке фото: {e}")
+                await message.answer(caption, reply_markup=get_product_keyboard(product['id']))
+        else:
+            await message.answer(caption, reply_markup=get_product_keyboard(product['id']))
 
 @dp.message(F.text == "🛒 Корзина")
 async def show_cart(message: types.Message):
@@ -147,15 +210,17 @@ admin_html = """
         input, textarea { width: 100%; margin: 5px 0; padding: 8px; }
         button { background-color: #4CAF50; color: white; padding: 10px 15px; border: none; border-radius: 5px; cursor: pointer; }
         button:hover { background-color: #45a049; }
+        img { max-width: 100px; max-height: 100px; }
     </style>
 </head>
 <body>
     <div class="container">
         <h1>Управление товарами</h1>
-        <form id="addProductForm">
+        <form id="addProductForm" method="POST" enctype="multipart/form-data" action="/add_product">
             <input type="text" name="name" placeholder="Название" required><br>
             <input type="number" name="price" placeholder="Цена" step="0.01" required><br>
             <textarea name="description" placeholder="Описание" required></textarea><br>
+            <input type="file" name="photo" accept="image/*"><br>
             <button type="submit">Добавить товар</button>
         </form>
         <h2>Существующие товары</h2>
@@ -164,6 +229,9 @@ admin_html = """
             <div class="product">
                 {{ product.name }} - {{ product.price }} руб.<br>
                 {{ product.description }}<br>
+                {% if product.photo %}
+                    <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/photos/{{ product.photo }}" alt="{{ product.name }}">
+                {% endif %}
                 <button onclick="deleteProduct({{ product.id }})">Удалить</button>
             </div>
             {% endfor %}
@@ -191,12 +259,6 @@ admin_html = """
         {% endif %}
     </div>
     <script>
-        document.getElementById('addProductForm').onsubmit = async (e) => {
-            e.preventDefault();
-            const formData = new FormData(e.target);
-            const response = await fetch('/add_product', { method: 'POST', body: formData });
-            if (response.ok) window.location.reload();
-        };
         async function deleteProduct(productId) {
             const response = await fetch(`/delete_product/${productId}`, { method: 'POST' });
             if (response.ok) window.location.reload();
@@ -210,7 +272,7 @@ admin_html = """
 def admin_panel():
     try:
         logger.info("Rendering admin panel with products and orders")
-        return render_template_string(admin_html, products=data['products'], orders=data['orders'])
+        return render_template_string(admin_html, products=data['products'], orders=data['orders'], repo_id=REPO_ID)
     except Exception as e:
         logger.error(f"Ошибка в шаблоне: {e}")
         return "Ошибка сервера. Проверь логи.", 500
@@ -223,8 +285,33 @@ def add_product():
             name = request.form['name']
             price = float(request.form['price'])
             description = request.form['description']
+            photo = request.files['photo']  # Получаем файл фотографии
             product_id = max((p['id'] for p in data['products']), default=0) + 1
-            data['products'].append({'id': product_id, 'name': name, 'price': price, 'description': description})
+
+            photo_filename = None
+            if photo:
+                photo_filename = secure_filename(photo.filename)
+                temp_path = os.path.join(".", photo_filename)  # Сохраняем временно локально
+                photo.save(temp_path)
+
+                try:
+                    api = HfApi()
+                    api.upload_file(
+                        path_or_fileobj=temp_path,
+                        path_in_repo=f"photos/{photo_filename}",
+                        repo_id=REPO_ID,
+                        repo_type="dataset",
+                        token=HF_TOKEN_WRITE,
+                        commit_message=f"Добавлено фото для товара {name}"
+                    )
+                    logger.info(f"Фото успешно загружено: {photo_filename}")
+                except Exception as e:
+                    logger.error(f"Ошибка при загрузке фото: {e}")
+                    return jsonify({'status': 'error', 'message': f'Ошибка при загрузке фото: {str(e)}'}), 500
+                finally:
+                    os.remove(temp_path)  # Удаляем временный файл
+
+            data['products'].append({'id': product_id, 'name': name, 'price': price, 'description': description, 'photo': photo_filename})  # Сохраняем имя файла фотографии
             save_data(data)
             return jsonify({'status': 'success'})
     except Exception as e:
@@ -249,7 +336,7 @@ async def on_startup(_):
 def run_flask():
     try:
         logger.info("Starting Flask server on port 7860")
-        app.run(host='0.0.0.0', port=7860)
+        app.run(host='0.0.0.0', port=7860, debug=True)
     except Exception as e:
         logger.error(f"Ошибка в Flask: {e}")
 
@@ -265,4 +352,4 @@ if __name__ == '__main__':
     except KeyboardInterrupt:
         logger.info("Stopping bot and Flask")
     finally:
-        flask_thread.join()  # Ждём завершения потока Flask при завершении программы
+        flask_thread.join()  # Ждём завершения потока Flask при завершении программы