From 30138bab38927b076a1614bded0930181eecabb8 Mon Sep 17 00:00:00 2001
From: xds <iam@voroninv.ru>
Date: Fri, 13 Feb 2026 11:18:11 +0300
Subject: [PATCH] feat: Introduce generation grouping, enabling multiple
 generations per request via a new `count` parameter and retrieval by group
 ID.

---
 .../generation_router.cpython-313.pyc         | Bin 10511 -> 11442 bytes
 api/endpoints/generation_router.py            |  37 ++++++++++++------
 api/models/GenerationRequest.py               |   8 +++-
 .../GenerationRequest.cpython-313.pyc         | Bin 3258 -> 3698 bytes
 .../generation_service.cpython-313.pyc        | Bin 26145 -> 27060 bytes
 api/service/generation_service.py             |  21 ++++++++--
 models/Generation.py                          |   1 +
 models/__pycache__/Generation.cpython-313.pyc | Bin 3268 -> 3329 bytes
 .../generation_repo.cpython-313.pyc           | Bin 5472 -> 6234 bytes
 repos/generation_repo.py                      |   8 ++++
 10 files changed, 58 insertions(+), 17 deletions(-)

diff --git a/api/endpoints/__pycache__/generation_router.cpython-313.pyc b/api/endpoints/__pycache__/generation_router.cpython-313.pyc
index fb99eafe262ca3d83034cdbbbc2883a81c565f81..f9e6ebaa2cb1c7ea7f1fe8d856f873845621c617 100644
GIT binary patch
delta 2790
zcmZ`)O-vg{6y8~{jlnql7>5lISZsrF2!;d`!cS-jX$&}!$U~~6Z0fjpAx^=zW=++G
zRFL)%sj1R7Q>mBUdT3gy>W?ouRi)~wQdJ{T(X4t%)l+X&q^Z(V=k4M(pkyU~`{wQY
z`QCdo<M*TAUfQ?ibe3`K`SI5~N!jz^K9LVvKWw|ZJl;SWBo1@9G;(mfku>UZS)^%P
zB9bmUBF*C-;?d>uh?jU}Z*&wQ?S4F*+|q*#9cg9poDu(c8)=g`S@3XpAFk*`xU!EY
z0XY!8VEWP?<m+uKIEHxalIwAe?8I)l5$~7VajjezD#KzOcTKp4dba(@VF*G$Kx6)*
zuYl`iUhc3m8{8lx*{Vz1aqjZ9)*y3=(FVR7wbAo@d(tp%mb<Wr{R#=aW)Pq34(-D&
zdr}|S#osIhx5^E;x{sF~tP@p*g7g(+YoVo<@)4t?T^U$`4J=FBo-9X!s~OxBaDaY?
z>UI7t%fhZk?TX4;c4$w;?&m~oe?IR*6>Kbr<&Hv!LN?q1u`74XBc!+Xun>qq`Nf&=
zxU=ZUO05mqan~MVJ<n<A5qeUnUp|Vv`*=p6uqQ$+o)M>KPn=^cPTK60Dd3>&iDt~D
zDd1zz9@dVc8Fy(%)xCQX_RW<Cj?@2yuaJ#?VVypp)rOUnLgLwEIu$1A#Y|LDGwGD7
zP^;|%I*^;TweqNyzHRTSY+^Ks1keoNp+DHuwssKPX>aKx)S45^R#0^d(**3q4<Fh9
z0`v>VU}Y<aL4a-VCk{{hVYXK745aOcbg+e1l_BYcOg1*lUa{mtd`=;~xsmb*yfgp-
z2EnFeQB_DRnII?F%FIW5D~6l;!N|1S&IdX$B}4S%igpL&ArT#NKUX~Aubc#V7~mAZ
z2!jA3XW->57Nfc5ld0E~L~Q12oWy6c28954=mo+hL&gBY0OJ5~g^?4~>++tN1W^Jo
znxKCH7y_7Jpq4P$)r^WvTFu7h%qbE4>3c4e%kA$$M^A#=9@4x{+GvZ-piyKsPU}S9
zkOXeeVsI8dJd*~2>raPE^oIClcU9*c?3T_Y{l(_+mS!)AGZtoBcuJY@4*PA{=%dPZ
zns2VE77E|YH*GF)(_fr!`cLygpNo4*Q29<y<z3up1IML=X-B~lTe`RlLexo*RJmxf
z-m!e1pJ?Pb2PbhUD>JMp7+$nuzK{2Dc5d>R%1bO*N|@z(Ewd;fOkO`Ur|fKLOlL05
z#OLRggfyEbQlXT7DZp#4V)G44R<7pImCO3UQ9m#4Kt{lr<|QU+;_gIh?A14o5TqJD
zmo20d%6u%bxR4>WOcAh=mmolKHhL5b%cVIP#K`vPR8<?ZlcZ(XD#oFy*_~=T4cy1q
z*iTg%{Kc(skL{e*aVv5&lCK`TfARi!zB2qk82wE=v>^u9#NdY5zb5u?h<HuJ`6+o_
zyu49;c_s2N99^%$KP-K3dEoZ1xd-lfzY2aHTt76{|G+)4y0kKpufe}NJYT=D=9tb4
z(|=nz*DL6$R)e3aK9>0HVRQ<iRSO#K7ghy3$mQPQL1DGC9OOakDU^G+CWx9(GGl#1
zUw01A%qXfVC6rWBNf_tRM*Q@l+kad?TX3ub0=WVn#eQ98OBFVl*%&ed764g#eE(7|
zaNsQgjnK~y{^TJqGi{(o<I}3<TFfTr)u8eQJJYH7d@R10y-Kdq4;ot?NeE!vXLPkO
zjKZ|E$+t{oaCYuE&a+_m8o*Tm-7Uo=&88|dM9C5Y3n3BD#<g-Y5zH1T1g-|v$pDC(
zk3L%RI#cU#6O0xROfr{i+Cb<uz1_U!9ARqKifxLwJ&j#gz{Bi{R>F3fl2sCWI&%Z8
z0KTRVJqqflFL*b2QcGPvUqEDPtxDgYYA{^9eGY)WRL#Z-i^zIaN3Zxib&y)Cj%SiV
zC6&melc}s4)H_&DKk<pn`oDxewO(dOqF`-dM#c?{L8sp@uv?9JLCKoUpsl$^r>4%w
zj&_A&ljmY1=O(9KoSf2m>jT_krkrsG?1PN(;=D3MK4Lo7Cr%Br%|L@Kz7m=Lm#O63
zrIv4yZE&SNk514^f6^UAk4v3s0c}}0erS^iannQp^k=ce@%CGFH|z4{P5G|yN?jg}
z{t=$cpHtSuvkyZ9`N6a6p^3k(HYB5au1yZ!zjdG9w1Dua<H%$E3xpj*n>;8ttLYv8
O6cXvhwwyp2yZ-^s=Uy2A

delta 1791
zcmZ`)No*Tc7@jvCPaG#s92{q{lX_<2-3eK2Ck;uLG;Km{a=?$Z9crca5V^#0^k#w(
zs3<2cfKYh1sy74(1Zs<~95{0*Jyb%n5X%jT0}%lv)C=!_6Whh5Bh5Ge{M+|0Z|tvL
z{-Drz<ng!#cHP^5zj7yZ&^IF9l@8(`&M5*GaCd$zH%`X6?9QLag-D3YJ$WUk5|zun
zd5vhgR#-bY^rymE1jz(}L|C{dAI-%`Oc8WR6L25ypGMfbD3Z7yFJOuyKDvqm3$bQQ
zM#MfnhzE5K_Uq$#NKfKnJ&<wZk$`Ybx`wiLYJsF$@t>mk)popugSx0sI+-6H(~<7v
z(tmS}cjTI4xnAM5q(ueA=&xc#?ZQeckE%~&jor;G6JVh?#GrGgSp|yHv(hM!ot?Jp
z=3bmVtovCP!<jxD?_i#7GNS~<r`vsqIH^yz7$Mq)QytNBt>{UrJA#b;4aZdKf7sdl
zraEHN&C2N+X(;Pry9wZFJhLdW4K$}e%PO#$=Q^S@Ec!K=m@}cPg>8^7q!&fKM-%jc
z<}Uf$V>-&nIMb1HVXHThrI(zykdwOQ%k*vec{EbLFGocbp$}X$-V<=9LI4VYN|(B8
zauUQOz1{sYiqsF>`{;=<(<EFZ3WgX!oKEyEdm|uD0Vn_}t@UcD6cbGs1LgQ4lkBB2
zuuE#yU90#idlW0x%`Jn>*T3$)FDg$#>=O8N?U@ECR?1|Vy|wu0?f#X}A~>}?kNH3c
zuH+Q;dQ&~XLvkGIYu>-bi8COd1y}(%#~^{o1$cOlo1sPDXgCg5-po2kPXnw0Tm*pc
zm@LzqzKNL)5FZ*Bw*Z_1c!q)5#h}$V{oUtBg^!$rsJ;=HL~5&`H$5w*l3|)k+1RNV
zWxF()MCpxS>?EfK)`UbbBvt@a1`WrSv13Y10t!A@@&XtEUZVd5->N(PSv1$;d8l!b
zH*Swsw%coWdy#Grg-<`WvpjG-+8$^xHK9cB4t-g-#;!}~9Q{N27JWv)QOD60`lA}{
zDT0@Md}zP6j@IalT6n49uGY%NcCoxy-DMkT)9c`V72pa0k2aXpkWHgR42x9gEiD*@
zb(kIoFpbt_kEIHM0I$~%wO^3D%EXd`di{g+w{YV!KMPX(L^lRlA%<lX*?<kQX;o@F
zeDRO2pv!hFcdWqzK7fy@5;f3C`cCwFdNrmdhMBH0z#BE6TyJL|0e!b=Z4$<78Dx}x
z5!2Xj-gmK)^#IspQrlzthO}EVt%g*tlq|wOO@2P+nG1Oq{2Y}X%Rcr6(DC!Yw^F6)
zxSHUFZ8FOdy9N7WbS<-b_O^}F<UOW?{}%HH2fZGTcAXdhVv6(i&*OKHys#g<$2!?e
pR{R_2&u$N@q9cbOE*^>?9xC*c)N^Q%{*<&(m_A6=rFU7Te*wM}Z&UyP

diff --git a/api/endpoints/generation_router.py b/api/endpoints/generation_router.py
index 85c4f61..79f68bd 100644
--- a/api/endpoints/generation_router.py
+++ b/api/endpoints/generation_router.py
@@ -8,7 +8,7 @@ from api import service
 from api.dependency import get_generation_service, get_project_id, get_dao
 from repos.dao import DAO
 
-from api.models.GenerationRequest import GenerationResponse, GenerationRequest,  GenerationsResponse, PromptResponse, PromptRequest
+from api.models.GenerationRequest import GenerationResponse, GenerationRequest, GenerationsResponse, PromptResponse, PromptRequest, GenerationGroupResponse
 from api.service.generation_service import GenerationService
 from models.Generation import Generation
 
@@ -68,12 +68,12 @@ async def get_generations(character_id: Optional[str] = None, limit: int = 10, o
     return await generation_service.get_generations(character_id, limit=limit, offset=offset, user_id=user_id_filter, project_id=project_id)
 
 
-@router.post("/_run", response_model=GenerationResponse)
+@router.post("/_run", response_model=GenerationGroupResponse)
 async def post_generation(generation: GenerationRequest, request: Request,
                           generation_service: GenerationService = Depends(get_generation_service),
                           current_user: dict = Depends(get_current_user),
                           project_id: Optional[str] = Depends(get_project_id),
-                          dao: DAO = Depends(get_dao)) -> GenerationResponse:
+                          dao: DAO = Depends(get_dao)) -> GenerationGroupResponse:
     logger.info(f"post_generation (run) called. LinkedCharId: {generation.linked_character_id}, PromptLength: {len(generation.prompt)}")
 
     if project_id:
@@ -85,16 +85,6 @@ async def post_generation(generation: GenerationRequest, request: Request,
     return await generation_service.create_generation_task(generation, user_id=str(current_user.get("_id")))
 
 
-@router.get("/{generation_id}", response_model=GenerationResponse)
-async def get_generation(generation_id: str,
-                         generation_service: GenerationService = Depends(get_generation_service),
-                         current_user: dict = Depends(get_current_user)) -> GenerationResponse:
-    logger.debug(f"get_generation called for ID: {generation_id}")
-    gen = await generation_service.get_generation(generation_id)
-    if gen and gen.created_by != str(current_user["_id"]):
-        raise HTTPException(status_code=403, detail="Access denied")
-    return gen
-
 
 @router.get("/running")
 async def get_running_generations(request: Request,
@@ -113,6 +103,27 @@ async def get_running_generations(request: Request,
     return await generation_service.get_running_generations(user_id=user_id_filter, project_id=project_id)
 
 
+@router.get("/group/{group_id}", response_model=GenerationGroupResponse)
+async def get_generation_group(group_id: str,
+                               generation_service: GenerationService = Depends(get_generation_service),
+                               current_user: dict = Depends(get_current_user)):
+    logger.info(f"get_generation_group called for group_id: {group_id}")
+    generations = await generation_service.dao.generations.get_generations_by_group(group_id)
+    gen_responses = [GenerationResponse(**gen.model_dump()) for gen in generations]
+    return GenerationGroupResponse(generation_group_id=group_id, generations=gen_responses)
+
+
+@router.get("/{generation_id}", response_model=GenerationResponse)
+async def get_generation(generation_id: str,
+                         generation_service: GenerationService = Depends(get_generation_service),
+                         current_user: dict = Depends(get_current_user)) -> GenerationResponse:
+    logger.debug(f"get_generation called for ID: {generation_id}")
+    gen = await generation_service.get_generation(generation_id)
+    if gen and gen.created_by != str(current_user["_id"]):
+        raise HTTPException(status_code=403, detail="Access denied")
+    return gen
+
+
 
 
 @router.post("/import", response_model=GenerationResponse)
diff --git a/api/models/GenerationRequest.py b/api/models/GenerationRequest.py
index 40e9d18..33a4010 100644
--- a/api/models/GenerationRequest.py
+++ b/api/models/GenerationRequest.py
@@ -1,7 +1,7 @@
 from datetime import datetime, UTC
 from typing import List, Optional
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 from models.Asset import Asset
 from models.Generation import GenerationStatus
@@ -17,6 +17,7 @@ class GenerationRequest(BaseModel):
     use_profile_image: bool = True
     assets_list: List[str]
     project_id: Optional[str] = None
+    count: int = Field(default=1, ge=1, le=10)
 
 
 class GenerationsResponse(BaseModel):
@@ -45,10 +46,15 @@ class GenerationResponse(BaseModel):
     progress: int = 0
     cost: Optional[float] = None
     created_by: Optional[str] = None
+    generation_group_id: Optional[str] = None
     created_at: datetime = datetime.now(UTC)
     updated_at: datetime = datetime.now(UTC)
 
 
+class GenerationGroupResponse(BaseModel):
+    generation_group_id: str
+    generations: List[GenerationResponse]
+
 
 class PromptRequest(BaseModel):
     prompt: str
diff --git a/api/models/__pycache__/GenerationRequest.cpython-313.pyc b/api/models/__pycache__/GenerationRequest.cpython-313.pyc
index b958d6046962717bf8d4280f85aad8b55bb9d2cc..0726ddb09aee4939f0106e13d72247b80b8f7d21 100644
GIT binary patch
delta 1779
zcma)6O>7%g5Ps`j|E>SB@oy66&ux`#LhMp#Lum<Vsfla{Q)>PW7M5eb7=z<=XV;=P
z!-dcbQnVMtg^CkM9FVvHXO7TNsf3kiE}$wm)L0xi@#fjMK}2fT+HYsSnVmOpzVZ8Z
z>_Oar=<~^hUH5+exbAcA_+!HFJ9DYGvQBW$kOC=4Iak&VZf@Iia@GSLZaZ?`tPgxy
zKlrl&2+WWeNsxkboD^I|$5|0m5&|Qfbu(5z!YVsohu22A%g0>)Bd#E41B@Lx!iG4j
zFn06^8?FV_(9U0K&}O+Wmv#D*R;3Ng{$iars^+4wE<5^Fki$Lm1L3<wF~lnLvY)Tu
zKQBVK7$`amfjJgb7=cL5rYhzY+xGl{z?cJUiYi*pD%~u%8-``8QOnk#mc632o5q(!
zjha8%&IyX?5T`pH7HoO)nS8pCdGpnDIuE0a0WZ?h%lUK;g7_ZBcd_0yAi_Nxn$`e2
zdSVDJgemldOC{YX8}&-5Y#31AXd6^7m0+6r4%tntZ0a?gf}Xlj)l*e!ek!g60*nU)
zEJnX!-WTcQ9qHt+N~C-J)+=9}*;SJ7r$2h?u9SpD9%{ZWQ6Xx6FP+I?;=JmD8Fb7d
zBoIy@@cK@nHIJ}>fbsn-PoS0j7oz5}eR7tkN3O!|;WZcua`=N_UYA1O+_bL<oq`18
zMORV6MT+)96vZe`Tzh8t)zC<g;^bLWhFvgKoM6$h$Bdrl(PLcbahRz|>I9r-X%6C-
z8Cte_)pB6F+j`#eZI<f|S}g%B>sk{OT#akV1q@-l$DozBOD)j0S_VW-!})n9Sffn}
zOfaoks)5#Sl~_)>!jc#)SJ|-ScB?wDdGplSV|#rPm*c(0@q=Xq9O8Z*C@h~T!2a}w
zIgeK7S;PwrEc<3dD;sbTiRTcwXe($vkFbhx2?6&GFCd^Yme|zZ0|{TRAfyqF(;6Gc
zyBBAM9%71Y6Z3D^J-@niTZ(pLw@!S*#%{^{#XaZXgK(P7S<1X=_jH^Cl|`<E1g@H0
zby(50Vei(kb=Rog(waIwIEO3o0-r>yKcntP^bMEb<^8{SRykzQjeyB@lyjTxjm7Us
z@jWGcJ2JnksCOl`KiU~nk}EarDsbI>je7H4lvCHKp+9CldHkFgbzC#o(Ti}w{866j
z_iNpJAYXlF6+O7@U>)Zp_I+ITtN1y>D!X+j1GZUj;KN*hf9T1p5%Vk0VuznSmUjzJ
zP(67GcL;9gx0C_(P$a<79EQ?ndrOChlj&nlehT#u1=jOH)A?3;MgKiKEZ4;?tyFC{
z=nA~SqVX@TZ!q{%6a-<PB=*VdeRAp#GW8ot?vv$RvfOjHg;dwwBka+O*@SRc?Gg6q
e$x&gcJJuuY(R<n^EOhJ0FZ5*dGjCMFY5WH+;$}Jk

delta 1364
zcma)5O=uHA6rM?Tlg(x~O=^-R>95f$Yq4r=o2pf4C5I+;6<0+eps`tNG)=<n7LV~{
zPfEjFdJ}r{;6-nWdKZKip_B!wARYt{qVev$S&a$-U6OC#eD81H&wTBBmJEGURT;5o
z<L>=(#@7rb@ayLOLyaM1AVbLea}trbEan3_naEuB<%2neC^?m=xey5rp#(}J!=FNi
zRParBAsj-ai?ad7%73t7&ITE){J};zt1>pkS-%^iyZJsQ?plm!-876f1v`P+qGg_`
zmCOpA$Mf`!9KbK>I({x03W+XyG?g|yE5akB#ptMa5{Iej9h`;woS;6ZXN|eb7qVHM
zMBw{^KnrK}Y@T$(ISOZQxoVRb*IcjFDntZL58d$g;spKdz1S530s!1Clb#byc}v(!
z_Hv%SkPlHTPU1M762}?|M(UbGlAufh^aAt&@W=xo?FQHb0QPRAVUYI!f;bZ__`Vgi
z9f@tk4{Sz8-U=hzk?58XbsfNVf&JSL8w1^tefY`KFQS1Ph1B`UwyrItQK4@FyGPPr
zNEJk*2O|B4^#Hd{fVGh%gNuTeA{sk<f)X81x#ajPyJ#<4x})AGj+QHC=_)acR;^0H
z%)*@kZ~QKB2-@bt&8u}%TdLb6P6cVbAulj{o4FNjM_#U%e#*Q{+EB-)_CsXe)Bs3_
z0ek>FB`g9TJ2?UppTHPMM*+qeIO2^;t!R@8CSN1aFn2l`kfs2R0UQT70RYSHc&oKL
zM1bQoz)65k^Mv;B9zlyNkv<H3piOz7kMEAgb|*<+1#^p#obaZom2&kqG{CY<+v0`m
zTHqg*p9XhcTc>ZB0WH7;tt$g=>KUH;v68vP3xm`@bNT;~t5b$7xC1~i%g=yqt<rTb
z?=C86bS+ACbtLR4OK_L1(HSU^Nc4f)(}2mf6=rpL$s)f8r!XbIe(vSq9pxqQtbgAh
zN1A5gmn-HeGSA##&8<rezIZXlJ1D(_c7H?z?Dr0h?x4&j%CvkEJ~_M=Xdwv?x8fcg
mThm&Iy;^b{A6!ee5PP-8JvhBO2Yk9EV;o++@*S}kFYgD&ZT}Yl

diff --git a/api/service/__pycache__/generation_service.cpython-313.pyc b/api/service/__pycache__/generation_service.cpython-313.pyc
index 24461f4c16b12f85ff199df4a5eacfea0d9fa315..1970192c370f5adde4006310cc1fba07cd7354a9 100644
GIT binary patch
delta 4716
zcmb7HYj9h|6~24*vL#!VAF?G|v1Iuv$8j9|$ivRViH)(Gh~SGuh?OX^bfYMdC9_uw
zq%YBs($IFAqzfdp4J`vREi+KsQHP;J%aD*1+J+(*+GK8%Kxz4rG6SLX{g~-FdnG$z
zpfkN=e>&&v*>iW#WB1(0&XXU$POKl6mzxE6&iwkT@$A%btAiXm-oCriFk$sq&<c+r
z2@;vGQQPs8-C53=C!GFzTJLY54SpAO`IpcoTyB|Y^t-9s@1Y)l6K&$MvI(!hnKt`d
zXbb1dCtCe&w9Vg6+x;E1gUhTFK7S|e<aEWvQhyii@(4jPxdT^q!xI<0La@7^XhOPN
zvITpj%AhH@!T=(vD%d7f2iv8ZphL0;ON;WLCD;=*_lc-ez%r?}Uf8FnH#xwfR97!Z
zj()LVwKqn5PWC7nHySdTc(jlGflLrR>k)@4)w(e`DbsK|o=T3<RAw$06|d+VjWAQS
z#Am}XSqUZLN}75=LYvqQ;{jr2^TsASs!%tAHiVRynM}_bRsz|OJ7c<?)EtA2JtPDL
zN+f}b5}`WD%Q{P&iI)XR-GIr`R?^HKES(}P?2n~Ovn`S%Xoqc^f@baj=K^I8*7lW1
z^|;A>gw{#_OS1|z8w#2Z$rZHqiP9489%q4T<XpXE=3{QoHAwbVK{JS4eR|2mMN6b6
z+87gkt!m|>pCp+ch)3kBqp;6_s*8qGu<?`Ty@P|?2M9w5YXN))Rj<g2J@jM76x^W)
zA3DGWEk1HHlPxV)?t5ccQRK8jn{y{DH9E4MJ!SQf0rskOoJ_I0inblx-Ut{_D`Rpx
zq^XBy4u;}UMK#Lzq~&Ck`fwSY2wv1Ik7==%LupW;Z8*>l;4{$;Nb``6uoD&I@T%Wc
z43jOnW?N8P#^tT3sQ-vb8qlPC`V`vO#-6Hj!}Rl2x030ctNNd0*)+<72s;5(bA-y_
zv>b)S(hvxII?6S|s1ahD>@7yVxfwQV?-;7&Q)W|9IT4CxX6I-XPASzEe4}N$i(5dL
zL5Q$<d!2CrNtwN8ZzIv1z4iyh9|eiephW}~RV#Y1;2Wp|itnT`)Y571*@uF?0E!NP
z2Mi?%W^=TX{kZ{$g>2H658~@hQyKLf0IF$nDi9bCj1_GzEOyZ*JM0`Il%01bR;tyD
z>j}lMFd?uJJs2uJMj-Bk{}0~=XcjzBXH$f2sHvGu>Zf#(YC)LB#qwtwGmSs}#a}2)
ze3H85Leem8(l|kw(ihL7YT=e{7&lG{lckI}Yrtr~j(OUhI>V%eRW?><O+{6Yu%$;3
zEU<>@@}i;b{SsL%wg@Fcz`Ihnhxo*RPo!xW=E3oqR6`_{Nv2gD94^!Xa@9b?$(XD<
zLt27_lz1|hkgpHSGMo-O6MQ9f!v<s&1nx?1`T(%1iONbQkyca^Qp>=NA>ijyNkvvl
zU>KsuAE>#d&7;{2h?E4tRpDOYqRo53)-iAEIAVCq(l~EvyX0+sLVi5<cvrs6{Z>`Y
z5!0p0+GClAT}Os5Hnm=8T0h^kKEGiyzjN;Orn@dQ+;wF0#j@(7fd>L7hkjGm@qt08
zYW%ZNu(jt01M?Npd`T2UE=vBPU=kl1E*V}vN=}=}@G|3ROY`t@azlV|u{^-v7X@Kg
zA;gl9)Wg4_QJ5}75N-(ro2z$hlE?}oiL1#786Zi+l-L3D@d-nwY;R3~5$&1M-7wOh
zSV)uG+3AK_R<f&FoGix(69coAIUDrbM-z()BJ>eS2Zv)x(l~8no#U=9uP|kVe?u`H
ziocK!_Toz=Ll0<7>e%l7dUmW{-^?>(v`Lt*EvlG=@F@rnq~Od+L1+{fR>M}fYuJ1B
z73@@RxtKIDXP~is%9N-mPFzqFC!6@PP3%~&%PR{Xu?}SjOY#<{O}4V*ORC`>JnL$$
zG6pE-j*qA&`JPB36P0uCx?DP=c1E&tPumjhPy!{2>7o_2@#f+2&7&hU#kP7{vU@?U
z8ly9qvi#sEl=i(84<Xe^GfB8>)xoiaE4Of!bbn&XcMl~}k#Is;*R4%H3Wk)c0Ds4P
zy=1FBVL5KeH~L@ceQh9j&2x-o`RP_o>p+K}VKj^sua3nXU|!Kf2zW)PmO{m)!^(c`
z^gE1W3;U#BKsJSNKLYp47lFb$M8RSik~I6Pw=w$wO3@#@=5}i(2lpI`4+5x5{|_g8
z9_<X0!lo})+K(77Hn{U8j%)h1(yB{t-@JQgzH8luuJQS<@w~JnKk1&%1JxD8fil;X
zirS0a{cqToUa9b2T<L$q*2kV{{#Uk^JTTm_u}e7JLN<2lPqz;1fjQ$NqqXErCqcTa
ze6(73W=+RPm-v#GY-%;W)I2N#bGDU?bQsUJ8IWE|K<R9kd(>(;XAyxuS5A<&8j!Bm
zfjQL-8Byr6BE>??`C6Z-`h+&>?Umm$&`-m*+2K_;*FK64vnJxn{cvkW_J%3k1$2zP
zwW`O+gB!}#uHH#3TCRdFJ;Xk@#%XLsa*&-`Q}4uZs+Ac<4$aZjo;aMUytktxEWgIH
znIEfTsJ$8WEQMx*_kH+oiZ!Yl#<vWQjnXGjfc=eHg4ZS*qj;cboxzvc(4b=!J3#6|
zXhOh}Q+1$Ck0bjv1cvY+f)xR;$m_D~ZeZCXgH6U6l;6)@9NcPr7|BjnJJe;YLDJ4{
z8FCdI!1fKzHDkY}niXh#LUVgl=@iAwh&~43b82aNl>KMOOWe$}wug+f+t&6PZ$gz`
z_NBG`n*Ar)xwTuhp5Z>$u&#qVof}#AdrOv=i9t!o@*F*jbIt*%y4iS=o=0}Ew0MEM
zh%#P7JSaZ_7TSUhD8&j^ZIKM6ax#tKh{5%wsHO2F6yYq$XfJ47-x~re(&yNB{XXK)
z{mTE2x$hwmeX10jJemwAmeUi!Q#dpQo-_0jq`r>u4TNtZoMP`zdC9k#bzAogFXU%X
z>OtT!;>q$VvOh=YLEt<44pOfnd|dFf{DaZOwl18e&$CCj^^hogZCgkC`#5Mn5KhFS
z(Di9~7gACg8j%+}P#R{9cQlh9v5j~5vuE&SJlg1Wt$)Qkihd8_C4_|#Xr1;?QO1*k
z?+R}n`Z5A0E`1i^6$ETIwNQ;9^)m!aHSH*@L&_n*M;iSFk{*PQIOh``p+{w&t$4Qj
z%6YKWvSN(*mGC~3w>J;)on1!tClRh7yoqoW;VS?NCT`)#)PA5@d13n&BInBPyv<nk
zGz|O7uFJUuONH#4xO@0Io<~bO5f3A^n1gyP2YElsU&nI|=iqzbEiZ=#$%{GNZZlbi
z<)pm>>sNF1dZ2uUz*RF!M1TNvb@1<IYr+Fr7fLG-mLM1r$`FPTxTCQVr4oQ@m^ql1
z6?!Mqx;UggcfX9|y#G!hbr*sSp&H?{2zY%^et|uQl#YM_{Y)yA7^Eho?V*s8h7a0E
zD4b5y_)G>~6AIBNDsMs<Kv-RbQ6QA}a-Nwkk~JG-l3%~|n!t(aNIn{ypWdsn*92iq
zyj6sst)e7;fZXlCp(e<7i+78E;mik)PD53GqjXKc-(7q1v3>Kq_Hz#O{Nf~<ki>0h
z!Y|$d9O!{4B4$d!1Tx#15?Sx!YsF>y`n0q4$GGIP2>)ikjI7+F>hWuy_s;ZhAXZJ`
zIGiToS?+YTyufv*56*$3F{jk3a5Ovzr-sr!dQU{2!>@jR{;G8g1F=*JyY2!{E7<O+
zH*3SU@~~j%%7dYn@sBd4dl<h|wd&)Ahc%~G76un@c`d$p&M(ZAlbKnCI?-Z1LIVz#
zQh6?=bmLbw=95|tE1A*Otn@A{3lmH&g)3!qDi)I|HACdJJm+a<MRq0H<Z-?+_G-xx
ztS*y~*HI7X!JnYKz;?+V$7P}Rw?f%FLfvIy>1AQnWnsme!e?F=KEs%tt?(O(^G^aI
HPow_=;crK{

delta 3811
zcmZ`+Yj9gf72eh3I+FahWy`iK$(HR%#8K@=P9Kh6X<TDDHmR=-iLF?crE6KPEV+B-
zkdPL8LV*O@Lb8<U04)vhr_h$BDo`jb&7=I`F$u|(%xzO(XqW~V2n0GEpujnMt=Q?b
zccL$6&pCT`&z{GwpF2;F|C|<ID=M-Q_+EVW!DQp;(P9T>M>kHF(V=2iEECy0R1zv>
zr5<whk-$`_hSh{>S*?eF#5z<Lax!PAp4Eq3%q4iXA$Q2bJfQ~GAjXA5jUg}dhMHJY
zsF^hjUeQoXXd~Mw=;9$?$j|&95~ibL{N=LB+3=<girXQCwaO*o&2nkDARI8mSS|~D
z<??WoToJC4E5nvNKWqzc4qJl;)LF$s*<MTLP3$HIX(V!0Es?7`3}_Rqwk4?F!Jnps
z#|$5)Mm}iRmV3<bs?p&Dp=L`iL=%b{NhQ@RbD_GMKOpU+Mci1>U`G{JkD!Z?^6=XV
zzHjaTvNrdRf-zbdhV4I0!h}(oFoR5)QTFh##Y;VW#o`8h+|o=N_)jh4w2?clo0dJY
zBWwr93c^<5rZrq1l;j$8V4kunxi(xHG{|+rq3YF<Q;Zz4Rmkeas7AJjt-z@bnq-&Y
z)X8q<Oc?xLt!zL^D=eB#X4104_9bJ=l^w9<5zQDkMKj>Uf30(!w+Vj`Zb#?@@S8Q0
zs-$Mw`^Z$>tp`6A<i~A(+RuMxYbqA!C`DCO$*Qa&_pYtd*w}@psAh^svr)f6GX&Z+
zv#x<r5eBxD4AL=vUrCE3mfFsrC~<de1S#_&cmTAbgzkkB$twG^tO*6p0R95jjkJi*
zcK$Ay3i5%{KDsk^q%>?87JM%%n%Ipd*-o7+rpst!3_vT*DzUkAG8Rom7FlLtF{_$@
zz3S@C6(6A!xkDBIqMP^Nj0uEEoMw$NC7M;@k(mQ*FEIQ@CNv_bv6nw#Z_=Hc=I8Cc
zsZudzArn_pk@(WWA}hlw7TuK_Ewiw&fG~{^1<;Ie8ma{-+?V)R5QQ-mHt>TEk3o&)
z4pyC@p&4NL&3X{0QCyr*vXRx(gq&cw7p9<=QD1)!8M6SY5df^~w&(diM>Czzea&H`
zcj#{6q;;mn@~jHDTEXb}$jIQxK;GtcC)RB8Kh_M;d-%<@sqPpql>a0kJ_WyPUxoyk
zB3|NwnvzM{2tSjHOs=*FXbQ$QGEPU0#~dzC+B9y!WWg^A7IrtKR*RmHF)}LghMEdK
z3qOy&lE2|9;rDuMwQ1w-U<#KL-#7_UnL_bwNSnE3m&Z45ZY1NTd`rsz?;(>F`6*%t
zCO+lz*H`9w1tj_?oEjN7&OAbh>&DfeYOq7!`K_z2NSbuyXN=bJb8<zAlq$}1*6-Lb
z>P9io6j}INu5!NOF1TSaFauli<fo2$Ic=<<2L8Oe*<c#U{nhO>N_w^MGH*-0sIY0!
zX7dOlSkj$+gFE|oumwKmZSrB{H7P!WIU*8bHDoIy4K#@@rJ<i_xAG~cuPm@Fm4Q-G
zy8}9VCHK1bJ9JqjT{*Mk5|~;{<jdOHOUTP2U`=W^y?7$osCu`akNZ$|eTVGhI9x#3
zk0AUx093yE??Eo+m?pulBfPZ5UVRXG=!{wbFfFPC+f|>lzXz9A{g6rTTp=suVwwH0
zbg|BTc7x-psmW6Qy4!!wz4L5q_pe(A&$SMoZymy6q4SNBs*8b+3#FU*>6U*lTj-I#
z;_Ys7vYKwMH=T6!nP70rN&7AIR6Rx7UDPibPi^+?a2rnB=^YN~bXA`L24@^}N1b%0
z#*DO!g3=kcyPulRbQ^$vg;1bhp=P8dBbd{yYvmftr**E&_pR~|xR&0&{p&`JM)+4c
z@1)20?>Z}?Kfm4CCh1k1Yu&b&_Uk!;XM~5LRiu*XyP<u>=AtYb%PK6AjO$Gmqo$dW
zuVNR@uBotn{FhzeJ^w?OXL%Ev>PHCS1e@OD_1p9qvav!m^WbpbKtF@Z&~F-T16rI-
zrWDqKJkg@>L(TGtnp{{)K~Kd#oM4KoGQ4Wn-@m8_tuV{(Cm#aIr&0VFgwG-rAz*^9
zrH^KWyPbcxyFr>p{(k;y_b%xOl0M$r(<+rCS;41zoU3QWS9%t`cpGX~RY}Jqi*uQ5
zh83dP!vOvoy=5Na6}^qLo^S1KqXRtAyG3e4m3IDkZ-;LGQT|%*E-8rICEniWqmSh#
z`hIU468UddQ;M?4eu&Z^0cggBWSX7C;abgz%>N$pL~)4tJqttVz~Tx!#19WQ*xP}#
z{&q(FC;3ame%hIPYxuG?co;aJu)rOvuyiza6Z;%$d>-Kega;8GLihs07ZHwfx~GwT
zi8t*D%!nfVCUV!_LE?@*g}k321Q5jLzlziug!>WR6QApM$lB!@&9Z0sb9>t848Od`
zN8jYd6Lq$=OpWm1gqJ?WcTa?te}EQ6=tRzlmjYfR>>CKrBV0dE{ZaD*@<fEi{;`X*
zA0uF9vTq~2h=A9(e(u|mdI{la1S}N2r&STS`Ps`je!pY*K$lH8V;TWFpub2&PAkku
zL$usC`N2u=R^i#NP!nzw!p<XHKsbc(NdOh|TkmSC2+i%n<S_kct}%R<G=Ch#{=#($
zG!G2Tnp7k3>QLaR!q$#U?CTkj>;3gT31-q0OuP%k@*>G_4lWF<UYhBoCvr70E8UE>
zqOa*e>UM-K0Ka+UiWLV2fRPcu0e<y-SA6SoEpqYdV|54;f(@YuK^(jpsk;C)^UQ&)
zqOy-6ZA_-KdN^Vz7w;JKk!27{5LOWIsM(VU?*M3~nM@|d{)41F5>eZ;@HCD^qS-7<
z&MblPNQBLxavwqv;g&q?0788z*Q>PBjXrbv*`8fj3I0DBI~z}&o17D)tAs4k0mDC0
za&?Sfo9%H5Q)tECu6MW-Xi=o+-}p%4mM->ZU}y!=Bs99{f^c6eS{((l2NuC)>^WL_
zG#*`q%2fmX`(w%?zUf&J|3#vn7V|$P8kdFB;vmXUy->U`)j%J<$o140Szn35vQ^>0
z^*;~txxH?tl3rR+SvBZ02SN?bv@m5cqXy#e*8>JmtH|$aZ$RA=SsjRIY8L3uJ2Q!d
z!r1?CeiMR-sgVNJfkSb;Hy*jRVH@mjDW!C?O3;IUM16t(WzJ)NjTBxc)tAVoOXSu|
Zr2RJ}c8<h&OLDp7UNf!vJ3$no`#*=mTmt|A

diff --git a/api/service/generation_service.py b/api/service/generation_service.py
index 8cf3b49..74a388a 100644
--- a/api/service/generation_service.py
+++ b/api/service/generation_service.py
@@ -5,13 +5,14 @@ import base64
 from datetime import datetime, UTC
 from typing import List, Optional, Tuple, Any, Dict
 from io import BytesIO
+from uuid import uuid4
 import httpx
 
 from aiogram import Bot
 from aiogram.types import BufferedInputFile
 from adapters.Exception import GoogleGenerationException
 from adapters.google_adapter import GoogleAdapter
-from api.models.GenerationRequest import GenerationRequest, GenerationResponse, GenerationsResponse
+from api.models.GenerationRequest import GenerationRequest, GenerationResponse, GenerationsResponse, GenerationGroupResponse
 # Импортируйте ваши модели DAO, Asset, Generation корректно
 from models.Asset import Asset, AssetType, AssetContentType
 from models.Generation import Generation, GenerationStatus
@@ -113,14 +114,28 @@ class GenerationService:
     async def get_running_generations(self, user_id: Optional[str] = None, project_id: Optional[str] = None) -> List[Generation]:
         return await self.dao.generations.get_generations(status=GenerationStatus.RUNNING, created_by=user_id, project_id=project_id)
 
-    async def create_generation_task(self, generation_request: GenerationRequest, user_id: Optional[str] = None) -> GenerationResponse:
+    async def create_generation_task(self, generation_request: GenerationRequest, user_id: Optional[str] = None, generation_group_id: Optional[str] = None) -> GenerationGroupResponse:
+        count = generation_request.count
+        
+        if generation_group_id is None:
+            generation_group_id = str(uuid4())
+        
+        results = []
+        for _ in range(count):
+            gen_response = await self._create_single_generation(generation_request, user_id, generation_group_id)
+            results.append(gen_response)
+        return GenerationGroupResponse(generation_group_id=generation_group_id, generations=results)
+
+    async def _create_single_generation(self, generation_request: GenerationRequest, user_id: Optional[str] = None, generation_group_id: Optional[str] = None) -> GenerationResponse:
         gen_id = None
         generation_model = None
 
         try:
-            generation_model = Generation(**generation_request.model_dump())
+            generation_model = Generation(**generation_request.model_dump(exclude={'count'}))
             if user_id:
                 generation_model.created_by = user_id
+            if generation_group_id:
+                generation_model.generation_group_id = generation_group_id
             
             gen_id = await self.dao.generations.create_generation(generation_model)
             generation_model.id = gen_id
diff --git a/models/Generation.py b/models/Generation.py
index 6c74100..9a75c84 100644
--- a/models/Generation.py
+++ b/models/Generation.py
@@ -35,6 +35,7 @@ class Generation(BaseModel):
     output_token_usage: Optional[int] = None
     is_deleted: bool = False
     album_id: Optional[str] = None  
+    generation_group_id: Optional[str] = None
     created_by: Optional[str] = None  # Stores User ID (Telegram ID or Web User ObjectId)
     project_id: Optional[str] = None
     created_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
diff --git a/models/__pycache__/Generation.cpython-313.pyc b/models/__pycache__/Generation.cpython-313.pyc
index b052293dfd5f8e05d03ae7ee0feab8dd2fb2bb8d..cb9ecdb2d77fb9c455b1c77b3cb8f4ea7aa01bbf 100644
GIT binary patch
delta 306
zcmX>i*(k;PnU|M~0SLG+^<~;_<ekXGsIz%K(^<xpV5TC~7}gjzR-j}Id$623h!#;|
z2$q*(2<Ee7EK-Y62Z}4e#TB9A8enlHu()Qha*T`#Q?QB@Ll%oE$Q~dl(h62B(oSd6
zRNK6VC74lAI6XBlwJ5P9Ge0jry(qu5U@|w`T1L&uJK4%LY(eVnK!hlWkOL9&Ai@De
zI06YR5W{J61p6FDMxDuDIkJIdBIhY7Ymm4zh+qa0t{?*B(<0x=4qR%CRh!efq!{@$
vbuX}p&yZN4JX7{Ei|*tG?#qn+lS6pa8KWlG@L2LIF$zzR{K^2Nz*YkQZu&u9

delta 248
zcmZpaIwHyYnU|M~0SGQM^k(vJ<ekXGD6x4y(^<xV7!f6gU^yv<U_ML6BGnkR81`U!
zb)Z79xB^sM9W1U07S{+?ijgs43Rae4$YL=CnF$0%n!zeXTIo!hs+%vc1T#*yU|Y$k
zG5HZ&xrPl$))qvFf(SVfArB(#L4*U4&;&6YH#f4+VPw>vEXSD*B)d3Ii8_JAnL&gL
yhyXdi$Y*jMml|XC=BZp#jFb0qUt;u|JdsD8F>>-w9!q{jM&SvPUm1WD*ct!?r!$BE

diff --git a/repos/__pycache__/generation_repo.cpython-313.pyc b/repos/__pycache__/generation_repo.cpython-313.pyc
index 7ef0c95795a538ea8205ef9fb45e67414a306b5c..34fb78ee00d29c6f06f9e6a6a106d024003dfc50 100644
GIT binary patch
delta 739
zcmZ9I&ref95XWcsz1Nl>ZLx0~ORG(*M*27?qP18^kOmAvY$KZ&qcj$Ol%^gEZi~j4
zQ1zr26Wtj94GkBL5dVQz4kb+pUN~@T#H8`&uI7cr@8L7w&Cblu?2qC0i0{bj?FA|8
z`}=al|J4^nnm~Y0kOz(oIAO>&HYid!iDB^^Z(>Az!_{cs&5XP^?=ocV6`h?7!xNnc
zScsdY9_!S`HIf+UZ+l97Wo@-ke*B1t6K+akC?dw_u#F5Exj>TgA*BV?<UcWu=}96M
zy%7=hgzMxYR5Y8IFq;Ob=!-p0x~(ttIsP~@!|ClrTW78znP6)R-s3#SE&3hD+#_J?
z(|8q~${5_5r~~vtOctL54`gRk+A><6><x>w%~G*a<PU1)l}f?cmS2|Lcv9j}ZOG3n
zsK<Fi!t(Hd@+pNe1-TONsKT(qxPn^HGPtQcg9*vFvI7c(5>@qQF(n?9t{o~BYlY)k
zs)dIy{>{@K60iMd?7y?!R)rrW4&fE-($MGJhg^#$nl!N$YTUiwv<f?PsYQ!TT5K%y
zMs00}K5NnSCSC6UTqo1y2p~8^x&Zze@>bgw_1>A&ed_SLswRCig&7}iYnbWRxAm!>
zj87~D!v1^mDx0;8tdd2=yTD7r?+7%<aDF3gUc%@vK%<5om&=HJsdhky&Dz#nb28Ue
tUOc6n$&zfMezSU3qN5?=KEw|oz6XPQFnR#v`;hLs<`E-5L6L`K-XDeBl@b5|

delta 151
zcmca*@IZ_2GcPX}0}yn7?#ayI-N+}(C?defz%Z2|hM|Ztn4yR%n6ZdCSbVZMqdlX<
z<Z?!DM#;&C84K8ogwi=Rr8WmLX$VYq5S8Q)0h;0Gr^)IE0-C~;GeoN=zYsNLG}$aC
tX2{6)oq>_HgmH4N_ztc_ph}<(48^LG^(ABkT^Jcl7(X$9sUiuWH~`MqA|(I-

diff --git a/repos/generation_repo.py b/repos/generation_repo.py
index c668e97..b561548 100644
--- a/repos/generation_repo.py
+++ b/repos/generation_repo.py
@@ -77,3 +77,11 @@ class GenerationRepo:
 
     async def update_generation(self, generation: Generation, ):
         res = await self.collection.update_one({"_id": ObjectId(generation.id)}, {"$set": generation.model_dump()})
+
+    async def get_generations_by_group(self, group_id: str) -> List[Generation]:
+        res = await self.collection.find({"generation_group_id": group_id, "is_deleted": False}).sort("created_at", 1).to_list(None)
+        generations: List[Generation] = []
+        for generation in res:
+            generation["id"] = str(generation.pop("_id"))
+            generations.append(Generation(**generation))
+        return generations