HollyCorn commited on
Commit
0d8ae5d
·
verified ·
1 Parent(s): 8e30815

Upload 5 files

Browse files
Files changed (5) hide show
  1. Dockerfile +20 -0
  2. generator.py +64 -0
  3. main.py +514 -0
  4. prompts.py +0 -0
  5. requirements.txt +7 -0
Dockerfile ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 1. Берем официальный легкий образ Python
2
+ FROM python:3.11-slim
3
+
4
+ # 2. Указываем рабочую директорию внутри сервера
5
+ WORKDIR /app
6
+
7
+ # 3. Копируем файл со списком библиотек в сервер
8
+ COPY requirements.txt .
9
+
10
+ # 4. Устанавливаем эти библиотеки
11
+ RUN pip install --no-cache-dir -r requirements.txt
12
+
13
+ # 5. Копируем весь остальной код (main.py и прочее) в сервер
14
+ COPY . .
15
+
16
+ # 6. Отключаем буферизацию логов (чтобы ты сразу видела ошибки в консоли Hugging Face)
17
+ ENV PYTHONUNBUFFERED=1
18
+
19
+ # 7. Команда для запуска твоего бота
20
+ CMD ["python", "main.py"]
generator.py ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import fal_client
3
+ import asyncio
4
+
5
+ # Ключ берем из переменных окружения
6
+ FAL_KEY = os.environ.get("FAL_KEY")
7
+ fal_client.api_key = FAL_KEY
8
+
9
+
10
+ def clean_url(url):
11
+ """
12
+ Очистка URL от лишних параметров.
13
+ Для ВК-фото оставляем как есть, для остальных (включая Supabase)
14
+ отсекаем query-параметры, если они мешают загрузке в API.
15
+ """
16
+ if not url:
17
+ return url
18
+ if "userapi.com" in url:
19
+ return url
20
+ # Для прямых ссылок из бакета Supabase часто требуется чистый путь
21
+ return url.split('?')[0]
22
+
23
+
24
+ async def generate_image(prompt, gender, photo_url=None, lora_model=None):
25
+ """
26
+ Основная функция генерации (Face Swap через Nano Banana).
27
+ lora_model игнорируется, так как функционал отключен.
28
+ """
29
+ # Используем актуальный эндпоинт Nano Banana 2 для редактирования и Face Swap
30
+ endpoint = "fal-ai/nano-banana/edit"
31
+
32
+ # Очищаем промпт от плейсхолдера, если он попал из старой логики
33
+ final_prompt = prompt.replace("{trigger}", "a person")
34
+
35
+ try:
36
+ # --- ВАРИАНТ Б: Face Swap (Nano Banana) ---
37
+ # Оставляем только этот функционал с текущими настройками
38
+ print(f"🚀 Генерация через Face Swap. Промпт: {final_prompt[:50]}...")
39
+
40
+ # Подготавливаем URL (будь то ссылка из ВК или из вашего бакета)
41
+ user_photo = clean_url(photo_url)
42
+
43
+ arguments = {
44
+ "image_urls": [user_photo],
45
+ "prompt": final_prompt,
46
+ "negative_prompt": "deformed face, blurry, low quality, bad anatomy, extra fingers, distorted features",
47
+ "guidance_scale": 7.5,
48
+ "image_size": "portrait_16_9",
49
+ "num_inference_steps": 30,
50
+ "strength": 0.55 # Оптимальная настройка для сохранения сходства и наложения стиля
51
+ }
52
+
53
+ # Вызов API Nano Banana 2
54
+ result = await fal_client.subscribe_async(endpoint, arguments=arguments)
55
+
56
+ if result and 'images' in result and len(result['images']) > 0:
57
+ return result['images'][0]['url']
58
+ else:
59
+ print("⚠️ API вернуло пустой результат")
60
+ return None
61
+
62
+ except Exception as e:
63
+ print(f"❌ Ошибка в генераторе (Endpoint: {endpoint}): {e}")
64
+ return None
main.py ADDED
@@ -0,0 +1,514 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import threading
3
+ import hashlib
4
+ import asyncio
5
+ import io
6
+ import aiohttp
7
+ import random
8
+ import uuid
9
+ from flask import Flask, request
10
+ from vkbottle import Keyboard, KeyboardButtonColor, Text, OpenLink, PhotoMessageUploader
11
+ from vkbottle.bot import Bot, Message
12
+ from supabase import create_client, Client
13
+ from loguru import logger
14
+ from dotenv import load_dotenv
15
+
16
+ load_dotenv()
17
+
18
+ # Импортируем вспомогательные файлы
19
+ import prompts
20
+ import generator
21
+
22
+ # --- КОНФИГУРАЦИЯ ---
23
+ TOKEN = os.environ.get("VK_TOKEN")
24
+ YOOMONEY_SECRET = os.environ.get("YOOMONEY_SECRET")
25
+ SUPABASE_URL = os.environ.get("SUPABASE_URL")
26
+ SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
27
+ YOOMONEY_WALLET = "4100118637309832"
28
+ GROUP_ID = 171587553
29
+ ADMIN_ID = 139711548
30
+ BUCKET_NAME = "photos"
31
+
32
+ # Инициализация бота
33
+ bot = Bot(token=TOKEN)
34
+ uploader = PhotoMessageUploader(bot.api)
35
+ supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
36
+ app = Flask(__name__)
37
+ loop = None
38
+
39
+ logger.add("bot_log.log", rotation="10 MB", level="INFO")
40
+
41
+
42
+ # --- ВСПОМОГАТЕЛЬНЫЕ ФУНКЦИИ ---
43
+
44
+ async def upload_to_bucket(url, user_id, prefix="temp"):
45
+ try:
46
+ async with aiohttp.ClientSession() as session:
47
+ async with session.get(url) as resp:
48
+ if resp.status == 200:
49
+ img_data = await resp.read()
50
+ file_name = f"{user_id}/{prefix}_{uuid.uuid4()}.png"
51
+ supabase.storage.from_(BUCKET_NAME).upload(
52
+ path=file_name,
53
+ file=img_data,
54
+ file_options={"content-type": "image/png"}
55
+ )
56
+ res = supabase.storage.from_(BUCKET_NAME).get_public_url(file_name)
57
+ return file_name, res
58
+ except Exception as e:
59
+ logger.error(f"Ошибка загрузки в бакет ({prefix}): {e}")
60
+ return None, None
61
+
62
+
63
+ async def download_file(url):
64
+ try:
65
+ async with aiohttp.ClientSession() as session:
66
+ async with session.get(url) as resp:
67
+ if resp.status == 200:
68
+ data = await resp.read()
69
+ return io.BytesIO(data)
70
+ except Exception as e:
71
+ logger.error(f"Ошибка скачивания файла для ВК: {e}")
72
+ return None
73
+
74
+
75
+ def get_user(user_id):
76
+ res = supabase.table("users").select("*").eq("id", user_id).execute()
77
+ if not res.data:
78
+ new_user = {
79
+ "id": user_id, "balance": 0, "free_used": False, "status": "main",
80
+ "gender": None, "primary_photo": None, "last_style": None
81
+ }
82
+ supabase.table("users").insert(new_user).execute()
83
+ return new_user
84
+ return res.data[0]
85
+
86
+
87
+ async def is_subscribed(user_id):
88
+ try:
89
+ res = await bot.api.groups.is_member(group_id=GROUP_ID, user_id=user_id)
90
+ return bool(res)
91
+ except Exception as e:
92
+ logger.error(f"Ошибка проверки подписки: {e}")
93
+ return True
94
+
95
+
96
+ # --- КЛАВИАТУРЫ ---
97
+
98
+ def get_main_menu(free_used=False):
99
+ kb = Keyboard(one_time=False)
100
+ if not free_used:
101
+ kb.add(Text("Одна фотография в подарок"), color=KeyboardButtonColor.POSITIVE).row()
102
+
103
+ kb.add(Text("Стили фото"), color=KeyboardButtonColor.PRIMARY).row()
104
+ kb.add(Text("Творческий режим"), color=KeyboardButtonColor.PRIMARY).row()
105
+ kb.add(Text("Замена фото"), color=KeyboardButtonColor.PRIMARY).row()
106
+ kb.add(Text("Кошелек"), color=KeyboardButtonColor.PRIMARY).row()
107
+ kb.add(Text("🆘 Поддержка"), color=KeyboardButtonColor.PRIMARY)
108
+ return kb.get_json()
109
+
110
+
111
+ def get_photo_change_menu():
112
+ kb = Keyboard(inline=True)
113
+ kb.add(Text("🔄 Сменить пол"), color=KeyboardButtonColor.PRIMARY).row()
114
+ kb.add(Text("🔄 Сменить фото"), color=KeyboardButtonColor.PRIMARY).row()
115
+ kb.add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY)
116
+ return kb.get_json()
117
+
118
+
119
+ def get_fast_packs_menu():
120
+ kb = Keyboard(inline=True)
121
+ packs = supabase.table("settings").select("*").eq("is_premium", False).order("price").execute().data
122
+ for pack in packs:
123
+ kb.add(Text(f"📸 {pack['name']} - {int(pack['price'])} р"), color=KeyboardButtonColor.PRIMARY).row()
124
+ kb.add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY)
125
+ return kb.get_json()
126
+
127
+
128
+ def get_gender_menu():
129
+ return (Keyboard(inline=True)
130
+ .add(Text("🙋‍♂️ Мужской"), color=KeyboardButtonColor.PRIMARY)
131
+ .add(Text("🙋‍♀️ Женский"), color=KeyboardButtonColor.PRIMARY)
132
+ ).get_json()
133
+
134
+
135
+ def get_styles_menu(gender, page=1):
136
+ # Используем inline=True. Лимит 6 строк и 10 кнопок.
137
+ kb = Keyboard(inline=True)
138
+ g = gender if gender else "female"
139
+
140
+ if g in prompts.FACE_SWAP_STYLES:
141
+ all_styles = [s for s in prompts.FACE_SWAP_STYLES[g].keys() if s != "Одна фотография в подарок"]
142
+
143
+ # Пагинация: по 6 стилей на страницу
144
+ per_page = 6
145
+ start = (page - 1) * per_page
146
+ end = start + per_page
147
+ current_styles = all_styles[start:end]
148
+
149
+ # Размещаем по 2 кнопки в ряд
150
+ for i in range(0, len(current_styles), 2):
151
+ kb.add(Text(current_styles[i]), color=KeyboardButtonColor.PRIMARY)
152
+ if i + 1 < len(current_styles):
153
+ kb.add(Text(current_styles[i + 1]), color=KeyboardButtonColor.PRIMARY)
154
+ kb.row()
155
+
156
+ # Кнопки навигации между страницами
157
+ if page == 1 and len(all_styles) > per_page:
158
+ kb.add(Text("Ещё стили ➡️"), color=KeyboardButtonColor.PRIMARY).row()
159
+ elif page > 1:
160
+ kb.add(Text("⬅️ Назад к стилям"), color=KeyboardButtonColor.PRIMARY).row()
161
+
162
+ # Кнопка возврата
163
+ kb.add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY)
164
+ return kb.get_json()
165
+
166
+
167
+ # --- ОБРАБОТЧИКИ ---
168
+
169
+ @bot.on.message(text=["Начать", "начать", "Меню", "start", "⬅️ В главное меню"])
170
+ async def start_handler(message: Message):
171
+ user = get_user(message.from_id)
172
+ supabase.table("users").update({"status": "main"}).eq("id", message.from_id).execute()
173
+
174
+ if not user['free_used']:
175
+ msg = ("Привет 👋\n"
176
+ "Здесь ты можешь создать стильные AI-фотографии с собой в главной роли ✨\n\n"
177
+ "🎁 Для начала — бесплатный бонус: одна генерация.\n"
178
+ "Нажми «Одна фотография в подарок» и попробуй 👇\n\n")
179
+ else:
180
+ msg = ("Выбери режим в меню ниже — и начнём создавать твои новые образы ✨")
181
+
182
+ await message.answer(msg, keyboard=get_main_menu(user['free_used']))
183
+
184
+
185
+ @bot.on.message(text="🆘 Поддержка")
186
+ async def support_handler(message: Message):
187
+ supabase.table("users").update({"status": "wait_support"}).eq("id", message.from_id).execute()
188
+ await message.answer("Опишите вашу ситуацию прямо здесь, в чате. Я передам сообщение специалисту.")
189
+
190
+
191
+ @bot.on.message(text="Замена фото")
192
+ async def photo_change_main_handler(message: Message):
193
+ await message.answer("Что вы хотите изменить?", keyboard=get_photo_change_menu())
194
+
195
+
196
+ @bot.on.message(text=["Одна фотография в подарок", "✅ Я подписался"])
197
+ async def free_trial_handler(message: Message):
198
+ user = get_user(message.from_id)
199
+
200
+ if user['free_used']:
201
+ return await message.answer("Вы уже использовали бесплатную генерацию!", keyboard=get_main_menu(True))
202
+
203
+ if not user['gender']:
204
+ supabase.table("users").update({"status": "wait_gender_free"}).eq("id", message.from_id).execute()
205
+ return await message.answer("Выберите ваш пол:", keyboard=get_gender_menu())
206
+
207
+ if not user['primary_photo']:
208
+ supabase.table("users").update({"status": "wait_primary_photo"}).eq("id", message.from_id).execute()
209
+ return await message.answer("📸 Отправь одно своё фото — я сделаю стильный образ\n\n"
210
+ "Чтобы результат получился классным, важно:\n"
211
+ "• на фото только ты\n"
212
+ "• лицо хорошо освещено и видно целиком\n"
213
+ "• без очков, шапок и эффектов\n"
214
+ "• без других людей в кадре\n\n"
215
+ "Через пару минут ты получишь стильное нейро-изображение ✨\n\n"
216
+ "Жду фотографию ⬇️")
217
+
218
+ if not await is_subscribed(message.from_id):
219
+ subscribe_kb = (
220
+ Keyboard(inline=True)
221
+ .add(OpenLink(f"https://vk.com/club{GROUP_ID}", "📢 Подписаться"))
222
+ .row()
223
+ .add(Text("✅ Я подписался"), color=KeyboardButtonColor.POSITIVE)
224
+ ).get_json()
225
+
226
+ return await message.answer(
227
+ "🎁 Для бесплатной генерации подпишись на на��е сообщество.\n\n"
228
+ "После подписки нажмите кнопку «✅ Я подписался» ниже 👇",
229
+ keyboard=subscribe_kb
230
+ )
231
+
232
+ await message.answer("⏳ Генерирую фотографию... Подожди около 30 секунд")
233
+
234
+ gender = user['gender']
235
+ prompt = prompts.FACE_SWAP_STYLES[gender]["Одна фотография в подарок"]
236
+
237
+ paths_to_clean = []
238
+ bucket_path, temporary_url = await upload_to_bucket(user['primary_photo'], message.from_id, "src")
239
+ if bucket_path: paths_to_clean.append(bucket_path)
240
+
241
+ res_url = await generator.generate_image(
242
+ prompt=prompt,
243
+ gender=gender,
244
+ photo_url=temporary_url if temporary_url else user['primary_photo'],
245
+ lora_model=None
246
+ )
247
+
248
+ if res_url:
249
+ res_path, res_url_bucket = await upload_to_bucket(res_url, message.from_id, "res")
250
+ if res_path: paths_to_clean.append(res_path)
251
+
252
+ if res_url_bucket:
253
+ try:
254
+ file_stream = await download_file(res_url_bucket)
255
+ photo_vk = await uploader.upload(file_stream, file_name="result.png")
256
+ await message.answer("Отлично! Твоя фотография готова ✨", attachment=photo_vk)
257
+
258
+ promo_text = (
259
+ "Теперь ты можешь выбрать один из 10 готовых стилей для фото ✨\n"
260
+ "или создать собственный образ в режиме Творца.\n\n"
261
+ "Выбирай, с чего начнём 👇"
262
+ )
263
+ await message.answer(promo_text, keyboard=get_main_menu(True))
264
+ supabase.table("users").update({"free_used": True}).eq("id", message.from_id).execute()
265
+ except Exception as e:
266
+ logger.error(f"Ошибка в free_trial VK upload: {e}")
267
+ await message.answer("Техническая ошибка при отправке фото.")
268
+ else:
269
+ await message.answer("Ошибка при обработке результата.")
270
+ else:
271
+ await message.answer("Произошла ошибка при генерации. Попробуйте другое фото.")
272
+
273
+ if paths_to_clean:
274
+ supabase.storage.from_(BUCKET_NAME).remove(paths_to_clean)
275
+
276
+
277
+ @bot.on.message(text=["Кошелек"])
278
+ async def balance_handler(message: Message):
279
+ user = get_user(message.from_id)
280
+ photo_st = "Есть" if user['primary_photo'] else "Нет"
281
+
282
+ gender_ru = "Не выбран"
283
+ if user['gender'] == "male":
284
+ gender_ru = "Мужской"
285
+ elif user['gender'] == "female":
286
+ gender_ru = "Женский"
287
+
288
+ kb = Keyboard()
289
+ kb.add(Text("🛒 Купить генерации"), color=KeyboardButtonColor.POSITIVE).row()
290
+ kb.add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY)
291
+
292
+ await message.answer(
293
+ f"🪙 Доступно: {user['balance']} генераций\n"
294
+ f"👤 Пол: {gender_ru}\n"
295
+ f"🖼 Закрепленная фотография: {photo_st}",
296
+ keyboard=kb.get_json()
297
+ )
298
+
299
+
300
+ @bot.on.message(text="🛒 Купить генерации")
301
+ async def shop_handler(message: Message):
302
+ await message.answer("Выберите количество генераций:", keyboard=get_fast_packs_menu())
303
+
304
+
305
+ @bot.on.message(text="Стили фото")
306
+ async def styles_handler(message: Message):
307
+ user = get_user(message.from_id)
308
+ if not user['gender']: return await message.answer("Сначала выберите пол:", keyboard=get_gender_menu())
309
+
310
+ await message.answer("Здесь можно получить готовое фото примерно за 30 секунд.\n\n"
311
+ "В каждом стиле есть 10 разных вариантов изображений.\n"
312
+ "При каждой генерации они выбираются случайным образом.\n\n"
313
+ "Даже если выбрать один и тот же стиль несколько раз,\n"
314
+ "фотографии каждый раз будут разными.\n\n"
315
+ "Выбери стиль ниже 👇", keyboard=get_styles_menu(user['gender'], page=1))
316
+
317
+
318
+ @bot.on.message(text="Ещё стили ➡️")
319
+ async def styles_page_2_handler(message: Message):
320
+ user = get_user(message.from_id)
321
+ await message.answer("Вторая страница стилей 👇", keyboard=get_styles_menu(user['gender'], page=2))
322
+
323
+
324
+ @bot.on.message(text="⬅️ Назад к стилям")
325
+ async def styles_page_1_back_handler(message: Message):
326
+ user = get_user(message.from_id)
327
+ await message.answer("Первая стран��ца стилей 👇", keyboard=get_styles_menu(user['gender'], page=1))
328
+
329
+
330
+ @bot.on.message(text="Творческий режим")
331
+ async def creator_mode(message: Message):
332
+ user = get_user(message.from_id)
333
+ if user['balance'] <= 0:
334
+ return await message.answer("В твоем кошельке 0 генераций. Пополни баланс:",
335
+ keyboard=get_fast_packs_menu())
336
+
337
+ supabase.table("users").update({"status": "wait_custom_prompt"}).eq("id", message.from_id).execute()
338
+ kb = Keyboard(inline=True).add(Text("⬅️ В главное меню"), color=KeyboardButtonColor.PRIMARY).get_json()
339
+ await message.answer(
340
+ "Опиши идею для будущего фото ✨\n"
341
+ "Напиши, где ты находишься, во что одет(а) и какое настроение хочешь передать.\n"
342
+ "Можно описать даже самую смелую задумку — я превращу её в фото👇",
343
+ keyboard=kb)
344
+
345
+
346
+ # --- ГЛАВНЫЕ ОБРАБОТЧИКИ ---
347
+ @bot.on.message()
348
+ async def main_handler(message: Message):
349
+ user = get_user(message.from_id)
350
+ text = message.text
351
+
352
+ if text == "⬅️ В главное меню":
353
+ supabase.table("users").update({"status": "main"}).eq("id", message.from_id).execute()
354
+ return await start_handler(message)
355
+
356
+ # Логика поддержки
357
+ if user['status'] == "wait_support":
358
+ try:
359
+ await bot.api.messages.send(peer_id=ADMIN_ID, random_id=0,
360
+ message=f"🆘 Новое обращение!\nОт: [id{message.from_id}|пользователя]\nТекст: {text}")
361
+ except Exception as e:
362
+ logger.error(f"Support notification error: {e}")
363
+
364
+ supabase.table("users").update({"status": "main"}).eq("id", message.from_id).execute()
365
+ return await message.answer("Ваше обращение передано в поддержку, вам скоро ответят, ожидайте.",
366
+ keyboard=get_main_menu(user['free_used']))
367
+
368
+ # 1. Оплата
369
+ packs = supabase.table("settings").select("*").execute().data
370
+ for pack in packs:
371
+ if pack['name'] in text and str(int(pack['price'])) in text:
372
+ pay_url = (
373
+ f"https://yoomoney.ru/quickpay/confirm.xml?"
374
+ f"receiver={YOOMONEY_WALLET}&"
375
+ f"quickpay-form=button&"
376
+ f"sum={pack['price']}&"
377
+ f"label={message.from_id}&"
378
+ f"targets=Оплата+пакета+{pack['name'].replace(' ', '+')}"
379
+ )
380
+ kb = Keyboard(inline=True).add(OpenLink(pay_url, "💳 Перейти к оплате")).get_json()
381
+ return await message.answer(f"Вы выбрали: {pack['name']}\nК оплате: {int(pack['price'])} руб.", keyboard=kb)
382
+
383
+ # 2. Пол
384
+ if text in ["🙋‍♂️ Мужской", "🙋‍♀️ Женский"]:
385
+ g = "male" if "Мужской" in text else "female"
386
+ supabase.table("users").update({"gender": g}).eq("id", message.from_id).execute()
387
+
388
+ if user['status'] == "wait_gender_change" or user['status'] == "wait_gender_free":
389
+ supabase.table("users").update({"status": "main"}).eq("id", message.from_id).execute()
390
+ if user['status'] == "wait_gender_free" and not user['primary_photo']:
391
+ supabase.table("users").update({"status": "wait_primary_photo"}).eq("id", message.from_id).execute()
392
+ return await message.answer("📸 Загрузи 1 фото для бесплатной генерации")
393
+ return await message.answer("✅ Сохранено", keyboard=get_main_menu(user['free_used']))
394
+
395
+ supabase.table("users").update({"status": "main"}).eq("id", message.from_id).execute()
396
+ return await message.answer(f"✅ Пол сохранен", keyboard=get_main_menu(user['free_used']))
397
+
398
+ # 3. Кнопки смены
399
+ if text == "🔄 Сменить пол":
400
+ supabase.table("users").update({"status": "wait_gender_change"}).eq("id", message.from_id).execute()
401
+ return await message.answer("Выбери пол:", keyboard=get_gender_menu())
402
+
403
+ if text == "🔄 Сменить фото":
404
+ supabase.table("users").update({"status": "wait_primary_photo"}).eq("id", message.from_id).execute()
405
+ return await message.answer("Отправь новое фото лица:")
406
+
407
+ # 4. Прием фото
408
+ if user['status'] == "wait_primary_photo" or (message.attachments and not user['primary_photo']):
409
+ if message.attachments:
410
+ photo_sizes = message.attachments[0].photo.sizes
411
+ photo_url = sorted(photo_sizes, key=lambda x: x.width)[-1].url
412
+ supabase.table("users").update({"primary_photo": photo_url, "status": "main"}).eq("id",
413
+ message.from_id).execute()
414
+
415
+ if not user['free_used']:
416
+ return await free_trial_handler(message)
417
+
418
+ return await message.answer("✅ Фото лица принято", keyboard=get_main_menu(user['free_used']))
419
+
420
+ # 5. ГЕНЕРАЦИЯ
421
+ gender = user['gender'] or "female"
422
+ styles_dict = prompts.FACE_SWAP_STYLES.get(gender, {})
423
+
424
+ if text in styles_dict or user['status'] == "wait_custom_prompt":
425
+ if user['balance'] <= 0 and user['status'] != "wait_custom_prompt":
426
+ if not (text in styles_dict and not user['free_used'] and text == "Одна фотография в подарок"):
427
+ return await message.answer("В твоем кошельке 0 генераций. Пополни баланс:",
428
+ keyboard=get_fast_packs_menu())
429
+
430
+ if user['status'] == "wait_custom_prompt" and user['balance'] <= 0:
431
+ return await message.answer("В твоем кошельке 0 генераций. Пополни баланс:", keyboard=get_fast_packs_menu())
432
+
433
+ if not user['primary_photo']:
434
+ supabase.table("users").update({"status": "wait_primary_photo"}).eq("id", message.from_id).execute()
435
+ return await message.answer("Сначала отправь фото своего лица")
436
+
437
+ if text in styles_dict:
438
+ options = styles_dict[text]
439
+ last_val = user.get('last_style') or ""
440
+ last_idx = -1
441
+ if "|" in last_val and last_val.split("|")[0] == text:
442
+ try:
443
+ last_idx = int(last_val.split("|")[1])
444
+ except:
445
+ last_idx = -1
446
+ available_indices = [i for i in range(len(options)) if i != last_idx]
447
+ new_idx = random.choice(available_indices)
448
+ prompt = options[new_idx]
449
+ new_last_style = f"{text}|{new_idx}"
450
+ else:
451
+ prompt = text
452
+ new_last_style = "custom_prompt|0"
453
+
454
+ await message.answer("⏳ Генерирую фотографию... Подожди около 30 секунд")
455
+ paths_to_clean = []
456
+ bucket_path, temporary_url = await upload_to_bucket(user['primary_photo'], message.from_id, "src")
457
+ if bucket_path: paths_to_clean.append(bucket_path)
458
+
459
+ res_url = await generator.generate_image(prompt=prompt, gender=gender,
460
+ photo_url=temporary_url if temporary_url else user['primary_photo'])
461
+ if res_url:
462
+ res_path, res_url_bucket = await upload_to_bucket(res_url, message.from_id, "res")
463
+ if res_path: paths_to_clean.append(res_path)
464
+ if res_url_bucket:
465
+ try:
466
+ file_stream = await download_file(res_url_bucket)
467
+ photo_vk = await uploader.upload(file_stream, file_name="style.png")
468
+ await message.answer("Отлично! Твое фото готово ✨", attachment=photo_vk)
469
+ supabase.rpc("increment_balance", {"user_id": message.from_id, "amount": -1}).execute()
470
+ supabase.table("users").update({"last_style": new_last_style, "status": "main"}).eq("id",
471
+ message.from_id).execute()
472
+ except Exception as e:
473
+ logger.error(f"VK Style Error: {e}")
474
+ await message.answer("ВК не принял фото. Баланс сохранен.")
475
+ else:
476
+ await message.answer("Ошибка нейросети. Попробуйте другой запрос.")
477
+ if paths_to_clean: supabase.storage.from_(BUCKET_NAME).remove(paths_to_clean)
478
+ return
479
+
480
+ await message.answer("Выбирай действие в меню 👇", keyboard=get_main_menu(user['free_used']))
481
+
482
+
483
+ @app.route('/yoomoney', methods=['POST'])
484
+ def yoomoney_webhook():
485
+ d = request.form
486
+ check = f"{d.get('notification_type')}&{d.get('operation_id')}&{d.get('amount')}&{d.get('currency')}&{d.get('datetime')}&{d.get('sender')}&{d.get('codepro')}&{YOOMONEY_SECRET}&{d.get('label')}"
487
+ my_hash = hashlib.sha1(check.encode('utf-8')).hexdigest()
488
+ if my_hash == d.get('sha1_hash'):
489
+ try:
490
+ u_id = int(d.get('label'))
491
+ received_amount = float(d.get('withdraw_amount') or d.get('amount'))
492
+ packs = supabase.table("settings").select("*").execute().data
493
+ for pack in packs:
494
+ if abs(received_amount - float(pack['price'])) / float(pack['price']) < 0.05:
495
+ supabase.rpc("increment_balance", {"user_id": u_id, "amount": pack['amount']}).execute()
496
+ if loop:
497
+ asyncio.run_coroutine_threadsafe(bot.api.messages.send(peer_id=u_id, random_id=0,
498
+ message=f"✅ Спасибо! Оплата прошла успешно! Теперь выбирай стиль фото в меню"),
499
+ loop)
500
+ break
501
+ except Exception as e:
502
+ logger.error(f"Webhook error: {e}")
503
+ return "OK", 200
504
+
505
+
506
+ @app.route('/')
507
+ def index():
508
+ return "<h1>VK Bot is Running</h1>", 200
509
+
510
+
511
+ if __name__ == "__main__":
512
+ loop = asyncio.get_event_loop()
513
+ threading.Thread(target=lambda: app.run(host="0.0.0.0", port=7860, use_reloader=False), daemon=True).start()
514
+ bot.run_forever()
prompts.py ADDED
The diff for this file is too large to render. See raw diff
 
requirements.txt ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ vkbottle
2
+ fal-client
3
+ pydantic-settings
4
+ flask
5
+ supabase
6
+ loguru
7
+ yoomoney