Upload 5 files
Browse files- Dockerfile +20 -0
- generator.py +64 -0
- main.py +514 -0
- prompts.py +0 -0
- 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
|