Update app.py

app.py

@@ -6,7 +6,7 @@ from datetime import datetime
 from aiogram import Bot, Dispatcher, types, F
 from aiogram.filters import Command
 from aiogram.utils.keyboard import ReplyKeyboardBuilder, InlineKeyboardBuilder
-from flask import Flask, request, jsonify, render_template_string
+from flask import Flask, request, jsonify, render_template_string, redirect
 import logging
 import threading
 from huggingface_hub import HfApi, hf_hub_download
@@ -23,7 +23,7 @@ bot = Bot(token=BOT_TOKEN)
 dp = Dispatcher()
 app = Flask(__name__)
 
-# Путь для хранения данных (товары и заказы)
+# Путь для хранения данных (товары, заказы и категории)
 DATA_FILE = 'data.json'
 
 # Настройки Hugging Face
@@ -37,23 +37,25 @@ def load_data():
         download_db_from_hf()
         with open(DATA_FILE, 'r', encoding='utf-8') as f:
             loaded_data = json.load(f)
-        # Проверка структуры JSON: ожидается словарь с ключами 'products' и 'orders'
+        # Проверка структуры JSON: ожидается словарь с ключами 'products', 'orders' и 'categories'
         if not (isinstance(loaded_data, dict) and 'products' in loaded_data and 'orders' in loaded_data):
             logger.error("Неверная структура JSON файла, ожидается словарь с ключами 'products' и 'orders'. Используем дефолтное значение.")
-            return {'products': [], 'orders': []}
+            loaded_data = {'products': [], 'orders': []}
+        if "categories" not in loaded_data:
+            loaded_data["categories"] = []
         return loaded_data
     except FileNotFoundError:
         logger.warning("Локальный файл базы данных не найден после скачивания.")
-        return {'products': [], 'orders': []}
+        return {'products': [], 'orders': [], 'categories': []}
     except json.JSONDecodeError:
         logger.error("Ошибка при загрузке данных: Невозможно декодировать JSON файл.")
-        return {'products': [], 'orders': []}
+        return {'products': [], 'orders': [], 'categories': []}
     except RepositoryNotFoundError:
         logger.error("Репозиторий не найден. Создание локальной базы данных.")
-        return {'products': [], 'orders': []}
+        return {'products': [], 'orders': [], 'categories': []}
     except Exception as e:
         logger.error(f"Ошибка при загрузке данных: {e}")
-        return {'products': [], 'orders': []}
+        return {'products': [], 'orders': [], 'categories': []}
 
 # Функция для сохранения данных в Hugging Face Hub
 def save_data(data):
@@ -102,8 +104,9 @@ def download_db_from_hf():
 # Загрузка данных
 data = load_data()
 
-# Клавиатуры
+# Функции для формирования клавиатур
 def get_main_keyboard():
+    # Теперь кнопка "Меню" покажет категории товаров
     builder = ReplyKeyboardBuilder()
     builder.button(text="📋 Меню")
     builder.button(text="🛒 Корзина")
@@ -111,6 +114,14 @@ def get_main_keyboard():
     builder.adjust(2)
     return builder.as_markup(resize_keyboard=True)
 
+def get_category_keyboard():
+    builder = InlineKeyboardBuilder()
+    for category in data['categories']:
+        builder.button(text=category['name'], callback_data=f"cat_{category['id']}")
+    # Добавляем кнопку "Назад" для возврата в главное меню, если понадобится
+    builder.adjust(2)
+    return builder.as_markup()
+
 def get_product_keyboard(product_id):
     builder = InlineKeyboardBuilder()
     builder.button(text="Добавить в корзину", callback_data=f"add_{product_id}")
@@ -121,22 +132,42 @@ def get_product_keyboard(product_id):
 async def cmd_start(message: types.Message):
     await message.answer("Привет! Я твой бот-магазин. Выбери действие:", reply_markup=get_main_keyboard())
 
+# При нажатии на "Меню" выводятся категории
 @dp.message(F.text == "📋 Меню")
-async def show_products(message: types.Message):
-    if not data['products']:
-        await message.answer("Нет доступных товаров.")
+async def show_categories(message: types.Message):
+    if not data['categories']:
+        await message.answer("Нет доступных категорий.")
+        return
+    await message.answer("Выберите категорию:", reply_markup=get_category_keyboard())
+
+# Обработчик для выбора категории (callback_data: "cat_{id}")
+@dp.callback_query(F.data.startswith("cat_"))
+async def show_products_in_category(callback_query: types.CallbackQuery):
+    try:
+        cat_id = int(callback_query.data.split('_')[1])
+    except (IndexError, ValueError):
+        await bot.answer_callback_query(callback_query.id, "Неверные данные категории")
+        return
+
+    # Фильтруем товары по выбранной категории
+    products_in_cat = [p for p in data['products'] if p.get('category_id') == cat_id]
+    if not products_in_cat:
+        await bot.send_message(callback_query.from_user.id, "В этой категории нет товаров.")
+        await bot.answer_callback_query(callback_query.id)
         return
-    for product in data['products']:
+
+    for product in products_in_cat:
         photo_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/photos/{product['photo']}" if product.get('photo') 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']))
+                await bot.send_photo(chat_id=callback_query.from_user.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']))
+                await bot.send_message(callback_query.from_user.id, caption, reply_markup=get_product_keyboard(product['id']))
         else:
-            await message.answer(caption, reply_markup=get_product_keyboard(product['id']))
+            await bot.send_message(callback_query.from_user.id, caption, reply_markup=get_product_keyboard(product['id']))
+    await bot.answer_callback_query(callback_query.id)
 
 @dp.message(F.text == "🛒 Корзина")
 async def show_cart(message: types.Message):
@@ -158,7 +189,11 @@ async def show_cart(message: types.Message):
 
 @dp.callback_query(F.data.startswith("add_"))
 async def add_to_cart(callback_query: types.CallbackQuery):
-    product_id = int(callback_query.data.split('_')[1])
+    try:
+        product_id = int(callback_query.data.split('_')[1])
+    except (IndexError, ValueError):
+        await bot.answer_callback_query(callback_query.id, "Неверные данные товара")
+        return
     product = next((p for p in data['products'] if p['id'] == product_id), None)
     if product:
         user_id = callback_query.from_user.id
@@ -175,7 +210,11 @@ async def add_to_cart(callback_query: types.CallbackQuery):
 # Изменённый обработчик "Оформить заказ" – заказ сразу отправляется в WhatsApp и удаляется из корзины
 @dp.callback_query(F.data.startswith("complete_"))
 async def complete_order(callback_query: types.CallbackQuery):
-    user_id = int(callback_query.data.split('_')[1])
+    try:
+        user_id = int(callback_query.data.split('_')[1])
+    except (IndexError, ValueError):
+        await bot.answer_callback_query(callback_query.id, "Неверные данные")
+        return
     cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o.get('completed')), None)
     if cart and cart['items']:
         total = 0
@@ -187,11 +226,11 @@ async def complete_order(callback_query: types.CallbackQuery):
                 total += product['price'] * item['quantity']
         cart_text += f"\nИтого: {total} руб."
         encoded_text = urllib.parse.quote(cart_text)
-        whatsapp_link = f"https://wa.me/996500398754?text={encoded_text}"
+        # Новый номер WhatsApp: +996709513331 (без знака + для ссылки)
+        whatsapp_link = f"https://wa.me/996709513331?text={encoded_text}"
         # Удаляем корзину, чтобы заказ не попадал в личный кабинет
         data['orders'].remove(cart)
         save_data(data)
-        # Отправляем пользователю сообщение с ссылкой на WhatsApp
         await bot.send_message(user_id, f"Пожалуйста, оформите заказ через WhatsApp, перейдя по ссылке:\n{whatsapp_link}")
         await bot.answer_callback_query(callback_query.id)
     else:
@@ -200,7 +239,7 @@ async def complete_order(callback_query: types.CallbackQuery):
 @dp.message(F.text == "📦 Заказы")
 async def show_orders(message: types.Message):
     user_id = message.from_user.id
-    # Поскольку оформленные заказы больше не сохраняются, здесь отображаются только те, которые остались в базе (если вдруг)
+    # Так как оформленные заказы больше не сохраняются, здесь показываются только те, которые остались (если вдруг)
     user_orders = [o for o in data['orders'] if o.get('completed')]
     if not user_orders:
         await message.answer("У вас нет оформленных заказов.")
@@ -215,7 +254,7 @@ async def show_orders(message: types.Message):
         response += f"\nИтого: {total} руб.\nДата: {order['date']}"
         await message.answer(response)
 
-# Админ-панель (Flask) с HTML, CSS и JavaScript в строке
+# Админ-панель (Flask)
 admin_html = """
 <!DOCTYPE html>
 <html>
@@ -224,64 +263,98 @@ admin_html = """
     <style>
         body { font-family: Arial, sans-serif; margin: 20px; background-color: #f0f0f0; }
         .container { max-width: 800px; margin: 0 auto; }
-        .product { border: 1px solid #ccc; padding: 15px; margin: 10px 0; background-color: #fff; border-radius: 5px; }
-        form { background-color: #fff; padding: 15px; border-radius: 5px; margin-bottom: 20px; }
-        input, textarea { width: 100%; margin: 5px 0; padding: 8px; }
+        .section { background-color: #fff; padding: 15px; margin-bottom: 20px; border-radius: 5px; }
+        input, textarea, select { 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; }
+        .item { border: 1px solid #ccc; padding: 10px; margin: 5px 0; }
     </style>
 </head>
 <body>
     <div class="container">
-        <h1>Управление товарами</h1>
-        <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>
-        {% if products %}
-            {% for product in products %}
-            <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 %}
-        {% else %}
-            <p>Нет товаров.</p>
-        {% endif %}
-        <h2>Заказы</h2>
-        {% if orders %}
-            {% for order in orders %}
-            <div class="product">
-                Пользователь: {{ order.user_id }}<br>
-                Дата: {{ order.date }}<br>
-                Товары: 
-                {% for item in order['items'] %}
-                    {% for product in products %}
-                        {% if product.id == item.product_id %}
-                            {{ item.quantity }} x {{ product.name }}<br>
-                        {% endif %}
+        <h1>Админ-панель</h1>
+        <div class="section">
+            <h2>Управление категориями</h2>
+            <form id="addCategoryForm" method="POST" action="/add_category">
+                <input type="text" name="name" placeholder="Название категории" required><br>
+                <button type="submit">Добавить категорию</button>
+            </form>
+            <h3>Существующие категории</h3>
+            {% if categories %}
+                {% for category in categories %}
+                <div class="item">
+                    {{ category.name }} (ID: {{ category.id }})
+                    <button onclick="deleteCategory({{ category.id }})">Удалить</button>
+                </div>
+                {% endfor %}
+            {% else %}
+                <p>Нет категорий.</p>
+            {% endif %}
+        </div>
+        <div class="section">
+            <h2>Управление товарами</h2>
+            <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>
+                <label>Категория:</label>
+                <select name="category_id" required>
+                    <option value="">Выберите категорию</option>
+                    {% for category in categories %}
+                    <option value="{{ category.id }}">{{ category.name }}</option>
                     {% endfor %}
+                </select><br>
+                <input type="file" name="photo" accept="image/*"><br>
+                <button type="submit">Добавить товар</button>
+            </form>
+            <h3>Существующие товары</h3>
+            {% if products %}
+                {% for product in products %}
+                <div class="item">
+                    {{ 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 %}
-            </div>
-            {% endfor %}
-        {% else %}
-            <p>Нет заказов.</p>
-        {% endif %}
+            {% else %}
+                <p>Нет товаров.</p>
+            {% endif %}
+        </div>
+        <div class="section">
+            <h2>Заказы</h2>
+            {% if orders %}
+                {% for order in orders %}
+                <div class="item">
+                    Пользователь: {{ order.user_id }}<br>
+                    Дата: {{ order.date }}<br>
+                    Товары: 
+                    {% for item in order['items'] %}
+                        {% for product in products %}
+                            {% if product.id == item.product_id %}
+                                {{ item.quantity }} x {{ product.name }}<br>
+                            {% endif %}
+                        {% endfor %}
+                    {% endfor %}
+                </div>
+                {% endfor %}
+            {% else %}
+                <p>Нет заказов.</p>
+            {% endif %}
+        </div>
     </div>
     <script>
         async function deleteProduct(productId) {
             const response = await fetch(`/delete_product/${productId}`, { method: 'POST' });
             if (response.ok) window.location.reload();
         }
+        async function deleteCategory(categoryId) {
+            const response = await fetch(`/delete_category/${categoryId}`, { method: 'POST' });
+            if (response.ok) window.location.reload();
+        }
     </script>
 </body>
 </html>
@@ -290,8 +363,8 @@ admin_html = """
 @app.route('/')
 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'], repo_id=REPO_ID)
+        logger.info("Rendering admin panel with products, orders and categories")
+        return render_template_string(admin_html, products=data['products'], orders=data['orders'], categories=data['categories'], repo_id=REPO_ID)
     except Exception as e:
         logger.error(f"Ошибка в шаблоне: {e}")
         return "Ошибка сервера. Проверь логи.", 500
@@ -304,13 +377,14 @@ def add_product():
             name = request.form['name']
             price = float(request.form['price'])
             description = request.form['description']
-            photo = request.files['photo']  # Получаем файл фотографии
+            category_id = int(request.form['category_id'])
+            photo = request.files.get('photo')
             product_id = max((p['id'] for p in data['products']), default=0) + 1
 
             photo_filename = None
-            if photo:
+            if photo and photo.filename:
                 photo_filename = secure_filename(photo.filename)
-                temp_path = os.path.join(".", photo_filename)  # Сохраняем временно локально
+                temp_path = os.path.join(".", photo_filename)
                 photo.save(temp_path)
 
                 try:
@@ -328,17 +402,18 @@ def add_product():
                     logger.error(f"Ошибка при загрузке фото: {e}")
                     return jsonify({'status': 'error', 'message': f'Ошибка при загрузке фото: {str(e)}'}), 500
                 finally:
-                    os.remove(temp_path)  # Удаляем временный файл
+                    os.remove(temp_path)
 
             data['products'].append({
                 'id': product_id,
                 'name': name,
                 'price': price,
                 'description': description,
+                'category_id': category_id,
                 'photo': photo_filename
             })
             save_data(data)
-            return jsonify({'status': 'success'})
+            return redirect("/")
     except Exception as e:
         logger.error(f"Ошибка при добавлении товара: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500
@@ -354,6 +429,29 @@ def delete_product(product_id):
         logger.error(f"Ошибка при удалении товара: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500
 
+@app.route('/add_category', methods=['POST'])
+def add_category():
+    try:
+        name = request.form['name']
+        category_id = max((c['id'] for c in data['categories']), default=0) + 1
+        data['categories'].append({'id': category_id, 'name': name})
+        save_data(data)
+        return redirect("/")
+    except Exception as e:
+        logger.error(f"Ошибка при добавлении категории: {e}")
+        return jsonify({'status': 'error', 'message': str(e)}), 500
+
+@app.route('/delete_category/<int:category_id>', methods=['POST'])
+def delete_category(category_id):
+    try:
+        logger.info(f"Deleting category with id={category_id}")
+        data['categories'] = [c for c in data['categories'] if c['id'] != category_id]
+        save_data(data)
+        return jsonify({'status': 'success'})
+    except Exception as e:
+        logger.error(f"Ошибка при удалении категории: {e}")
+        return jsonify({'status': 'error', 'message': str(e)}), 500
+
 # Запуск бота и Flask
 async def on_startup(_):
     logger.info("Бот запущен!")
@@ -377,4 +475,4 @@ if __name__ == '__main__':
     except KeyboardInterrupt:
         logger.info("Stopping bot and Flask")
     finally:
-        flask_thread.join()  # Ждём завершения потока Flask при завершении программы
+        flask_thread.join()