Upload 2 files

main.py

@@ -28,6 +28,7 @@ YOOMONEY_WALLET = "4100118637309832"
 GROUP_ID = 171587553
 ADMIN_ID = 139711548
 BUCKET_NAME = "photos"
+REFERRAL_BONUS = 2  # Бонус за первую покупку друга
 
 # Инициализация бота
 bot = Bot(token=TOKEN)
@@ -72,12 +73,18 @@ async def download_file(url):
     return None
 
 
-def get_user(user_id):
+def get_user(user_id, ref_id=None):
     res = supabase.table("users").select("*").eq("id", user_id).execute()
     if not res.data:
+        # Логика определения пригласителя
+        inviter = None
+        if ref_id and str(ref_id).isdigit() and int(ref_id) != user_id:
+            inviter = int(ref_id)
+
         new_user = {
             "id": user_id, "balance": 0, "free_used": False, "status": "main",
-            "gender": None, "primary_photo": None, "last_style": None
+            "gender": None, "primary_photo": None, "last_style": None,
+            "invited_by": inviter, "ref_bonus_done": False
         }
         supabase.table("users").insert(new_user).execute()
         return new_user
@@ -98,12 +105,12 @@ async def is_subscribed(user_id):
 def get_main_menu(free_used=False):
     kb = Keyboard(one_time=False)
     if not free_used:
-        kb.add(Text("Одна фотография в подарок"), color=KeyboardButtonColor.POSITIVE).row()
+        kb.add(Text("🎁 Одна фотография в подарок"), color=KeyboardButtonColor.POSITIVE).row()
 
-    kb.add(Text("Стили фото"), color=KeyboardButtonColor.PRIMARY).row()
-    kb.add(Text("Творческий режим"), color=KeyboardButtonColor.PRIMARY).row()
-    kb.add(Text("Замена фото"), color=KeyboardButtonColor.PRIMARY).row()
-    kb.add(Text("Кошелек"), color=KeyboardButtonColor.PRIMARY).row()
+    kb.add(Text("🖤 Стили фото"), color=KeyboardButtonColor.PRIMARY).row()
+    kb.add(Text("✨ Творческий режим"), color=KeyboardButtonColor.PRIMARY).row()
+    kb.add(Text("🔄 Замена фото"), color=KeyboardButtonColor.PRIMARY).row()
+    kb.add(Text("💰 Кошелек"), color=KeyboardButtonColor.PRIMARY).row()
     kb.add(Text("🆘 Поддержка"), color=KeyboardButtonColor.PRIMARY)
     return kb.get_json()
 
@@ -121,6 +128,8 @@ def get_fast_packs_menu():
     packs = supabase.table("settings").select("*").eq("is_premium", False).order("price").execute().data
     for pack in packs:
         kb.add(Text(f"📸 {pack['name']} - {int(pack['price'])} р"), color=KeyboardButtonColor.PRIMARY).row()
+    # Добавляем кнопку бесплатных генераций в самый низ списка пакетов
+    kb.add(Text("🎁 Бесплатные генерации"), color=KeyboardButtonColor.PRIMARY).row()
     kb.add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY)
     return kb.get_json()
 
@@ -133,33 +142,27 @@ def get_gender_menu():
 
 
 def get_styles_menu(gender, page=1):
-    # Используем inline=True. Лимит 6 строк и 10 кнопок.
     kb = Keyboard(inline=True)
     g = gender if gender else "female"
 
     if g in prompts.FACE_SWAP_STYLES:
-        all_styles = [s for s in prompts.FACE_SWAP_STYLES[g].keys() if s != "Одна фотография в подарок"]
-
-        # Пагинация: по 6 стилей на страницу
+        all_styles = [s for s in prompts.FACE_SWAP_STYLES[g].keys() if s != "🎁 Одна фотография в подарок"]
         per_page = 6
         start = (page - 1) * per_page
         end = start + per_page
         current_styles = all_styles[start:end]
 
-        # Размещаем по 2 кнопки в ряд
         for i in range(0, len(current_styles), 2):
             kb.add(Text(current_styles[i]), color=KeyboardButtonColor.PRIMARY)
             if i + 1 < len(current_styles):
                 kb.add(Text(current_styles[i + 1]), color=KeyboardButtonColor.PRIMARY)
             kb.row()
 
-        # Кнопки навигации между страницами
         if page == 1 and len(all_styles) > per_page:
             kb.add(Text("Ещё стили ➡️"), color=KeyboardButtonColor.PRIMARY).row()
         elif page > 1:
             kb.add(Text("⬅️ Назад к стилям"), color=KeyboardButtonColor.PRIMARY).row()
 
-    # Кнопка возврата
     kb.add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY)
     return kb.get_json()
 
@@ -168,7 +171,9 @@ def get_styles_menu(gender, page=1):
 
 @bot.on.message(text=["Начать", "начать", "Меню", "start", "⬅️ В главное меню"])
 async def start_handler(message: Message):
-    user = get_user(message.from_id)
+    # Ловим ref_id из сообщения (параметр в ссылке vk.me/bot?ref=ID)
+    ref_id = message.ref
+    user = get_user(message.from_id, ref_id)
     supabase.table("users").update({"status": "main"}).eq("id", message.from_id).execute()
 
     if not user['free_used']:
@@ -188,12 +193,24 @@ async def support_handler(message: Message):
     await message.answer("Опишите вашу ситуацию прямо здесь, в чате. Я передам сообщение специалисту.")
 
 
-@bot.on.message(text="Замена фото")
+@bot.on.message(text="🔄 Замена фото")
 async def photo_change_main_handler(message: Message):
     await message.answer("Что вы хотите изменить?", keyboard=get_photo_change_menu())
 
 
-@bot.on.message(text=["Одна фотография в подарок", "✅ Я подписался"])
+@bot.on.message(text="🎁 Бесплатные генерации")
+async def free_gens_handler(message: Message):
+    link = f"https://vk.me/club{GROUP_ID}?ref={message.from_id}"
+    msg = (f"🎁 Получай бесплатные генерации за друзей!\n\n"
+           f"Твоя личная ссылка: {link}\n\n"
+           f"Как это работает?\n"
+           f"1. Отправь эту ссылку другу\n"
+           f"2. Когда он совершит свою первую оплату, ты мгновенно получишь +{REFERRAL_BONUS} генерации на свой баланс!\n\n"
+           f"Количество друзей не ограничено. Больше друзей — больше фото! ✨")
+    await message.answer(msg)
+
+
+@bot.on.message(text=["🎁 Одна фотография в подарок", "✅ Я подписался"])
 async def free_trial_handler(message: Message):
     user = get_user(message.from_id)
 
@@ -232,7 +249,7 @@ async def free_trial_handler(message: Message):
     await message.answer("⏳ Генерирую фотографию... Подожди около 30 секунд")
 
     gender = user['gender']
-    prompt = prompts.FACE_SWAP_STYLES[gender]["Одна фотография в подарок"]
+    prompt = prompts.FACE_SWAP_STYLES[gender]["🎁 Одна фотография в подарок"]
 
     paths_to_clean = []
     bucket_path, temporary_url = await upload_to_bucket(user['primary_photo'], message.from_id, "src")
@@ -257,7 +274,7 @@ async def free_trial_handler(message: Message):
 
                 promo_text = (
                     "Теперь ты можешь выбрать один из 10 готовых стилей для фото ✨\n"
-                    "или создать собственный образ в режиме Творца.\n\n"
+                    "или создать собственный образ в Творческом режиме.\n\n"
                     "Выбирай, с чего начнём 👇"
                 )
                 await message.answer(promo_text, keyboard=get_main_menu(True))
@@ -274,7 +291,7 @@ async def free_trial_handler(message: Message):
         supabase.storage.from_(BUCKET_NAME).remove(paths_to_clean)
 
 
-@bot.on.message(text=["Кошелек"])
+@bot.on.message(text=["💰 Кошелек"])
 async def balance_handler(message: Message):
     user = get_user(message.from_id)
     photo_st = "Есть" if user['primary_photo'] else "Нет"
@@ -287,6 +304,8 @@ async def balance_handler(message: Message):
 
     kb = Keyboard()
     kb.add(Text("🛒 Купить генерации"), color=KeyboardButtonColor.POSITIVE).row()
+    # Кнопка бесплатные генерации в кошелек
+    kb.add(Text("🎁 Бесплатные генерации"), color=KeyboardButtonColor.PRIMARY).row()
     kb.add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY)
 
     await message.answer(
@@ -302,7 +321,7 @@ async def shop_handler(message: Message):
     await message.answer("Выберите количество генераций:", keyboard=get_fast_packs_menu())
 
 
-@bot.on.message(text="Стили фото")
+@bot.on.message(text="🖤 Стили фото")
 async def styles_handler(message: Message):
     user = get_user(message.from_id)
     if not user['gender']: return await message.answer("Сначала выберите пол:", keyboard=get_gender_menu())
@@ -327,7 +346,7 @@ async def styles_page_1_back_handler(message: Message):
     await message.answer("Первая страница стилей 👇", keyboard=get_styles_menu(user['gender'], page=1))
 
 
-@bot.on.message(text="Творческий режим")
+@bot.on.message(text="✨ Творческий режим")
 async def creator_mode(message: Message):
     user = get_user(message.from_id)
     if user['balance'] <= 0:
@@ -423,7 +442,7 @@ async def main_handler(message: Message):
 
     if text in styles_dict or user['status'] == "wait_custom_prompt":
         if user['balance'] <= 0 and user['status'] != "wait_custom_prompt":
-            if not (text in styles_dict and not user['free_used'] and text == "Одна фотография в подарок"):
+            if not (text in styles_dict and not user['free_used'] and text == "🎁 Одна фотография в подарок"):
                 return await message.answer("В твоем кошельке 0 генераций. Пополни баланс:",
                                             keyboard=get_fast_packs_menu())
 
@@ -492,7 +511,28 @@ def yoomoney_webhook():
             packs = supabase.table("settings").select("*").execute().data
             for pack in packs:
                 if abs(received_amount - float(pack['price'])) / float(pack['price']) < 0.05:
+                    # 1. Начисляем баланс самому пользователю
                     supabase.rpc("increment_balance", {"user_id": u_id, "amount": pack['amount']}).execute()
+
+                    # 2. Логика реферального бонуса
+                    u_data_res = supabase.table("users").select("invited_by", "ref_bonus_done").eq("id", u_id).execute()
+                    if u_data_res.data:
+                        u_data = u_data_res.data[0]
+                        if u_data['invited_by'] and not u_data['ref_bonus_done']:
+                            inviter_id = u_data['invited_by']
+                            # Начисляем бонус тому, кто пригласил
+                            supabase.rpc("increment_balance",
+                                         {"user_id": inviter_id, "amount": REFERRAL_BONUS}).execute()
+                            # Помечаем, что бонус за этого реферала выдан
+                            supabase.table("users").update({"ref_bonus_done": True}).eq("id", u_id).execute()
+
+                            if loop:
+                                asyncio.run_coroutine_threadsafe(
+                                    bot.api.messages.send(peer_id=inviter_id, random_id=0,
+                                                          message=f"🎁 Твой друг совершил первую оплату! Тебе начислено +{REFERRAL_BONUS} бонусных генераций."),
+                                    loop
+                                )
+
                     if loop:
                         asyncio.run_coroutine_threadsafe(bot.api.messages.send(peer_id=u_id, random_id=0,
                                                                                message=f"✅ Спасибо! Оплата прошла успешно! Теперь выбирай стиль фото в меню"),

prompts.py

@@ -3,7 +3,7 @@
 # Единая база стилей для всех режимов
 FACE_SWAP_STYLES = {
     "female": {
-        "Одна фотография в подарок": (
+        "🎁 Одна фотография в подарок": (
             ""
         ),
         "Портрет": [
@@ -89,7 +89,7 @@ FACE_SWAP_STYLES = {
         ]
     },
     "male": {
-        "Одна фотография в подарок": (
+        "🎁 Одна фотография в подарок": (
             ""
         ),
         "Деловой": [