Update states.py

states.py

@@ -1,4 +1,8 @@
 # states.py
+"""
+🎨 AI PhotoStudio — FSM States (COMPLETE + дополнено)
+"""
+
 from aiogram.fsm.state import State, StatesGroup
 
 class UserStates(StatesGroup):
@@ -24,5 +28,11 @@ class UserStates(StatesGroup):
     waiting_for_shot_type_photoshoot = State()
     waiting_for_prompt_photoshoot = State()
     
+    # 🔹 ДОПОЛНИТЕЛЬНЫЕ СОСТОЯНИЯ ДЛЯ УПРОЩЁННОГО ПРОЦЕССА
+    waiting_for_face_photo = State()      # получение фото лица
+    choosing_references = State()         # выбор, добавлять ли референсы
+    waiting_for_reference_photos = State() # ожидание референсных фото
+    choosing_style = State()              # выбор стиля
+    
     # 🔹 ОБЩЕЕ
     generating = State()