Update keyboards.py

keyboards.py

@@ -1,103 +1,67 @@
-# keyboards.py
 from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton, ReplyKeyboardMarkup, KeyboardButton
-from aiogram.utils.keyboard import InlineKeyboardBuilder
-
+from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder
 
 def get_main_menu():
-    """Главное меню — INLINE кнопки с callback_data"""
-    builder = InlineKeyboardBuilder()
-    builder.button(text="🔄 С заменой лица", callback_data="mode_generate")
-    builder.button(text="✨ Просто генерация", callback_data="mode_simple")
-    builder.button(text="🖼️ Замена лица на своём изображении", callback_data="mode_swap_own")
-    builder.button(text="✨ ИИ фотосессия", callback_data="mode_photoshoot")
-    builder.button(text="📊 Моя статистика", callback_data="stats")
-    builder.button(text="❓ Помощь", callback_data="help")
-    builder.button(text="🤖 О боте", callback_data="about")
-    builder.adjust(2, 2, 2, 1)
-    return builder.as_markup()
+    """Главное меню (ReplyKeyboard)"""
+    builder = ReplyKeyboardBuilder()
+    builder.row(
+        KeyboardButton(text="🎨 Создать фото"),
+        KeyboardButton(text="🖼️ С референсами")
+    )
+    builder.row(
+        KeyboardButton(text="📊 Моя статистика"),
+        KeyboardButton(text="⚙️ Настройки")
+    )
+    builder.row(KeyboardButton(text="❓ Помощь"))
+    builder.adjust(2, 2, 1)
+    return builder.as_markup(resize_keyboard=True)
 
-
-def get_gender_keyboard():
-    """Выбор пола"""
+def get_reference_menu():
+    """Меню выбора референсов (Inline)"""
     builder = InlineKeyboardBuilder()
-    builder.button(text="👨 Мужчина", callback_data="gender_male")
-    builder.button(text="👩 Женщина", callback_data="gender_female")
+    builder.row(
+        InlineKeyboardButton(text="📸 Только лицо", callback_data="ref_face_only"),
+        InlineKeyboardButton(text="🎨 + Стиль", callback_data="ref_face_style")
+    )
+    builder.row(
+        InlineKeyboardButton(text="🧘 + Поза", callback_data="ref_face_pose"),
+        InlineKeyboardButton(text="🖼️ Всё вместе", callback_data="ref_all")
+    )
+    builder.row(InlineKeyboardButton(text="⬅️ Назад", callback_data="back_to_main"))
     builder.adjust(2)
     return builder.as_markup()
 
-
-def get_style_keyboard():
-    """Выбор стиля — 22 варианта"""
+def get_progress_keyboard(gen_id: str):
+    """Клавиатура с кнопкой отмены (Inline)"""
     builder = InlineKeyboardBuilder()
-    
-    # Реализм
-    builder.button(text="📸 Фотореализм", callback_data="style_photorealistic")
-    builder.button(text="🎯 Гиперреализм", callback_data="style_hyperrealistic")
-    builder.button(text="🎬 Кино кадр", callback_data="style_cinematic")
-    
-    # Художественные
-    builder.button(text="🎨 Арт", callback_data="style_art")
-    builder.button(text="🖼️ Масло", callback_data="style_oil_painting")
-    builder.button(text="💧 Акварель", callback_data="style_watercolor")
-    builder.button(text="✏️ Скетч", callback_data="style_sketch")
-    
-    # Фантастика
-    builder.button(text="🔥 Киберпанк", callback_data="style_cyberpunk")
-    builder.button(text="🧙 Фэнтези", callback_data="style_fantasy")
-    builder.button(text="🚀 Sci-Fi", callback_data="style_scifi")
-    builder.button(text="🌌 Космос", callback_data="style_space")
-    
-    # Стилизации
-    builder.button(text="✨ Винтаж", callback_data="style_vintage")
-    builder.button(text="🎭 Нуар", callback_data="style_noir")
-    builder.button(text="🎪 Поп-арт", callback_data="style_popart")
-    builder.button(text="📰 Комикс", callback_data="style_comic")
-    
-    # Аниме/Игры
-    builder.button(text="🌸 Аниме", callback_data="style_anime")
-    builder.button(text="🎮 3D Render", callback_data="style_3d_render")
-    builder.button(text="👾 Пиксель-арт", callback_data="style_pixel_art")
-    
-    # Другие
-    builder.button(text="🌅 Импрессионизм", callback_data="style_impressionism")
-    builder.button(text="🏛️ Классика", callback_data="style_classical")
-    builder.button(text="🎲 Сюрреализм", callback_data="style_surrealism")
-    
-    builder.adjust(2, 2, 2, 2, 2, 2, 2, 1)
+    builder.button(text="❌ Отменить", callback_data=f"cancel_{gen_id}")
     return builder.as_markup()
 
-
-def get_shot_type_keyboard():
-    """Выбор типа кадра"""
+def get_result_keyboard(gen_id: str):
+    """Клавиатура после генерации (Inline)"""
     builder = InlineKeyboardBuilder()
-    builder.button(text="👤 Портрет", callback_data="shot_portrait")
-    builder.button(text="🚶 В полный рост", callback_data="shot_fullbody")
-    builder.adjust(2)
+    builder.row(
+        InlineKeyboardButton(text="🔄 Ещё раз", callback_data=f"retry_{gen_id}"),
+        InlineKeyboardButton(text="🎨 Другой стиль", callback_data="choose_style")
+    )
+    builder.row(
+        InlineKeyboardButton(text="🏠 В меню", callback_data="back_to_main")
+    )
     return builder.as_markup()
 
-
-def get_reply_keyboard():
-    """
-    🔥 НОВОЕ: Нижнее меню — ReplyKeyboard (серые кнопки внизу)
-    Эти кнопки отправляют текст, а не callback_data
-    """
-    return ReplyKeyboardMarkup(
-        keyboard=[
-            [
-                KeyboardButton(text="🔄 С заменой лица"),
-                KeyboardButton(text="✨ Просто генерация")
-            ],
-            [
-                KeyboardButton(text="🖼️ Замена лица на своём изображении"),
-                KeyboardButton(text="✨ ИИ фотосессия")
-            ],
-            [
-                KeyboardButton(text="📊 Моя статистика"),
-                KeyboardButton(text="❓ Помощь"),
-                KeyboardButton(text="🤖 О боте")
-            ]
-        ],
-        resize_keyboard=True,  # Кнопки подстраиваются под размер экрана
-        one_time_keyboard=False,  # Меню не исчезает после нажатия
-        input_field_placeholder="Выберите действие 👇"  # Подсказка в поле ввода
-    )
+def get_style_menu():
+    """Меню выбора стиля (Inline)"""
+    builder = InlineKeyboardBuilder()
+    styles = [
+        ("🎬 Кинематографичный", "style_cinematic"),
+        ("🌟 Портрет", "style_portrait"),
+        ("🎨 Арт", "style_art"),
+        ("📸 Реализм", "style_realistic"),
+        ("🌃 Киберпанк", "style_cyberpunk"),
+        ("✨ Фэнтези", "style_fantasy"),
+    ]
+    for text, callback in styles:
+        builder.button(text=text, callback_data=callback)
+    builder.adjust(2)
+    builder.button(text="⏭️ Пропустить", callback_data="style_skip")
+    return builder.as_markup()