From 8291a3b662813018ede3ad139c5c3b27fcc01566 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Kr=C3=BCger?= <valknar@pivoine.art>
Date: Wed, 26 Nov 2025 15:01:27 +0100
Subject: [PATCH] feat: rewrite CivitAI and HuggingFace download scripts with
 curl
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Complete rewrite of both model download scripts with:
- Beautiful colorful CLI output with progress indicators
- Pure bash/curl downloads (no Python dependencies for downloading)
- yq-based YAML parsing (consistent with arty.sh)
- Three commands: download, link, verify
- Filtering by --category and --repo-id (comma-separated)
- --dry-run mode for previewing operations
- Respects format field for file extensions (.safetensors, .pt, etc.)
- Uses type field for output subdirectories (checkpoints, embeddings, loras)

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 artifact_civitai_download.sh     |  Bin 22536 -> 17843 bytes
 artifact_huggingface_download.sh | 1924 ++++++++----------------------
 2 files changed, 513 insertions(+), 1411 deletions(-)

diff --git a/artifact_civitai_download.sh b/artifact_civitai_download.sh
index 7fea697facfc91a998730094707852da4616d053..14d23bfbd804478bd70854dee03b42771f7b2a2d 100755
GIT binary patch
literal 17843
zcmdU1UvC>pa)0Ni=pE4(wd)Zjdy@_1)!}H0k_k(qKuLBC+eeJZp)@CQhMpNp=E?-_
zAxQF;T(Xw{hXBcA9`XSa<Sid%zd-U4@~i6Z>7F5}KehxUI^dRUc2!q(RoCBcFMU%z
z3Bu}$mz<TBN=u?1Tm)&YDfXhCKM-5dWjKhuo*xTW)Wi$l8>c~kJP`HWrZ|eC0U-38
zL;!3^V2S9*(J({hg+0-YeJ}MBkxYhzAUsb*8i{C}j>f6z1+m{vqd4#rrr<d7PW=tB
zTJ>VA<#p3emuT{WPES#@nw*Ju_14~AZGY>7c-J~OYPR+dKS=tc;SdWZ8=y=G4S|mt
zotj2a(Dk}!en!AVRC<2j8xPVI1|~uYkpAjSzagI)l;E?ROc?yakAwb1g1?rZGwy@1
z$UR^H)f%Ni6e{A~?i1m<-6-q_r((Og+mN`38Xg<R9?ceSYkRw*9}Ij^iT#gbY-dk0
z0AOO!#SV+DW;+81f;{6)uj@*I%ua8ypsGQbq~2g47u8aeywXake;UOTQLi1XKf(V!
zmCJ1hgB03xGQno68I>-{$W8o_2M!Feu|JC3peLGJ>ra|n3LaPt3YyWuj71oroo8+u
zo%^9UYQ1XgYmT*2*AALcJp3#f4rzaT@x+bCp`ObBJv@sp#pRioipyv`=!p|3Nfi1L
z>WrYy{J|&(f-ek3KR}eF#7~9mk0UV(Mo_9?Py+PkJ^le*y|vqFi-X#3<LIb?18R=v
zksl9((8D>622qT2b-zoO+Kt1;(dP2|wP(-Xtqqrnx9uqQPvdBe!>$%>H@5VY=g-zL
zYp304?3<yd7<${--EF-wW6v=5V)wXVM!v_$UTvqbe^fJre}cjK+nR;!PcZ^2UPOal
z=|yXIt7BGQm#g!rO#;S=1Rl@PfghL-LSqqwVIqd-=PAN3L=pdDk%eI*i-+I5Y#wQ@
zF#ZF^iCnLHE;p_8VyBZw`aPk<oWpc7fN9)n?kWC>XpG;ZE^Pwc@qV-3+G>dU%UZiu
zKWelO?|Y3>{bi&63S72dZ92dF)89L#db@Ram<{}+Q);*Bub^1Y|NQeGf9I68T5tAu
zTeU6B{<pvS?|=Urr?lJLf7Jn<`!wZe0KBQS_nZ4WnDNJd;Tg^S?H0cM#eeX(!vid~
z+c>24{?#ef+U?exPP@7D5^MbOmw)F#Hh<pPRo{QclJtG6b^KztfeC;3Tjv2b8q{w$
zcaGaN@>=)5$Bey0?qbuy4ofe&>1a=gU2VVZw2${U``#e&OR&_9ofd`w0ihF2yK&I!
zG`Fk~P`M>NW9BU3tzmI`ehS|$+T-vb>WL^8gQ)8bN<m+|dneqUaH^M4e4dQFuJ4Es
zAD)Z!491rXXT1h(=)fy&I_0bE%Z6(Jsxp?SD(^pa7_N1EbZ~s60nN7@WIY=8C&x{N
z0g#AE{A-sjU%hPYHKx^7_L#R`Q-jSc*bhi^Y`=5vPh?o+hZjK{g+o6~OZDdK=25NL
zfn(~tYP{vW$PaC}ARnlX;93!BP%tLZtO5^v{iH;L9enMuZ>y|`E9O;6Arwxzuuef1
zB#+on$8jju7?P-`L6)W<%oJWEn-w0VUm!HKVfa-_CO3Q#N56$;BvrW9QPisvwbnFC
z%8P&T0t;+KQq+fzX?g^rDA1w5ChvazVd^}Gr_QKUJJ828J!=l@M7ZQs#s>6V|MAFX
zxpIooLpZ-F?Bk}B!~FDD)t-M*4ab9l_$<1JZQb7T=CW}6PhAY82%>vp*;&p$JIjvv
zOh1hMm5gXw+7ca8uz+fsIzv)fLget##GHg_fc_w9`62trfzI>+?A~r`X9uQQY#;B}
zIbgqc+c%1XFzuXpVd%%%DNvI2KczTb>rw~h*k#a5&o+OMDsQHB27b6%zFJBFGUeTo
zzm2?JF9=Ve3>8sfSQtQ>L;x$IDxQiJrKNuNEE2G!w6*-%^+{9h$YDxfwXd0$D@=P|
zEI(RXP2hNzMY)prAH}*T^HM7-`T18Ifhn7~Xo+9`@OumKw+WCcCdBeGBSe5h@d}7x
zZi}-k8J`-?;&{bEq9T_mt#~Wa^4xqq_t<RctW<ymh6j1kwHlCZ+54~{m;d@tp?6~u
zCz@MpyGUw3|ISG5hc%Lz7Lp-t-0k{Fl985?p@`t{6w)w}(o~e!=OFUqI5Ik3M1$Cu
zkYv$S4vF~H<ELh`F1<K}RxhMT2FAdqBCQ-MgC+?3(Lx+b#V9me0G_1&C<B+rtvilm
zxX9w6O(W!>7<_7<WKv#?J)3{xD1@KBD_<#$(<<akSf(HJ#zNwmOs{m*37)0=v_4oW
zhPVeQT}E-X{{nLnr~}mCvx3eU?h_=MolzXYy;xdt_YIJy?0egkjr_Qa{X;E<ESBP3
zd@I)1)?ll617NPmLdoUe;Rk*%KUZ40d9(fDD4oz`Z3rc}Hh{{SIqzfIk(}K_WWd~R
z;0L+C@1)o?4CZI45H`|ePt!-`E4j=x(XjBJ<X4yJf4}_0UoSI51{28wP)zM2px@_#
zK4^lK=0AGYdju4jDiF%bqh9q<&m3?VceC9T6R7Q`kG#{YI`^Kkap4CU&LO$g$~eI}
z7L$)u`?=TLu!I^?_q;RtnCUyhlWzo9aggo17g$t0HnEQt<&%0zqs#_NIG7nEB8*Z5
zZ(L3Cd#mDL;CqR$M+jR#h?8_x>V?5UMn*VI3AiE7(sYz;RI8^!dNw{;1wGYaaE?4B
z_Rgx4j|STR6r^IERRi%3$2L1C&#+{`#yCtx)K_J}(y5<zIP>W6FB#A3P;%wX7$wl^
zU!cyiyb47`9B`VREtC9tnj>-%;b$2OIBQB9hz4FspwJYAJ^z!nct7w5y&D#nBr#;s
zA~Lu!iE<_g9m_O`KjHXl@}1RpWybSiwalQRS{3h|_YN!bEHlz@EQz*1j4pifaU4-`
z7iuwrA)~~GgvM6UDf3!%P7@!|Z&Lk@Q>{Kvs!q9DU6#VAlYSAj+>N1^LgP?G=F(wO
zQr{^DK`GHVPsx<32^x_oL8)q^+!&?@`4}&h6(S`<{>=)w3mNg8&|9v+WT`Bg*rmVb
z_TlF8lV!x(NYmWdqoQXbe*E!A=tL1OuS-Cb(_o%TX9!|8lc9O|1tL?siZ8UMpS<`q
zSrK9hRxlW%W*I;blx}C(2~u(gwN#tpz(A0e5{_I}PLuVY<d1Tb94`9#1pJ-nMy))8
z69|n&ugu7JSuQ)wUsYu&@}v?5g%$td#k2r*8F;?@^5u&;4BkO}w33!z0dK}WiI3(f
z`7y|R0n$%Kg7od~A|xo3z?B3gj{ip@%#zTD$S`%}_JAp-E~7~oQLxWSHbql}Mo@x$
z1X7d6h%{U3AXEnj1EVT(m7_gB>Ba$fg7DM#DF0XxsKbQb(8mu~YPsnu!Xieb(m{zu
zHgh^74>>L-XC*cYe{paa3ze1#w{+1ms1;5dDwYK3$W16lv*Itaq>hKI$gQKn(d#01
z(Z(Tg!@Qy{zt%PrP7`x1Q1y_CJJwdvK^Bi67iirwdqqgB`E!?(oh+KEauREovUp!@
zincZ|VcviY`a~d*NYaG2OgRiHEm=6`SR2W2A`?I1)D$YMNPdr*Hv%Df&|h{^=!>y{
zTdG|}uF@jJDXX$B8wN7kvnEzjB7-Q*c2NZ05O27_rK!Qj<m8<xN?zdC(5D)uvNdX&
zgak1>r$(fUIvRYTs{Uqsy`_4);u6suYG?7l(t3Q`bac3<>cJfEd844JY1j4Q8+A@E
zMd7mDs9s$~#A&Tc3nB<q`<#7)1t;=^yDQvF;XJy^ewjMrJ;M{X?n~jIPn#kda*rk>
zbhN$EXb`~X;&4=dn?zx8b`9N2w84Mk85`8ii_tlSs^%-g`Bhw<J-R1eY%h;UIj@<9
zG^Y&uDL8$+zM|`oc@GQi*4rvrb^V^-v((L<qxW<;>D(s8g=gH8G*C(KlZh<o3pTbl
z7VcRQUX@b&>!wCpI8<7q2rzF_U{NVw(PGmTlON=cTGlEcA<A`B?y`-PA1j~=pY4rh
zN=4=6(a`}NW}^@G8bg$Ui<}kGgIuhY3x;t1AHrc$JM96TC0(Gni=&ijSBk>|kCiL%
z*pAXVq1rs%><yD>jB?10-jHN2wKGLvjZ6?0i;$Z>6S+@AXU^j?yV~Cx_?sQ>ArohL
zC5u|iD;b5<uJ(wDv;Z>_EHK_L%b7{!v%c&bsi4-{3%!{t8VW~JPMMm|NqwLC|JOUT
zEDu+7J&Z%+yE`}90<?f}UM+D}7Z?TOf0cie;?26D6K2M8a7O%@hf}#S#}MRt)MX%s
zRMEsP3`8cvIlCV#BVy(-v_DYr)MYTgEs6j#>1~tPWK1I}RFm;EQ7d<oVg7a$(=-UB
zC&;OjO&!lF*zsK$3>gqzAB8|@*-Ze{ROd)AGwBauHeWZ|&F!}jXgCox<LcP|5!1<`
z+#M`urhwNP&N*3#G`w;-)CSXWLKs*RW*Qp#cvM32!nX%x1=MztSLG2aus58~Ig2+3
zF*f*|7-lY`a~eRJ)nepg*(5G&eStl>rP*MwNWQ$Vxz`YD>}W2YT7+q)%Vd~7iRr?g
z6-j|_LZjKw^<HvjL+8gDm!!#>ukJyTt+R!poO*5|S*8u@Wmrl!74!VZLl~UBT615#
zX|-Q%<Feenm-mrDP*YNxn__tt_ivG};2@%PZkue`V-*hCIRs1&ZBO3)MSv}VD8yzE
z6;V;ojlXl@9npTF)DGt*??&<MB}etrrH<-83KuTd<{U@qNix^AAE|mis;hJ*5m!ZI
zO<rA^HTs!(3R|fp#cE^F6|);_Sw55-r7NDriRZy+#A8xnQ75cag3Yz(BKVQF03W{l
zZe=BRWks&6R9BeliDZW^zgjg8bEnqCFjP)PQHhr%mK3VocFPife!H>z0jm9KI(DCc
z>Cez`TO0<$LZfTXw?(4oTPO^Zw?#s^goQw*5w_-CvlVi-1#a#9Z3(bIfGFG3B~e2$
z02jhki$m@-7wW$eTr2kz-S2>4v{3Ld+m=F?jW8ri;n0><AHQ2yJzl*44nZ1(R!Lu*
zQJP@%9T>7NIUKV4U`~$;;LyGYZ5ar4jVcw^aaU2YvO<H>o15OaVyu|05Hb27B)E^S
zP8tn&sZ;2brG;^DuCqO*Uv*+`w3kNdPyZsV957gWYUwL}P-4d?TX<9_L2rwQG5raN
zT8QV*b5xLzn5BR{2di*?b)(g^LwfZ)EC=!tEzT||$LBFTcgY+J#F@pFpLkuPz0xFO
z2WdNS%l)HRo6R){gE5t)l2ihvI#OPt^vz6~Pk+jg0ziJs%~rmjTTD!^t&fT<rEtPM
zP3O@iHC9$Z59NV@TUS6+X=W>Esa53J#PbBH?P}>9_!#dPtSY&VhcZs5W(RZ-@&t=n
zA+6+bs<_sZxIxT!L+Cj{)@8uiM=wz<H(J}c{(i@YIwke4=FYEbO1Rvrp45o!7?GrO
z)lGZaYai6$K;u*p82#lFM)h8ZV=yYO$%1+=xt6*tfnUqVH1a6PkQ6UM_(cmnHPVl4
zU~aSLG$_Y@&0`vX-})&H&6Nz?^&yR1<tv`i5CSr5U;2oK;BWneMpO{b&?}chd77^0
ziVH6$d*t5rg$-B#v3u}==M++Q&%E#yj~{^Q`e!%k8LpM^qnsh4QCxZ|l^W)rwRr8t
zfp>x{%o{*eBv@({tED<q=A95`>sE!CrarZxlyT6*+aPr}Uh1IT^e}b9BJ>RKjMA4H
zpS<Bn(QKaCp&2#MCr-XX<LepV>xLI|+5_S!D0uy_Zgt`f@ji)e5oBjN3K5?0#h+1x
zTeT#DufxwrNGZ3CmzSNf2hc4|FG}uw)Neroo0#j^Q>Mdg;mvF*%X1x8UBZOv!u^+v
zz`qm`?t^h3=N43JMi&SM8>il=L8n{W)Cd@Qfw|tQOBje8aCHg{AxNTv04{7vm%Su*
z>D2&V3!iyAK-2T@2Q<)Dkc<Z2M9BNpj8{58Ie=K@N<P<_X73#KC<DxNh?`gX1rCyG
zMuGQC^teuOBviu2Kj}>G=1yjonk!#AQ$B!mk_}F=M{k;3fqYQe5||hqAiU@s;#n)4
zvA)-;eee4)^Y@D2*WMuL;pR*}gS0k8T{z}`AKq$aDW=7h`xExd&TeM|Cb_w)*M!cY
z!6fVJ`fcZ`I7SHUcB}ersENk&7u~GoRc-(42%rkFF7Vr!cpcKCIm=6?I3P;U7?f;T
z4GRm8l>`v7A|&~D8&I3&>%m(5bk%}07WsNW74#4!hifKhQBSHYaNYoY49R!!TOZ4W
zS&|?v;D^id29G}bb1EA0I7r3GcYx<iHan&jC=bHj8F*`>D+WTv!f@pmfk*CJKTjez
z&Tf-lcG;o9Vk~s0ae{W1P}^W`Y1(c3apK2Suge!K^SCS(A_7&=ZD`FjiSWPj`G$(T
za9!}~8*3YXHbqBYA8Wac&UTd*Qt3C{kHg-ul=RJ$;%z;gS9AMB?pR&jrc+sbrYii@
z6Y5bwVJQspqZcY%E1R=U8zCKT(=1vhkHA_*spWyJ(7}$=1x#GbQ5SdlXO*(6+8^5}
zsh!$dZ}T%^9aItFcFI+B2${9V8(@lEU^L5D+2F(6g08i9j`#397!RTOCiK16M{%hV
zeKN?ayksSn>X{kWBg*S`A;Rr`)}}e1lv~CQ1y$AHrY)tv<Sgi?;wei-j^r|MD*-7W
z)rRm3P?|g?kJA{*FF>j*RC$zz!|(-oRc4b%n^*oXz-scHJmP{mxjW_%N}!^gJZ2T%
zZ02rzWE!wY{@PJ|rWp6L3`+BOv!}lLm7CTg)~rQ%-80ZhiBld~R(#08v2~WUaE;t5
z61m^zapVG;<M2Gh6)yzF)K7q|R8R3V9%)rLvXTF|0)?|<%H(jgWKj%o*E$PAWsy^$
zW=R~-twtchb$#5M7}En>dcjBkvWM^z72R>Hm1ElW!{%}I0v}8?<d1oQPs^jx+=a5?
Uikqx+;hH`EeF}>CIq)z2FK<-g`v3p{

literal 22536
zcmcg!&2k$_a<)&7z{Lk2e649w18{~QNJ+Ea<qXz_NRWaD{0Wd8twNF!O`t*c3P5AJ
z8=|lj96tJLKY@>a2m9ie@!fCWFEi_}n;=Kha+eb`gRZX1%D>9W%*yIVUo|e`Y2zZy
zt}2fzkJN7bKF(Vmb(jpJvD!<nr{g3XMyaZ+mU<qA^E@8S$7;9RQD;dq#t73VQ%9%!
zFBQg4Xl$lNX)>|!!s$>A(kRTMOl6D7IG(=CRGz5aWHMTuchoRWqd}geag;G0=UI3e
z?WoO0m}16okoO0KSs3?+8mG<dO1<7aK0It4?Y&j6lUW`o)9kHKkxV9-G22n5_lBY=
zH`GOvU#Z$K8in(5zK+h7EXq|qnkQ-&&(Iu?D<G=*fPci*<L+@!owT~`v$Hl>=@@wy
zrIUCXf}_beNi$VDnWwXH)HnqB(>zpXSJ5N_o)3!=;PoO9Ug@<@+h@({o2{o$UvEvS
zMEZV`Mwe+ahrk4<%JXh(_eay|`=`{e*WPmt+h{oGwcAIo=?R)%wY%Npm#*z8+MajM
z+pgsqS`J$W?W41nYy1w4yRTXvuwS9&<%`alH}db$1YDmd<6-6bad)rp<G3w2a=Swm
zt%NAI`q-kKu;umw%6P#j3wVn+nh9_2Uql{n2zh?vQGjNmfSY|X&~C^G2k%NOKPXW2
zjClV1pzo-8#<U#teM+7&9S8ja70;Lk0%s|B#`qufi`YM7+z<L5=Fb@KgT9aTGmZ0U
zz8GVz?sX2e=xFlJS*zRG)eSW9`O#jx*X<k`(ecc9HtIxk_S!$Sd#9#{2LFKTuL(st
zKkDor@3qzLi&n3-dxrV23J=3~k&1J5l16#HK(7a#w6go6z564W-21T^{7q2V?H!+<
z>c_tgD!t>~AF=d;=D!4$6Ra;R13dh1P}w_vdDJ~_?KOkH`s<)_dV=xY_GvSy1(p3y
zk2*aDur7u-ga1R1eduSqhvxqYD$LoS3Gk<9Ewr5km6xsFQRnEO8T<$Oc8>Oso564J
ztJUirzwGxq2QPrbztgPe&%5nDH#dX-3ZSmP>v!e#U&zyocI&5C&EPNPNe1u~tsY?f
z@Sj9S=ZHx9Z%pve>?9+4wBI>6@3qc4$43th9`H^FX*|pAPPC|;?)Ew-XZ^hnIJs6E
z3>E&1eQFv`BK*90-a38JKRrJ0?Y3WUy<G<Df%^JuHM<_JgA={ue{Aob^?S$1V2D4=
z0mT5~(w^)^YLHAv@#Q=XX&+OgcpRx&m}6H~XX!#suwBL27}PAuviJf&#>s%YW|erP
zUcXlL5h3;B_^`cfoTSO`v7Kg(bUvLW!-nk0W((b1gp+Zg-oE`_<yX<PqVOlo?eBNG
z(6G+|Q;Ej@{668!&!6{~hnMDa?EHe+erYh~%A(us36y%IjuI^~5!J;)X{VwxiVI>5
z2Un3(Dwv&NltVVEH=mv)LzSdzo{2QaMv~C9#&wdu%VyyqDhUcZw+xJ?@8dL?PT(d8
ziY*u%t*ke@=3%GqAZc+lv?sL;c--;%*~z)l@*DS57^&Sf4FMx9XCA386mhIR6&<5n
zn*(CZC*xv<-9n_qZE^f6+AnMd<nQq^GmdTRq@&(N3-L}pq)8B)cxG{8(RHSA9*5)j
z50TpK{Dh6I)5p;MkL_0q`aVr^MgCLi6)l}o(@q`SDlcNYcGajbtzJogpp{&Lvm`wr
z17nwZajg*UB~>XwXs*>R(`cstsnkxBt%CJY4d!r-ht*~k#^p&}eN;K(fT5}eRr@=r
z2I?cdn8QnjEIx~Z)yC?NqvSrAe`<=nrS``VaSP0!5$`Z=JP_&@(idTm74x9p7+k=!
zPPcRRO6{K??UDyU9w4GCc!8JkK)ncOvuK*B7f~8L^xS}VlE%|qjpow<N2%&MMi_cw
z!)aK7NtoyH^s+Lep8bn(I*roWy1HQ-P4;4}^5{bj3$rbDTiUMUVSd&8^_JId77mAi
zgymjSHGU6D*KQus@FjjeSyv79M6It^cu+LBN)Q06K)~8ZI5-sf=pbQ9kH71S*SB1&
zF`6S)eZ0AqRbfE9mw{TVWzo-STdncz>+6NrYd0Kp>fPrHrc+fnfu>ca0Japo097Wn
zh~#~vR@3;$;`tzK2`bBq)`<*W16;#<mBO;~&GeT1d>4NSg%Bp)MWyA*N>OuUfGx>L
z`qx_|0wY+5@_aCevdp4v(bLFr^a^>oPzi`_uK*OKY2ubc84SW=0%8NK0uc4vub((3
zUWe%vi{nm=Bz6Wg9XAy~IWX~bl-vo!vhIfBtveuOc{H;SWGxToX$o(yyu|Y)565VH
z;;(yc3!biCv=r<wU$5P0h|30~O;DC^4R0ZTho%WRLAHp{XoshL@LG`@vlv?uDD*`V
z07D%-FSxu#J=wwpen^(7i!ik&v65XS*ZrA(WvBUk#)v`uu3!+ObU?|p%ZO~E=85`7
zZEtPC5;9|83@Hkd=@5I3qhYbH*wIzJqsc5^P-kOmWw<e~+E4~>QsPv>q#8!-;Dzx0
zwTRP<L4TX20fSuzY592VMy6S2W-R<L^3SUD&%aig90!LagfWck>LLGD9P-v6s~5!M
zr^CkpVAX>F)*cTVkB4sE!;zd>RtI>}tfrsLWmYFBK$s%pdDS}Xs#+FLD5~ZEQYgbA
z?RsTKUHTNKmQ;b7`V<_OR|$gQ=rTzcvQUKlMShh`pJK}k*6cXOOZN@b4?nbzPY9C=
zFi6r|Q9tuLTV$1r9jKZ}TxXkMdinnK_S=e?&u@P6wo=K{#SYKTTYLhiqGmyCsA^hW
zS0Qra(uZ_T;(makZkt&+iu#mtuZ{Hn43FB))l^l49porvRR_W}0f<1`yo~ai>$#zB
zmZIP^A!|h~upwgeW@{O2iVK2`Tc8UBj?wzac~pVA=ax!7!31yBb-jE%=q`@D2YfQm
zXY+go{9=^13O>?et!}s7g-0UkveSF!fi{I$(qnsekb$1<6<8y;)JE3pu!DHG-EF<P
z2PX0~5L7&gfqyg(7l8PkfygP5g3$B>PB%QUQS65oeNHIF(}L`n!+mcU%WxBsD|S;S
zdB<@BNaS3^;FDlGp!)pADAVieEo9!l)|f5cvaS;vL=AUTm`*@Akko<6Y|1N9FKQvp
zsygMQj_;#1gQL@rhfrvSB;8&6ia~;7d5CFUF<=;FobaP;TVWdS#fo7WG6Ma}i{cR7
zi0)fOQn1azs3DI~>~EVfVn0NZsW?io&KI+&IE-2dB_vJ;?`8=Cg{;InS=&{t=#Mws
z`1c9^eTsjd;otA@@2{3sv(kzA8y>GWgw^2s22xa(HycV5x1lW7rYqs2mp<i#oVdwQ
zMjzt5w!L17J`5tHRonbOVm8d5?y4<O8@cG-j`Gb=w7$WKm?lBWavVUW!Xh8CG2L)C
zksT(U0C5=HVKg1kIU=(OtJ5{v&4g#}YmZt5JQzk5$ynDJX&cf;@d+j|(yr4>f|fOF
z+BBP4Pk5tIPq{zr^;cNRd;qEH@6}B2LSO$@HHOjq#&kX&dx=+?!Uz6svli$ffsjlY
zF6w1w!#U;ZiQ?H)^*OVk{~c4L(a&=n$PSgJ40*_RaGx?auc4|wP=U8~2I|Rg@1jz8
zt{rd{oGjP_piEjk&ETBE=h`qo!RpP;P4R|^cF_FPyaPvRt5o%$86CqO=KBh?h?XeJ
zv(G@y4B<52XCi0E-bN3ZJA~oBuwVbfTSVlSx!Njb<C$bm1aRP^f<sZif_{_RSOE47
z4k4H&G@O>7+AeTNE4{>IDrk+z1$Zih^OcR_XsG?PpOF3>Vw=+&lRlcLyw}Zr@ip#R
zQ+hP5s$KtRLoX|wx4`2^ijiX;vtSrs!IMHPtV~%*L~Fr!Mz|+O|2AgPaCHK53ZKtA
z(<o&|i9yowD>*m0zbkbYq_j8+1jzva>G6n8&4H>9%?gBoctW8^s4o4F{z|f0F@XW@
z#aZ4w9vcXpP0u^ncH!BRi^5q`Hj4(a2%pB7RRyC{%6=!)@j_VxLA$=Gu~1ax!37rD
z{8@Ve%_nhitww;vo(jwu&T1ope7gr{?r$#(EniBJmoQr)V@JIaT!|Q+$KO;Z<0wQ>
z0{53rgyF$Z7E~k)3cYgz&nP+QqN@iOyh!G$^3ND{)K#9(vYm$UmN&ti29A8vMmQKi
zVRE--3E-!zh_2KzlgVnF!)nuFk&wK^p`ERr?X4vYs|n;}YERb1!z$KJVfe#}537z&
zQ*8)xujsTPpyDAiCLEKv12rPnWgaZV(t04y(%<AAf*Dnx6(+Zf@SgQpWDN8_94<a>
z>J>{-Q&fccY0#XkjyZU8jJR7^RQdGT>&!R8<*ACOx=^pgbl}Ra*`BW0QXzcF=_SH_
zc5n*?VijW$%prB3``FDzyfE@v%Yna(3d%V$&x^$?ohlPR+NVYHGreyJy^`fp!4W1R
zD!r7;+6~QKBC$B?Ckb3sXOG4@#PqgY?7ZiEZ0hYHyR!Q!;tq?E%~EPQmIu$>l4eD4
zb$K{mre!0Z9KW+pl1_h^5ZdHD5|k83HU!?b`I*O)$ow`wPK8VSs7hfRUr-K*GAJih
zw1D&oyl;x^5y4ckS)23dWwjE{;y#_g%8Q~w#%sqm>EWL7l0+cQEDSCJ+vT#tvR$YU
z9yN=``X(>^8d3oW$g<`c4lEJTe2BBWpS<HoIQ7`=JgbUGf`hEeJf-+;6s)r3pr&DT
z*Br1HMSlfb0mT3WIm`)KbL6Gc_z#@x3U&Y(^?)gAEN?Vxmca%K3f|8&2fFn$1X}1A
z;E-w@BlJcX)%bmuOxaPJgdh5-V4~7V%<;5BTQUxlDAz|DSsG3+qng)Arw?qRitOSe
zDTdO5<;kfTY1$czDv9Mr<{XNHAsb55)ztq-pqN7v6U|!q^GN7oH{!xhVYm*PWWpEl
z7GMe}kAm7v<WZodc4BNgW^ROkVLs2`Z4QA)Q$2g~8y9E_k|3oYj&g9Y35xTY9jZ~2
z^g$0iTGU(A4?*;=aAIMnBQU%=YonlN5sHSJ>Sf4?sT<<3o@EI)0lV!mqY&keIF%sM
zA1#~BY#brk??EWe3|Jt39!Z!-qbvbw7$Xk~iS~sHCi0|Cj}TMmDN1A6P2rUxCAV5>
z_F@Px1Nw-$g%Orsnh|=1_365hh(+3)1-2O-^uSRL!b?0rSXD~N5(=mlkZe;Keo<zo
zI|Q}@m(2`njs&LZ)vT@~<qAbWfmRLG2kZJLiVqwoix~)FkQt6iMuAvw&aVQs47Ik}
z@KaBnh$Ksf&Lzu%GXew2Nyr4_pq*Z-aLg`&SQPPl2{2xJuqh{kNOW}qZxrm`2867L
zHy(C@XS<aZp{eLkbT~LY-`#DWo_fromo{~R9F^4sZ=c|VwK;*Y9XN5{r+wz$#igZK
zhZbDsS5Odtc^UhNghErv$qXTMfxY;h0sZUk-#l?>x#YP=KD6%o;#U5gAd$%%T+OHN
z*x(s8j<M1yNF`OQ@oS!}>E<nEQVA3!#$Aj0N^j_Iw0WHZtUz<RCoko{9<kpx75H&q
zXUygpyn0H4s*hE`y_yGJpGGlJ>vEoD#pYwg5Gb1f>-Hwy)7}`&EwoER-r3w9EgwU@
zu;(?a5FaUZLCq{=tz#=xk;~$uYpS%zf?#`mOP%i2&|BK6^M-cck!KjQY_lgRomDAC
zpVl@V8iU5@-gN^JnBDRR<!zHVHrg&q<QjWHpA0(W@I-y{jUC6P7#k(@j8eT9$d)Ce
zk>!|=6K_J~kHJdbSb$%+quB9I6K`8V&2gP%Jq$m<&Wp1Eiv=fbxI&Tz)U9NY%#~7v
zTA-or<472b8uvJiTCM`-1E8F(^*4DM@ShT6e~%iJpYP6tWhQH7)ECqKvU^UP2etEb
zs?E-dU`%HQZj%D|>}0z5T!XS%HAy^+`>8Lruqg^9z^PE+^?>n=I~I^|hS|sBMOvCp
z=T*?fjHF$8La~C2i@g{PwEzwfRR)-<4F-pOP3RH3tFrlpcyKscCEJMP$S{erX*HLv
zjt5ChmE&kM;?q2!=M|FD9Nvt=PLaCK1GteZ5&=SBDG6Y4Dw62{mEtr~1TXe?{mL1T
z{TMkc##Z)qa%MPAuZ@%B9ZT2uA-Gc}%TKr<l#=-rFTi_7k-&(+7ZM+JKIR#zIy-Q|
zH$L=2%G#A#ddsFxc`$E^JNf5l{~UQZh~|qvmP*=4e^Ko{nUjV`Lu|6hV>P#A|3*bH
zTHz6mHFR%29(sL6j!tL)lw8hmsOqW8kTx5@(Z!4Jp<vvNWF~6DrR`pq6;x7Z5dgtp
zPjsz@TW$OkJn~07Yw1SuQYHV7bl0f|>8=5Yu-xKB1L+RcBzYgnLZDS=Y=Ep2arO#B
zG&x-$WGB3X7@8rSNOs!wEkLuC_jDb>^|_rhC0O()OUbIjK8c~|xT+duM3JYFD|+j+
zygd2^jQcM+R~)&xCYL%kZPuh{pamMnQ&rDO@+nYL=v_2+Yd3V?Wy$L1XFV;#&bAp`
z=>NnitFx%v@t?m%Re+sQrvr!B*pVsu0ciOvO_V_-%G@_F??2G78(Ejp0Dt}YmuygK
z**aUAKhXw96N=%|vwX@nhhilv<rCY8+N}kZE^x&Sd;)%DsudQcq@C%9l~pKRHSHXQ
zqJjeNQR2qj7dhMGSroGAxlCWP79S}}y0ciwmZiAR5@!xG7B|<<N|!#4vqGN1dxlwd
z<wg*)cdAk-7Sd6ZUc=V#W(;)2DlV-y9J^RN>2f98T;U+CGDD$G{t@N{&&4Eu^<GQW
ze3BxSZAPL9nE(WFd7jo+gWyMXJy7+rzgg+iH&|}RDOS*Ehypmjl`h6|6ZaIhyVMQN
z5g|ZH=$Z;54`jJkruiMe1Sd`B`(<Ey752mR(joe-Nh}xK@Q+GvK-{YHLG4`evV(BR
zFx~A~)sI!XY4MKj(vu=ys#@`Kxu$x;6;J4HwCxwRIwK2LqR7UGIYrVDkveBouMw1y
zLTAgBc3L$^f6M2l9&(<OWTz4R(pIWT_d49J=OqguJ5%F;1y?ohf0eG$RO#qW$_WW^
zSunJ&9^(8@L;T7{97P8k1d~{Za3>jh^2nw~oj4ClN<`4<&(J@j;vCBxdSg_Ae$yYk
z0Glwx2Aj~`E(EECF~R{OB&@+)NkOC*le^ioX0`_;I*Iqcw3i#d-=rzmZpzDGDRz;X
zCC%y~Stzj(azHvIE5V^MAg6iMQ}2!Xr=?c=ekJXo4d;_JgF2HnC4cvmPlX8&OlsJ7
zhg8wyE8hM;QD@kk_GzWwr6)A+-MT_=Jw}xteNDO_cb<VwGK_<LDEKOcz?LTxgN*@V
zx(mJisSAbIko#Sxgk@azh4kJcCc@T)zrn!-;UFf;oOa4yr%vZ6jY5$u_B=jaGj{+i
zT}Xyduut?F2X4c3mP&u22)+{2XgcYt70Of7W|YZ170Pkrxv!r~&I!`$sU&sE?g=ts
zoa_;|gvjx?USsgECzg&DWhd7Ar8{4g!9!hhf&H##CP1ReVWHhm>5VT#%<;ht>TCug
z8EsE4w5gV}h(f2w(#aEoTqsh3$-Ush@8JnPN72?EmzCYE{eo_iFOAe3uZbbvAe(CV
zg6kp$P_{~*KRU2HBS*)6V+3Bl!N6#QQ6FKu6N}y3wAUpXX)#_o%|7dsn}QJ2{1C<S
zW&3#(>~a!ePk-fP;mzTvY1}cXA@hE;kEI4+1y~=dm;CuOuzOldJuHb8_|x35=g(S0
zmlM3vhhpCK0@gpvau201P!fg(GfBBZUkrDp4ehk{CX~?>^GSuJq6#?elRL#R_8gzN
z885y9UADhro-uC)n|pgc%Orc1=rlp@*aQKrYkul<Q*jw8%*<t`!V6>;Y@Lpo5q6t%
zjt_h_@(|x`u=gS#{PyD{jLVnSoV{bQx<^<st~mVmF_LQh4nC(g=<d3t7&Cq?Ta2-8
zYd827#nK+)AJT`2objO47Js6`M2}WxC8M|MOGg}C?zyuQ`?VvU#NEPStUSO}h;@=&
z<60?d;}#Y*le^gUOC#%hY5YE#lFpIQ_4Q5jLNvrLY33J5<KrZ)(5wgCQNu}Vxm3q#
z*i8X7<?eJqN`=(aTb3T@8w&w);VTl<Q6!?X>c3EHxT(%jTu>i0>UE@&@c9c`4{)O2
z$2jXp+Qloj^XlS^_aggW2~YP3UAiZ(cl~foTBgu@dD{ao6UYnNV;3~V#P_~%o&S6%
zDPq^G_frVKeVgYpttKl&CpiX%1ykcfRvWKssuqQ?Q$l(33gu+tJ#lP{@|hXFxy2tr
zpxRDcELQwY3Z3yLaTkQJF!>oqfnFQH<^_bpivxb9PU+CSwG4}+nl<_j)wrZ@3+U%J
z7jJ5tZ`K=c=5My~Ld%K3mQzmTw_14ax0%(;+~OkmOcymHXH9iEHO?6!isid#Tm!p7
zb)40qn!*bum83EYWV8qQ9L~seKDod+mnSTX4ena0$vWSj|E%%F`qJ1rOJ`hHqEz^6
zd`0G~c_<i5>RWPk!4zj%Dd;r=>30c3*Cd{+cBUVAqrsQHdau;UF+LOY;M<J3S7_1_
zh$E5RPA}(ltuU*c9G`aR+JCEyj~pGG(j5+2BIHm!@^ZHN^kG2dz-!w`(^*}v4?Y?@
z<%Cas_Y9(o7U%{2WmoYiS5GVyYWx0s`v{2S8rE^5RRU^C=Dq-=9NyiA={&*D25KCn
zj})JOqzq39UCHhMu1@k2o%F}_TXLzzr~V2oy^6-OA}}!#`Wt|2TdThF=gcXAlmV;S
zcFm^%SG~lNF}0}##r~On-Z*Pq^KQNxxWhH>_;{1;jvs=694`3n@>`#&>I0ArTN`TV
zYuXc>%rn|?=_qW0Z$em<BKnY&56jHKCHDRc{%vt4OD&&TG{-owapsGMt01mwJC@HW
z?yP#Juw!p1vlJA4Gm0@N_%eLJVCUUY_;O@YN4g5>u{+4}R)WGew={tgIN9^w&w_Z1
zN8BLaPufv;-e+A+^J-vSI$hXN|J*2k&uBG>dfhEdN1_hm{=_6=9ngn1b&2X~{C<JN
zPXSTCBF&%`?E~YhKD^EHn+Xn*aGh)w606$U?n_tdr!b8}eEuxELo-V^9yAhuRWFqu
zqZU&xxZR<OKJM<fx!-~4EdT6l9rTqI{fQL=QPy+&Lr7QP?;;4+&)O(lwiBCS@|W@@
zx0OZ$<H%A&7@p2G@RiiYr-%`w*~+Y*k=DN1v~Z~rXKLnhi?o-7NH@`3)gMvs?*QKv
zO@G0=KA*mu!aytexGPeIrM*Qs$T5WI93279Nw=_X<2OF4gIxRybfkpG2*N|BgiXn7
zpv#%Sp@{LF8(fry1Kewv;soUpJ}|`P&ZZ(4ex&hDR87Y|bZ*Vx{9D(b(ZX%P5(0N!
f7%X}qH|0_Ec|!eNI#@qcx#959e2l0<PnG`%h6l9B

diff --git a/artifact_huggingface_download.sh b/artifact_huggingface_download.sh
index 7c2cf9e..5e4fbf5 100755
--- a/artifact_huggingface_download.sh
+++ b/artifact_huggingface_download.sh
@@ -1,40 +1,24 @@
 #!/bin/bash
 #
-# ComfyUI Model Downloader - A Beautiful CLI Tool
-# Downloads AI models from HuggingFace and creates symlinks to ComfyUI directories
+# HuggingFace Model Downloader - A Beautiful CLI Tool
+# Downloads AI models from HuggingFace and creates symlinks to output directories
 #
 # Usage: ./artifact_huggingface_download.sh [COMMAND] [OPTIONS]
 #
 # Commands:
-#   download    Download models from HuggingFace to cache directory
-#   link        Create symlinks from cache to ComfyUI models directory
-#   both        Download and link (default)
-#   verify      Verify symlinks in ComfyUI models directory
+#   download    Download models to cache directory (default)
+#   link        Create symlinks from cache to output directory
+#   verify      Verify symlinks in output directory
 #
 # Options:
-#   -c, --config FILE       Path to YAML configuration file
-#   --cache-dir DIR         HuggingFace cache directory (default: auto-detect)
-#   --comfyui-dir DIR       ComfyUI installation directory
-#   --filter-repo REPO...   Only process specific repositories
-#   --category CAT1[,CAT2]  Filter by category (comma-separated for multiple)
-#   --cleanup, --clean      Remove unused cache files (link/both only)
-#   --dry-run, -n           Show what would be done without making changes
-#
-# Examples:
-#   # Download and link all models from config
-#   ./artifact_huggingface_download.sh both -c models.yaml
-#
-#   # Download only specific categories
-#   ./artifact_huggingface_download.sh download -c models.yaml --category image_models,video_models
-#
-#   # Link with cleanup (remove unused cache files)
-#   ./artifact_huggingface_download.sh link -c models.yaml --cleanup
-#
-#   # Dry-run to preview operations
-#   ./artifact_huggingface_download.sh both -c models.yaml --dry-run
-#
-#   # Process only specific repositories
-#   ./artifact_huggingface_download.sh both -c models.yaml --filter-repo black-forest-labs/FLUX.1-schnell
+#   -c, --config FILE         Configuration YAML file (required)
+#   --cache-dir DIR           Cache directory
+#   --output-dir DIR          Output/installation directory
+#   --category CAT1,CAT2      Filter by category (comma-separated)
+#   --repo-id ID1,ID2         Filter by repo_id (comma-separated)
+#   --auth-token TOKEN        HuggingFace token
+#   -n, --dry-run             Show what would be done
+#   -h, --help                Show help
 #
 
 set -euo pipefail
@@ -43,21 +27,17 @@ set -euo pipefail
 # COLOR PALETTE - Beautiful Terminal Colors
 # ============================================================================
 
-# Reset
 RESET='\033[0m'
 
 # Foreground Colors
-BLACK='\033[0;30m'
 RED='\033[0;31m'
 GREEN='\033[0;32m'
 YELLOW='\033[0;33m'
 BLUE='\033[0;34m'
 MAGENTA='\033[0;35m'
 CYAN='\033[0;36m'
-WHITE='\033[0;37m'
 
 # Bold
-BOLD_BLACK='\033[1;30m'
 BOLD_RED='\033[1;31m'
 BOLD_GREEN='\033[1;32m'
 BOLD_YELLOW='\033[1;33m'
@@ -66,121 +46,98 @@ BOLD_MAGENTA='\033[1;35m'
 BOLD_CYAN='\033[1;36m'
 BOLD_WHITE='\033[1;37m'
 
-# Background Colors
-BG_BLACK='\033[40m'
-BG_RED='\033[41m'
-BG_GREEN='\033[42m'
-BG_YELLOW='\033[43m'
-BG_BLUE='\033[44m'
-BG_MAGENTA='\033[45m'
+# Background
 BG_CYAN='\033[46m'
-BG_WHITE='\033[47m'
 
 # Styles
 DIM='\033[2m'
-ITALIC='\033[3m'
-UNDERLINE='\033[4m'
-BLINK='\033[5m'
-REVERSE='\033[7m'
 
 # ============================================================================
-# UNICODE CHARACTERS - Make it Pretty
+# UNICODE CHARACTERS
 # ============================================================================
 
 CHECK_MARK="✓"
 CROSS_MARK="✗"
 ROCKET="🚀"
-PACKAGE="📦"
 DOWNLOAD="⬇️"
-SPARKLES="✨"
-FIRE="🔥"
-CLOCK="⏱️"
-FOLDER="📁"
-LINK="🔗"
-STAR="⭐"
+LINK_ICON="🔗"
 WARNING="⚠️"
 INFO="ℹ️"
+SPARKLES="✨"
 ARROW_RIGHT="→"
-DOUBLE_ARROW="»"
 BOX_LIGHT="─"
-BOX_HEAVY="━"
 BOX_DOUBLE="═"
+PACKAGE="📦"
 
 # ============================================================================
 # CONFIGURATION
 # ============================================================================
 
-# Script directory
-SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
-PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
+CONFIG_FILE=""
+COMMAND="download"
+DRY_RUN=false
+CATEGORY_FILTER=""
+REPO_ID_FILTER=""
 
-# Default configuration file path
-# Try multiple possible locations
-if [[ -f "${HOME}/Projects/runpod/comfyui_models.yaml" ]]; then
-    CONFIG_FILE="${HOME}/Projects/runpod/comfyui_models.yaml"
-elif [[ -f "${PROJECT_ROOT}/comfyui_models.yaml" ]]; then
-    CONFIG_FILE="${PROJECT_ROOT}/comfyui_models.yaml"
-elif [[ -f "${SCRIPT_DIR}/comfyui_models.yaml" ]]; then
-    CONFIG_FILE="${SCRIPT_DIR}/comfyui_models.yaml"
-else
-    CONFIG_FILE=""  # No config file by default
-fi
-
-# Default cache directory - detect RunPod or use HuggingFace default
+# Default directories - detect RunPod or local
 if [[ -d "/workspace" ]]; then
-    # RunPod environment
     CACHE_DIR="${CACHE_DIR:-/workspace/huggingface_cache}"
-    COMFYUI_DIR="${COMFYUI_DIR:-/workspace/ComfyUI/models}"
+    OUTPUT_DIR="${OUTPUT_DIR:-/workspace/ComfyUI/models}"
 else
-    # Local environment
-    CACHE_DIR="${CACHE_DIR:-${HOME}/.cache/huggingface}"
-    COMFYUI_DIR="${COMFYUI_DIR:-${HOME}/ComfyUI/models}"
+    CACHE_DIR="${CACHE_DIR:-${HOME}/.cache/huggingface/hub}"
+    OUTPUT_DIR="${OUTPUT_DIR:-${HOME}/ComfyUI/models}"
 fi
 
-# Default command
-COMMAND="both"
-
-# Feature flags
-CATEGORY_FILTER=""  # Empty = all categories, or comma-separated list
-CLEANUP_MODE=false  # Remove unused files from HuggingFace cache
-DRY_RUN=false       # Simulate operations without making changes
-
-# HuggingFace token from environment or .env file
-# Initialize HF_TOKEN if not set
+# HuggingFace token from environment
 HF_TOKEN="${HF_TOKEN:-}"
 
-# Try multiple locations for .env file
-if [[ -z "${HF_TOKEN}" ]] && [[ -f "${PROJECT_ROOT}/ai/.env" ]]; then
-    HF_TOKEN=$(grep ^HF_TOKEN "${PROJECT_ROOT}/ai/.env" | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
-fi
-if [[ -z "${HF_TOKEN}" ]] && [[ -f "${PROJECT_ROOT}/.env" ]]; then
-    HF_TOKEN=$(grep ^HF_TOKEN "${PROJECT_ROOT}/.env" | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
-fi
-if [[ -z "${HF_TOKEN}" ]] && [[ -f "/workspace/ai/.env" ]]; then
-    HF_TOKEN=$(grep ^HF_TOKEN "/workspace/ai/.env" | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
-fi
+# Try to load from .env files
+load_env_token() {
+    if [[ -n "$HF_TOKEN" ]]; then
+        return 0
+    fi
+
+    local env_files=(
+        "${HOME}/.env"
+        "${HOME}/Projects/runpod/.env"
+        "${HOME}/Projects/runpod/ai/.env"
+        "/workspace/.env"
+        "/workspace/ai/.env"
+    )
+
+    for env_file in "${env_files[@]}"; do
+        if [[ -f "$env_file" ]]; then
+            local token
+            token=$(grep "^HF_TOKEN=" "$env_file" 2>/dev/null | cut -d'=' -f2- | tr -d '"' | tr -d "'" || true)
+            if [[ -n "$token" ]]; then
+                HF_TOKEN="$token"
+                return 0
+            fi
+        fi
+    done
+}
 
 # ============================================================================
-# UTILITY FUNCTIONS - The Magic Happens Here
+# LOGGING FUNCTIONS
 # ============================================================================
 
-# Print functions with beautiful formatting
 print_banner() {
     local text="$1"
-    local width=80
-    local padding=$(( (width - ${#text} - 2) / 2 ))
+    local width=70
+    local text_len=${#text}
+    local padding=$(( (width - text_len) / 2 ))
 
-    echo -e ""
-    echo -e "${BOLD_CYAN}${BOX_DOUBLE}$(printf '%.0s'"${BOX_DOUBLE}" $(seq 1 $width))${BOX_DOUBLE}${RESET}"
-    echo -e "${BOLD_CYAN}${BOX_DOUBLE}$(printf '%.0s ' $(seq 1 $padding))${BOLD_MAGENTA}${text}$(printf '%.0s ' $(seq 1 $padding))${BOLD_CYAN}${BOX_DOUBLE}${RESET}"
-    echo -e "${BOLD_CYAN}${BOX_DOUBLE}$(printf '%.0s'"${BOX_DOUBLE}" $(seq 1 $width))${BOX_DOUBLE}${RESET}"
-    echo -e ""
+    echo ""
+    echo -e "${BOLD_CYAN}${BOX_DOUBLE}$(printf '%0.s═' $(seq 1 $width))${BOX_DOUBLE}${RESET}"
+    echo -e "${BOLD_CYAN}║$(printf '%*s' $padding '')${BOLD_MAGENTA}${text}$(printf '%*s' $((width - padding - text_len)) '')${BOLD_CYAN}║${RESET}"
+    echo -e "${BOLD_CYAN}${BOX_DOUBLE}$(printf '%0.s═' $(seq 1 $width))${BOX_DOUBLE}${RESET}"
+    echo ""
 }
 
 print_section() {
     local text="$1"
-    echo -e "\n${BOLD_YELLOW}${DOUBLE_ARROW} ${text}${RESET}"
-    echo -e "${CYAN}$(printf '%.0s'"${BOX_LIGHT}" $(seq 1 80))${RESET}"
+    echo -e "\n${BOLD_CYAN}» ${text}${RESET}"
+    echo -e "${CYAN}$(printf '%0.s─' $(seq 1 70))${RESET}"
 }
 
 print_success() {
@@ -203,1382 +160,555 @@ print_step() {
     local current="$1"
     local total="$2"
     local text="$3"
-    echo -e "${BOLD_BLUE}[${current}/${total}]${RESET} ${MAGENTA}${DOWNLOAD}${RESET} ${text}"
+    echo -e "${BOLD_BLUE}[${current}/${total}]${RESET} ${CYAN}${PACKAGE}${RESET} ${text}"
 }
 
 print_detail() {
     echo -e "  ${DIM}${CYAN}${ARROW_RIGHT} $1${RESET}"
 }
 
-# Progress bar function
 show_progress() {
     local current="$1"
     local total="$2"
-    local width=50
+    local width=40
     local percentage=$((current * 100 / total))
     local filled=$((current * width / total))
     local empty=$((width - filled))
 
     printf "\r  ${BOLD_CYAN}Progress: ${RESET}["
-    printf "${BG_GREEN}${BOLD_WHITE}%${filled}s${RESET}" | tr ' ' '█'
+    printf "${BG_CYAN}${BOLD_WHITE}%${filled}s${RESET}" | tr ' ' '█'
     printf "${DIM}%${empty}s${RESET}" | tr ' ' '░'
     printf "] ${BOLD_YELLOW}%3d%%${RESET} ${DIM}(%d/%d)${RESET}" "$percentage" "$current" "$total"
 }
 
-# Parse YAML (simple implementation)
-parse_yaml() {
-    local yaml_file="$1"
-    local category="$2"
+# ============================================================================
+# YAML PARSING (using yq)
+# ============================================================================
 
-    python3 - "$yaml_file" "$category" <<EOPYAML
-import yaml
-import sys
-
-yaml_file = sys.argv[1]
-category = sys.argv[2]
-
-try:
-    with open(yaml_file, 'r') as f:
-        config = yaml.safe_load(f)
-
-    if category == 'settings':
-        settings = config.get('settings', {})
-        print("CACHE_DIR={0}".format(settings.get('cache_dir', '/workspace/huggingface_cache')))
-        print("PARALLEL_DOWNLOADS={0}".format(settings.get('parallel_downloads', 1)))
-    elif category == 'categories':
-        for cat_name in config.get('model_categories', {}).keys():
-            print(cat_name)
-    elif category in config.get('model_categories', {}):
-        models = config['model_categories'][category]
-        for model in models:
-            repo_id = model.get('repo_id', '')
-            description = model.get('description', '')
-            size_gb = model.get('size_gb', 0)
-            essential = model.get('essential', False)
-            filename = model.get('filename', '')
-            print('{0}|{1}|{2}|{3}|{4}'.format(repo_id, description, size_gb, essential, filename))
-    else:
-        sys.exit(1)
-except Exception as e:
-    print("ERROR: {0}".format(e), file=sys.stderr)
-    sys.exit(1)
-EOPYAML
-}
-
-# Validate and get categories to process
-validate_and_get_categories() {
-    local config_file="$1"
-
-    # Get all available categories
-    local all_categories
-    all_categories=$(parse_yaml "$config_file" "categories")
-
-    # If no filter specified, return all categories
-    if [[ -z "$CATEGORY_FILTER" ]]; then
-        echo "$all_categories"
-        return 0
-    fi
-
-    # Split comma-separated categories and validate each
-    local requested_categories
-    IFS=',' read -ra requested_categories <<< "$CATEGORY_FILTER"
-
-    local validated_categories=()
-    for requested in "${requested_categories[@]}"; do
-        # Trim whitespace
-        requested=$(echo "$requested" | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')
-
-        # Check if category exists
-        if echo "$all_categories" | grep -q "^${requested}$"; then
-            validated_categories+=("$requested")
-        else
-            print_error "Invalid category: '$requested'"
-            echo ""
-            echo "Available categories:"
-            while IFS= read -r cat; do
-                echo "  - $cat"
-            done <<< "$all_categories"
-            exit 1
-        fi
-    done
-
-    # Print validated categories (one per line)
-    for cat in "${validated_categories[@]}"; do
-        echo "$cat"
-    done
-}
-
-# Parse file mappings for a specific model
-parse_file_mappings() {
-    local yaml_file="$1"
-    local category="$2"
-    local repo_id="$3"
-
-    python3 - "$yaml_file" "$category" "$repo_id" <<EOPYTHON
-import yaml
-import sys
-
-yaml_file = sys.argv[1]
-category = sys.argv[2]
-repo_id = sys.argv[3]
-
-try:
-    with open(yaml_file, 'r') as f:
-        config = yaml.safe_load(f)
-
-    if category in config.get('model_categories', {}):
-        models = config['model_categories'][category]
-        for model in models:
-            if model.get('repo_id', '') == repo_id:
-                files = model.get('files', [])
-                if files:
-                    for file_mapping in files:
-                        source = file_mapping.get('source', '')
-                        dest = file_mapping.get('dest', source)
-                        if source:
-                            print('{0}|{1}'.format(source, dest))
-                sys.exit(0)
-
-    # No file mappings found
-    sys.exit(0)
-except Exception as e:
-    print("ERROR: {0}".format(e), file=sys.stderr)
-    sys.exit(1)
-EOPYTHON
-}
-
-# Check dependencies
-check_dependencies() {
-    print_section "Checking Dependencies"
-
-    local missing_deps=()
-
-    # Check Python 3
-    if ! command -v python3 &> /dev/null; then
-        missing_deps+=("python3")
-    fi
-
-    # Check pip
-    if ! command -v pip3 &> /dev/null; then
-        missing_deps+=("pip3")
-    fi
-
-    # Check required Python packages
-    if ! python3 -c "import yaml" 2>/dev/null; then
-        print_warning "PyYAML not installed, installing..."
-        pip3 install pyyaml -q
-    fi
-
-    if ! python3 -c "import huggingface_hub" 2>/dev/null; then
-        print_warning "huggingface_hub not installed, installing..."
-        pip3 install huggingface_hub -q
-    fi
-
-    if [[ ${#missing_deps[@]} -gt 0 ]]; then
-        print_error "Missing dependencies: ${missing_deps[*]}"
+check_yq() {
+    if ! command -v yq &>/dev/null; then
+        print_error "yq is not installed. Please install yq first."
+        print_info "Install: https://github.com/mikefarah/yq"
         exit 1
     fi
-
-    print_success "All dependencies satisfied"
 }
 
-# Validate configuration
-validate_config() {
-    print_section "Validating Configuration"
+# Get total count of models
+get_model_count() {
+    local config="$1"
+    yq eval '. | length' "$config" 2>/dev/null || echo "0"
+}
 
-    # Show current command
-    print_info "Command: ${BOLD_CYAN}${COMMAND}${RESET}"
+# Get model field at index
+get_model_field() {
+    local config="$1"
+    local index="$2"
+    local field="$3"
+    local value
+    value=$(yq eval ".[$index].$field // \"\"" "$config" 2>/dev/null)
+    echo "$value" | sed 's/^"//;s/"$//'
+}
 
-    if [[ -n "$CONFIG_FILE" ]]; then
-        if [[ ! -f "$CONFIG_FILE" ]]; then
-            print_error "Configuration file not found: $CONFIG_FILE"
-            exit 1
-        fi
-        print_success "Configuration file found: ${CYAN}${CONFIG_FILE}${RESET}"
-    else
-        print_warning "No configuration file specified"
-    fi
+# Get files array length for a model
+get_files_count() {
+    local config="$1"
+    local index="$2"
+    yq eval ".[$index].files | length" "$config" 2>/dev/null || echo "0"
+}
 
-    # HF_TOKEN only required for download and both commands
-    if [[ "$COMMAND" == "download" ]] || [[ "$COMMAND" == "both" ]]; then
-        if [[ -z "$HF_TOKEN" ]]; then
-            print_error "HF_TOKEN not set. Please set it in .env file or environment."
-            exit 1
-        fi
-        print_success "HuggingFace token configured: ${DIM}${HF_TOKEN:0:10}...${RESET}"
-    elif [[ "$COMMAND" == "verify" ]]; then
-        print_info "Verify mode: HuggingFace token not required"
-    fi
+# Get file mapping at index
+get_file_field() {
+    local config="$1"
+    local model_index="$2"
+    local file_index="$3"
+    local field="$4"
+    local value
+    value=$(yq eval ".[$model_index].files[$file_index].$field // \"\"" "$config" 2>/dev/null)
+    echo "$value" | sed 's/^"//;s/"$//'
+}
 
-    # Validate flag combinations
-    if [[ "$CLEANUP_MODE" == true ]] && [[ ! "$COMMAND" =~ ^(link|both)$ ]]; then
-        print_error "--cleanup can only be used with 'link' or 'both' commands"
-        exit 1
-    fi
+# Check if model matches filters
+matches_filters() {
+    local repo_id="$1"
+    local category="$2"
 
-    # Cache directory
-    if [[ "$COMMAND" == "download" ]] || [[ "$COMMAND" == "both" ]]; then
-        if [[ ! -d "$CACHE_DIR" ]]; then
-            print_info "Creating cache directory: ${CYAN}${CACHE_DIR}${RESET}"
-            mkdir -p "$CACHE_DIR"
-        fi
-        print_success "Cache directory ready: ${CYAN}${CACHE_DIR}${RESET}"
-    else
-        # For link and verify commands, just show the directory
-        if [[ -d "$CACHE_DIR" ]]; then
-            print_success "Cache directory found: ${CYAN}${CACHE_DIR}${RESET}"
-        else
-            print_warning "Cache directory not found: ${CYAN}${CACHE_DIR}${RESET}"
-        fi
-    fi
-
-    # ComfyUI directory
-    if [[ "$COMMAND" == "link" ]] || [[ "$COMMAND" == "both" ]] || [[ "$COMMAND" == "verify" ]]; then
-        if [[ -d "$COMFYUI_DIR" ]]; then
-            print_success "ComfyUI directory found: ${CYAN}${COMFYUI_DIR}${RESET}"
-        else
-            if [[ "$COMMAND" == "verify" ]]; then
-                print_warning "ComfyUI directory not found: ${CYAN}${COMFYUI_DIR}${RESET}"
-            else
-                print_info "ComfyUI directory: ${CYAN}${COMFYUI_DIR}${RESET}"
+    # Check category filter
+    if [[ -n "$CATEGORY_FILTER" ]]; then
+        local match=false
+        IFS=',' read -ra cats <<< "$CATEGORY_FILTER"
+        for cat in "${cats[@]}"; do
+            cat=$(echo "$cat" | xargs)
+            if [[ "$category" == "$cat" ]]; then
+                match=true
+                break
             fi
-        fi
-    fi
-}
-
-# Find model files in HuggingFace cache
-find_model_files() {
-    local repo_id="$1"
-    local filename_filter="$2"
-
-    python3 - "$CACHE_DIR" "$repo_id" "$filename_filter" <<EOPYFINDFIND
-import os
-import sys
-from pathlib import Path
-
-cache_dir = sys.argv[1]
-repo_id = sys.argv[2]
-filename_filter = sys.argv[3]
-
-# HuggingFace cache structure: cache_dir/models--org--name/snapshots/hash/
-# Try both with and without 'hub/' subdirectory for compatibility
-cache_path = Path(cache_dir)
-repo_path = repo_id.replace('/', '--')
-model_dir = cache_path / 'models--{0}'.format(repo_path)
-
-# First attempt: direct path
-model_dir_original = model_dir
-if not model_dir.exists():
-    # Fallback to hub/ subdirectory if direct path doesn't exist
-    model_dir = cache_path / 'hub' / 'models--{0}'.format(repo_path)
-
-if not model_dir.exists():
-    sys.stderr.write('ERROR: Model directory not found for {0}\n'.format(repo_id))
-    sys.stderr.write('  Tried: {0}\n'.format(str(model_dir_original)))
-    sys.stderr.write('  Tried: {0}\n'.format(str(model_dir)))
-    sys.stderr.write('  Cache: {0}\n'.format(str(cache_path)))
-    sys.exit(1)
-
-# Find the latest snapshot
-snapshots_dir = model_dir / 'snapshots'
-if not snapshots_dir.exists():
-    sys.stderr.write('ERROR: Snapshots directory not found for {0}\n'.format(repo_id))
-    sys.stderr.write('  Model dir: {0}\n'.format(str(model_dir)))
-    sys.stderr.write('  Expected: {0}\n'.format(str(snapshots_dir)))
-    sys.exit(1)
-
-# Get all snapshot directories sorted by modification time
-try:
-    snapshots = sorted(snapshots_dir.iterdir(), key=lambda x: x.stat().st_mtime, reverse=True)
-except Exception as e:
-    sys.stderr.write('ERROR: Failed to list snapshots for {0}: {1}\n'.format(repo_id, str(e)))
-    sys.exit(1)
-
-if not snapshots:
-    sys.stderr.write('ERROR: No snapshots found for {0}\n'.format(repo_id))
-    sys.stderr.write('  Snapshots dir: {0}\n'.format(str(snapshots_dir)))
-    sys.exit(1)
-
-latest_snapshot = snapshots[0]
-
-# Find model files
-file_count = 0
-for file_path in latest_snapshot.rglob('*'):
-    if file_path.is_file():
-        # If filename filter is specified, only match those files
-        if filename_filter and filename_filter not in file_path.name:
-            continue
-        # Skip metadata files except important config files
-        # Allow: config.json, tokenizer.json, tokenizer_config.json, sentencepiece models, .index.json for sharded models
-        important_files = ('config.json', 'tokenizer.json', 'tokenizer_config.json', '.model')
-        if file_path.name.endswith(('.txt', '.md', '.gitattributes')):
-            continue
-        if file_path.name.endswith('.json') and not (file_path.name in important_files or file_path.name.endswith('.index.json')):
-            continue
-        print(str(file_path))
-        file_count += 1
-
-if file_count == 0:
-    sys.stderr.write('WARN: No files matched filter for {0} (filter: {1})\n'.format(repo_id, filename_filter))
-EOPYFINDFIND
-}
-
-# Download a single model
-download_model() {
-    local repo_id="$1"
-    local description="$2"
-    local size_gb="$3"
-
-    print_detail "Repository: ${BOLD_WHITE}${repo_id}${RESET}"
-    print_detail "Description: ${description}"
-    print_detail "Size: ${BOLD_YELLOW}${size_gb}GB${RESET}"
-
-    # Dry-run mode: skip actual download
-    if [[ "$DRY_RUN" == true ]]; then
-        print_info "DRY-RUN: Would download ${BOLD_WHITE}${repo_id}${RESET} (~${size_gb} GB)"
-        return 0
-    fi
-
-    # Download using Python
-    python3 - <<EOPYDOWNLOAD
-import os
-import sys
-from huggingface_hub import snapshot_download
-
-cache_dir = '${CACHE_DIR}'
-token = '${HF_TOKEN}'
-repo_id = '${repo_id}'
-
-os.environ['HF_HOME'] = cache_dir
-
-try:
-    snapshot_download(
-        repo_id=repo_id,
-        cache_dir=cache_dir,
-        token=token,
-        resume_download=True
-    )
-    print("SUCCESS")
-except Exception as e:
-    print("ERROR: {0}".format(e), file=sys.stderr)
-    sys.exit(1)
-EOPYDOWNLOAD
-
-    if [[ $? -eq 0 ]]; then
-        print_success "Downloaded ${BOLD_WHITE}${repo_id}${RESET}"
-        return 0
-    else
-        print_error "Failed to download ${repo_id}"
-        return 1
-    fi
-}
-
-# Create symlink for a model
-link_model() {
-    local repo_id="$1"
-    local filename_filter="$2"
-    local file_mappings="$3"  # Optional: explicit source|dest mappings
-
-    local linked_count=0
-
-    # If explicit file mappings are provided, use them
-    if [[ -n "$file_mappings" ]]; then
-        print_detail "Using explicit file mappings from YAML"
-
-        # Extract unique target directories from file mappings for display
-        local target_dirs=()
-        while IFS='|' read -r source_pattern dest_path; do
-            if [[ -n "$dest_path" && "$dest_path" == *"/"* ]]; then
-                local dir_path="${dest_path%/*}"
-                target_dirs+=("${COMFYUI_DIR}/${dir_path}")
-            fi
-        done <<< "$file_mappings"
-
-        # Remove duplicates and display
-        local unique_dirs=($(printf '%s\n' "${target_dirs[@]}" | sort -u))
-        for target_dir in "${unique_dirs[@]}"; do
-            print_detail "Linking to: ${CYAN}${target_dir}/${RESET}"
         done
-
-        # Dry-run mode: show what would be linked
-        if [[ "$DRY_RUN" == true ]]; then
-            local file_count=$(echo "$file_mappings" | grep -c .)
-            print_info "DRY-RUN: Would link ${BOLD_YELLOW}${file_count}${RESET} file(s)"
-            print_detail "Files that would be linked:"
-            while IFS='|' read -r source_pattern dest_path; do
-                if [[ -n "$dest_path" ]]; then
-                    print_detail "  ${LINK} Would link: ${DIM}${dest_path}${RESET}"
-                fi
-            done <<< "$file_mappings"
-            return 0
-        fi
-
-        # Process each file mapping - find each file individually by its pattern
-        while IFS='|' read -r source_pattern dest_path; do
-            if [[ -z "$source_pattern" ]] || [[ -z "$dest_path" ]]; then
-                continue
-            fi
-
-            # Extract just the filename from the source pattern (remove directory path)
-            local filename_only
-            filename_only=$(basename "$source_pattern")
-
-            # Find files matching the filename (HuggingFace cache flattens directory structure)
-            local source_file
-            source_file=$(find_model_files "$repo_id" "$filename_only" 2>/dev/null | grep "/$filename_only$" | head -n1)
-
-            if [[ -z "$source_file" ]] || [[ ! -f "$source_file" ]]; then
-                print_warning "Source file not found: ${source_pattern}"
-                continue
-            fi
-
-            # Construct full link path with directory included in dest_path
-            local link_path="${COMFYUI_DIR}/${dest_path}"
-            local link_dir=$(dirname "$link_path")
-
-            # Ensure directory exists
-            if [[ ! -d "$link_dir" ]]; then
-                mkdir -p "$link_dir"
-            fi
-
-            # Remove existing symlink or file if it exists
-            if [[ -L "$link_path" ]]; then
-                rm -f "$link_path"
-            elif [[ -e "$link_path" ]]; then
-                print_warning "File already exists (not a symlink): ${dest_path}"
-                continue
-            fi
-
-            # Create symlink
-            ln -s "$source_file" "$link_path"
-            print_detail "${LINK} Linked: ${DIM}${dest_path}${RESET}"
-            linked_count=$((linked_count+1))
-        done <<< "$file_mappings"
-    else
-        # Fallback: use automatic prefixing for files without explicit mappings
-        print_detail "No file mappings found, using automatic prefixing"
-
-        # Extract model name from repo_id for prefixing filenames
-        # e.g., "facebook/musicgen-medium" -> "musicgen-medium"
-        local model_name=$(echo "$repo_id" | sed 's/.*\///')
-
-        while IFS= read -r source_file; do
-            if [[ -f "$source_file" ]]; then
-                local filename=$(basename "$source_file")
-
-                # Add model name prefix to filename for better organization
-                # e.g., "pytorch_model.bin" -> "musicgen-medium-pytorch_model.bin"
-                local prefixed_filename="${model_name}-${filename}"
-                local link_path="${target_dir}/${prefixed_filename}"
-
-                # Remove existing symlink or file if it exists
-                if [[ -L "$link_path" ]]; then
-                    rm -f "$link_path"
-                elif [[ -e "$link_path" ]]; then
-                    print_warning "File already exists (not a symlink): ${prefixed_filename}"
-                    continue
-                fi
-
-                # Create symlink
-                ln -s "$source_file" "$link_path"
-                print_detail "${LINK} Linked: ${DIM}${prefixed_filename}${RESET}"
-                linked_count=$((linked_count+1))
-            fi
-        done <<< "$model_files"
-    fi
-
-    if [[ $linked_count -gt 0 ]]; then
-        print_success "Linked ${linked_count} file(s) for ${BOLD_WHITE}${repo_id}${RESET}"
-        return 0
-    else
-        print_error "Failed to link files for ${repo_id}"
-        return 1
-    fi
-}
-
-# Cleanup unused cache files that aren't symlinked
-cleanup_unused_cache_files() {
-    local repo_id="$1"
-    local cache_dir="$2"
-    local comfyui_dir="$3"
-    local file_mappings="$4"
-
-    # Find the latest snapshot directory for this repo
-    local repo_cache_dir="${cache_dir}/models--${repo_id//\//--}"
-    if [[ ! -d "$repo_cache_dir" ]]; then
-        print_warning "Cache directory not found for $repo_id"
-        return 1
-    fi
-
-    print_info "Analyzing cache for ${BOLD_WHITE}${repo_id}${RESET}..."
-
-    # Use Python to clean up old snapshots AND non-whitelisted files in latest snapshot
-    local cleanup_result
-    cleanup_result=$(python3 - "$repo_cache_dir" "$comfyui_dir" "$file_mappings" <<'EOPYCLEANUP'
-import os
-import sys
-import shutil
-from pathlib import Path
-
-repo_cache = Path(sys.argv[1])
-comfyui_dir = Path(sys.argv[2])
-file_mappings_str = sys.argv[3] if len(sys.argv) > 3 else ""
-
-# Parse whitelist from file_mappings (format: "source|dest\nsource|dest\n...")
-whitelist_sources = set()
-if file_mappings_str:
-    for line in file_mappings_str.strip().split('\n'):
-        if '|' in line:
-            source = line.split('|')[0].strip()
-            if source:
-                whitelist_sources.add(source)
-
-# Essential HuggingFace metadata files to always preserve
-ESSENTIAL_FILES = {
-    '.gitattributes',
-    'README.md',
-    'model_index.json',
-    '.huggingface',
-    'config.json'
-}
-
-# Find latest snapshot
-snapshots_dir = repo_cache / 'snapshots'
-if not snapshots_dir.exists():
-    sys.exit(0)
-
-snapshots = sorted(snapshots_dir.iterdir(), key=lambda x: x.stat().st_mtime, reverse=True)
-if not snapshots:
-    sys.exit(0)
-
-latest_snapshot = snapshots[0]
-old_snapshots = snapshots[1:]  # All snapshots except the latest
-
-# Calculate size of old snapshot directories
-old_snapshot_size = 0
-old_snapshot_paths = []
-for old_snap in old_snapshots:
-    try:
-        # Calculate size of old snapshot directory
-        for file_path in old_snap.rglob('*'):
-            if file_path.is_file():
-                old_snapshot_size += file_path.stat().st_size
-        old_snapshot_paths.append(str(old_snap))
-    except Exception:
-        pass
-
-# Find non-whitelisted files in latest snapshot
-unwanted_files = []
-unwanted_size = 0
-if whitelist_sources:
-    for item in latest_snapshot.rglob('*'):
-        if item.is_file():
-            # Get relative path from snapshot root
-            rel_path = str(item.relative_to(latest_snapshot))
-
-            # Check if this file is in whitelist or is essential
-            is_whitelisted = False
-
-            # Check exact match first
-            if rel_path in whitelist_sources:
-                is_whitelisted = True
-            else:
-                # Check if any whitelist entry matches this file
-                # (handles cases where whitelist has paths like "split_files/diffusion_models/file.safetensors")
-                for whitelisted in whitelist_sources:
-                    if rel_path == whitelisted or rel_path.endswith('/' + whitelisted):
-                        is_whitelisted = True
-                        break
-
-            # Check if it's an essential file
-            if item.name in ESSENTIAL_FILES:
-                is_whitelisted = True
-
-            # If not whitelisted, mark for deletion
-            if not is_whitelisted:
-                unwanted_files.append(str(item))
-                unwanted_size += item.stat().st_size
-
-# Output results: old_snapshot_count|old_snapshot_size|unwanted_files_count|unwanted_size
-print(f"{len(old_snapshot_paths)}|{old_snapshot_size}|{len(unwanted_files)}|{unwanted_size}")
-for snap in old_snapshot_paths:
-    print(snap)
-for unwanted_file in unwanted_files:
-    print(unwanted_file)
-EOPYCLEANUP
-)
-
-    # Parse results
-    local first_line
-    first_line=$(echo "$cleanup_result" | head -n 1)
-    local snapshot_count
-    snapshot_count=$(echo "$first_line" | cut -d'|' -f1)
-    local snapshot_bytes
-    snapshot_bytes=$(echo "$first_line" | cut -d'|' -f2)
-    local unwanted_count
-    unwanted_count=$(echo "$first_line" | cut -d'|' -f3)
-    local unwanted_bytes
-    unwanted_bytes=$(echo "$first_line" | cut -d'|' -f4)
-
-    # Check if there's anything to clean
-    if [[ "$snapshot_count" -eq 0 ]] && [[ "$unwanted_count" -eq 0 ]]; then
-        print_success "No cleanup needed - cache is optimal"
-        return 0
-    fi
-
-    # Convert bytes to MB
-    local snapshot_mb
-    snapshot_mb=$(echo "scale=2; $snapshot_bytes / 1048576" | bc)
-    local unwanted_mb
-    unwanted_mb=$(echo "scale=2; $unwanted_bytes / 1048576" | bc)
-    local total_mb
-    total_mb=$(echo "scale=2; ($snapshot_bytes + $unwanted_bytes) / 1048576" | bc)
-
-    # Get list of items to delete (skip first line which is summary)
-    # First snapshot_count lines are old snapshots, remaining lines are unwanted files
-    local all_items
-    all_items=$(echo "$cleanup_result" | tail -n +2)
-
-    local snapshots_to_delete
-    if [[ "$snapshot_count" -gt 0 ]]; then
-        snapshots_to_delete=$(echo "$all_items" | head -n "$snapshot_count")
-    else
-        snapshots_to_delete=""
-    fi
-
-    local files_to_delete
-    if [[ "$unwanted_count" -gt 0 ]]; then
-        files_to_delete=$(echo "$all_items" | tail -n "$unwanted_count")
-    else
-        files_to_delete=""
-    fi
-
-    if [[ "$DRY_RUN" == true ]]; then
-        if [[ "$snapshot_count" -gt 0 ]]; then
-            print_warning "DRY-RUN: Would clean up ${BOLD_YELLOW}${snapshot_count}${RESET} old snapshot(s) (~${snapshot_mb} MB)"
-            if [[ -n "$snapshots_to_delete" ]]; then
-                print_detail "Old snapshots that would be deleted:"
-                while IFS= read -r snapshot; do
-                    local basename
-                    basename=$(basename "$snapshot")
-                    print_detail "  ${CROSS_MARK} Would delete snapshot: ${DIM}${basename}${RESET}"
-                done <<< "$snapshots_to_delete"
-            fi
-        fi
-
-        if [[ "$unwanted_count" -gt 0 ]]; then
-            print_warning "DRY-RUN: Would clean up ${BOLD_YELLOW}${unwanted_count}${RESET} non-whitelisted file(s) (~${unwanted_mb} MB)"
-            if [[ -n "$files_to_delete" ]]; then
-                print_detail "Non-whitelisted files that would be deleted (showing first 10):"
-                echo "$files_to_delete" | head -n 10 | while IFS= read -r file; do
-                    local basename
-                    basename=$(basename "$file")
-                    print_detail "  ${CROSS_MARK} Would delete file: ${DIM}${basename}${RESET}"
-                done
-                if [[ "$unwanted_count" -gt 10 ]]; then
-                    print_detail "  ${DIM}... and $((unwanted_count - 10)) more${RESET}"
-                fi
-            fi
-        fi
-
-        print_info "Total space that would be freed: ~${total_mb} MB"
-        return 0
-    fi
-
-    # Actually delete items
-    local deleted_snapshots=0
-    local deleted_files=0
-
-    # Delete old snapshot directories
-    if [[ "$snapshot_count" -gt 0 ]]; then
-        print_warning "Cleaning up ${BOLD_YELLOW}${snapshot_count}${RESET} old snapshot(s) (~${snapshot_mb} MB)..."
-        while IFS= read -r snapshot; do
-            if [[ -d "$snapshot" ]]; then
-                rm -rf "$snapshot" && deleted_snapshots=$((deleted_snapshots+1))
-            fi
-        done <<< "$snapshots_to_delete"
-    fi
-
-    # Delete non-whitelisted files
-    if [[ "$unwanted_count" -gt 0 ]]; then
-        print_warning "Cleaning up ${BOLD_YELLOW}${unwanted_count}${RESET} non-whitelisted file(s) (~${unwanted_mb} MB)..."
-        while IFS= read -r file; do
-            if [[ -f "$file" ]]; then
-                rm -f "$file" && deleted_files=$((deleted_files+1))
-            fi
-        done <<< "$files_to_delete"
-    fi
-
-    # Report results
-    local success=true
-    if [[ "$snapshot_count" -gt 0 ]] && [[ $deleted_snapshots -eq $snapshot_count ]]; then
-        print_success "Cleaned up ${deleted_snapshots} old snapshot(s), freed ~${snapshot_mb} MB"
-    elif [[ "$snapshot_count" -gt 0 ]]; then
-        print_warning "Cleaned up ${deleted_snapshots}/${snapshot_count} old snapshots"
-        success=false
-    fi
-
-    if [[ "$unwanted_count" -gt 0 ]] && [[ $deleted_files -eq $unwanted_count ]]; then
-        print_success "Cleaned up ${deleted_files} non-whitelisted file(s), freed ~${unwanted_mb} MB"
-    elif [[ "$unwanted_count" -gt 0 ]]; then
-        print_warning "Cleaned up ${deleted_files}/${unwanted_count} non-whitelisted files"
-        success=false
-    fi
-
-    if [[ "$snapshot_count" -gt 0 ]] || [[ "$unwanted_count" -gt 0 ]]; then
-        print_info "Total space freed: ~${total_mb} MB"
-    fi
-
-    if $success; then
-        return 0
-    else
-        return 1
-    fi
-}
-
-# ============================================================================
-# VERIFICATION FUNCTIONS - Model Status Checking
-# ============================================================================
-
-# Get actual disk usage for a model's files
-get_model_disk_usage() {
-    local model_files="$1"
-
-    if [[ -z "$model_files" ]]; then
-        echo "0"
-        return
-    fi
-
-    local total_bytes=0
-    while IFS= read -r file_path; do
-        if [[ -f "$file_path" ]] || [[ -L "$file_path" ]]; then
-            local file_size
-            # Use -L to follow symlinks (HuggingFace uses symlinks to blobs)
-            # Try Linux stat first, then macOS stat
-            if stat -L -c "%s" "$file_path" >/dev/null 2>&1; then
-                file_size=$(stat -L -c "%s" "$file_path" 2>/dev/null)
-            elif stat -L -f "%z" "$file_path" >/dev/null 2>&1; then
-                file_size=$(stat -L -f "%z" "$file_path" 2>/dev/null)
-            else
-                file_size=0
-            fi
-            total_bytes=$((total_bytes + file_size))
-        fi
-    done <<< "$model_files"
-
-    echo "$total_bytes"
-}
-
-# Format bytes to human-readable size
-format_bytes() {
-    local bytes="$1"
-
-    if (( bytes < 1024 )); then
-        echo "${bytes} B"
-    elif (( bytes < 1048576 )); then
-        echo "$(( bytes / 1024 )) KB"
-    elif (( bytes < 1073741824 )); then
-        printf "%.2f MB" "$(bc <<< "scale=2; $bytes / 1048576")"
-    else
-        printf "%.2f GB" "$(bc <<< "scale=2; $bytes / 1073741824")"
-    fi
-}
-
-# Verify if model is downloaded
-verify_model_download() {
-    local repo_id="$1"
-    local expected_size_gb="$2"
-    local filename_filter="$3"
-
-    # Find model files in cache
-    # Capture both stdout (file paths) and stderr (error messages)
-    local find_output
-    find_output=$(find_model_files "$repo_id" "$filename_filter" 2>&1)
-
-    # Separate file paths from error/debug messages
-    local model_files
-    model_files=$(echo "$find_output" | grep -v "^ERROR:" | grep -v "^WARN:" | grep -v "^DEBUG:")
-
-    # Extract error messages for logging
-    local error_msgs
-    error_msgs=$(echo "$find_output" | grep "^ERROR:\|^WARN:\|^DEBUG:")
-
-    if [[ -z "$model_files" ]]; then
-        # Log error messages to stderr if they exist
-        if [[ -n "$error_msgs" ]]; then
-            echo "$error_msgs" >&2
-        fi
-        echo "NOT_FOUND|0|0|"
-        return 1
-    fi
-
-    # Count files
-    local file_count
-    file_count=$(echo "$model_files" | wc -l | tr -d ' ')
-
-    # Get actual size
-    local actual_bytes
-    actual_bytes=$(get_model_disk_usage "$model_files")
-
-    # Get cache path (first file's directory)
-    local cache_path
-    cache_path=$(echo "$model_files" | head -n1 | xargs dirname)
-
-    # Get modification time of first file
-    local mod_time="Unknown"
-    if [[ -n "$model_files" ]]; then
-        local first_file
-        first_file=$(echo "$model_files" | head -n1)
-        if [[ -f "$first_file" ]]; then
-            # Try Linux stat first (most common), then macOS stat
-            if stat -c "%y" "$first_file" >/dev/null 2>&1; then
-                mod_time=$(stat -c "%y" "$first_file" 2>/dev/null | cut -d'.' -f1)
-            elif stat -f "%Sm" "$first_file" >/dev/null 2>&1; then
-                mod_time=$(stat -f "%Sm" -t "%Y-%m-%d %H:%M:%S" "$first_file" 2>/dev/null)
-            fi
-        fi
-    fi
-
-    echo "FOUND|${actual_bytes}|${file_count}|${cache_path}|${mod_time}"
-    return 0
-}
-
-# Verify model symlinks
-verify_model_links() {
-    local repo_id="$1"
-    local filename_filter="$2"
-    local file_mappings="$3"
-
-    local total_links=0
-    local valid_links=0
-    local broken_links=0
-    local link_details=""
-
-    # Check if explicit file mappings exist
-    if [[ -n "$file_mappings" ]]; then
-        # Process each file mapping to verify links
-        while IFS='|' read -r source_pattern dest_path; do
-            if [[ -z "$source_pattern" ]] || [[ -z "$dest_path" ]]; then
-                continue
-            fi
-
-            total_links=$((total_links + 1))
-            local link_path="${COMFYUI_DIR}/${dest_path}"
-
-            if [[ -L "$link_path" ]]; then
-                # Symlink exists, check if it's valid
-                if [[ -e "$link_path" ]]; then
-                    valid_links=$((valid_links + 1))
-                    local link_target
-                    link_target=$(readlink "$link_path")
-                    link_details="${link_details}VALID|${dest_path}|${link_target}\n"
-                else
-                    broken_links=$((broken_links + 1))
-                    local link_target
-                    link_target=$(readlink "$link_path")
-                    link_details="${link_details}BROKEN|${dest_path}|${link_target}\n"
-                fi
-            else
-                link_details="${link_details}MISSING|${dest_path}|\n"
-            fi
-        done <<< "$file_mappings"
-    else
-        # No explicit mappings, check automatic prefixed filenames
-        local model_files
-        model_files=$(find_model_files "$repo_id" "$filename_filter" 2>/dev/null)
-
-        if [[ -z "$model_files" ]]; then
-            echo "NOT_DOWNLOADED|0|0|0"
+        if [[ "$match" == false ]]; then
+            return 1
+        fi
+    fi
+
+    # Check repo_id filter
+    if [[ -n "$REPO_ID_FILTER" ]]; then
+        local match=false
+        IFS=',' read -ra repos <<< "$REPO_ID_FILTER"
+        for repo in "${repos[@]}"; do
+            repo=$(echo "$repo" | xargs)
+            if [[ "$repo_id" == "$repo" ]]; then
+                match=true
+                break
+            fi
+        done
+        if [[ "$match" == false ]]; then
             return 1
         fi
-
-        local model_name
-        model_name=$(echo "$repo_id" | sed 's/.*\///')
-
-        while IFS= read -r source_file; do
-            if [[ -f "$source_file" ]]; then
-                local filename
-                filename=$(basename "$source_file")
-                local prefixed_filename="${model_name}-${filename}"
-
-                total_links=$((total_links + 1))
-                local link_path="${target_dir}/${prefixed_filename}"
-
-                if [[ -L "$link_path" ]]; then
-                    if [[ -e "$link_path" ]]; then
-                        valid_links=$((valid_links + 1))
-                        local link_target
-                        link_target=$(readlink "$link_path")
-                        link_details="${link_details}VALID|${prefixed_filename}|${link_target}\n"
-                    else
-                        broken_links=$((broken_links + 1))
-                        local link_target
-                        link_target=$(readlink "$link_path")
-                        link_details="${link_details}BROKEN|${prefixed_filename}|${link_target}\n"
-                    fi
-                else
-                    link_details="${link_details}MISSING|${prefixed_filename}|\n"
-                fi
-            fi
-        done <<< "$model_files"
     fi
 
-    echo -e "CHECKED|${total_links}|${valid_links}|${broken_links}\n${link_details}"
     return 0
 }
 
-# Verify models by category
-verify_category() {
-    local category="$1"
-    local category_display="$2"
+# ============================================================================
+# DOWNLOAD FUNCTIONS
+# ============================================================================
 
-    print_section "${category_display}"
+download_file() {
+    local repo_id="$1"
+    local source="$2"
 
-    # Get models for this category
-    local models_data
-    models_data=$(parse_yaml "$CONFIG_FILE" "$category")
+    # Convert repo_id to cache path (replace / with --)
+    local cache_repo_dir="${CACHE_DIR}/${repo_id}"
+    local source_dir
+    source_dir=$(dirname "$source")
+    local output_dir="${cache_repo_dir}"
+    if [[ "$source_dir" != "." ]]; then
+        output_dir="${cache_repo_dir}/${source_dir}"
+    fi
+    local filename
+    filename=$(basename "$source")
+    local output_path="${output_dir}/${filename}"
 
-    if [[ -z "$models_data" ]]; then
-        print_warning "No models found in category: ${category}"
+    print_detail "File: ${BOLD_WHITE}${source}${RESET}"
+    print_detail "Output: ${CYAN}${output_path}${RESET}"
+
+    # Check if already exists
+    if [[ -f "$output_path" ]]; then
+        local size
+        size=$(du -h "$output_path" | cut -f1)
+        print_success "Already downloaded: ${filename} (${size})"
         return 0
     fi
 
-    local total_models
-    total_models=$(echo "$models_data" | wc -l)
-    local current=0
-    local models_downloaded=0
-    local models_missing=0
-    local models_linked=0
-    local models_broken=0
-    local models_not_linked=0
-    local total_size_bytes=0
-    local expected_size_bytes=0
+    # Dry-run mode
+    if [[ "$DRY_RUN" == true ]]; then
+        print_info "DRY-RUN: Would download ${BOLD_WHITE}${source}${RESET}"
+        return 0
+    fi
 
-    while IFS='|' read -r repo_id description size_gb essential filename; do
-        current=$((current+1))
+    # Create output directory
+    mkdir -p "$output_dir"
 
-        echo ""
-        print_step "$current" "$total_models" "${BOLD_MAGENTA}${description}${RESET}"
-        print_detail "Repository: ${BOLD_WHITE}${repo_id}${RESET}"
-        print_detail "Category: ${CYAN}${category}${RESET}"
-        print_detail "Expected Size: ${BOLD_YELLOW}${size_gb} GB${RESET}"
+    # Build download URL
+    local url="https://huggingface.co/${repo_id}/resolve/main/${source}"
+    print_detail "Downloading from HuggingFace..."
 
-        expected_size_bytes=$((expected_size_bytes + $(echo "$size_gb * 1073741824" | bc | cut -d'.' -f1)))
+    # Download with curl (with resume support)
+    local curl_args=(-L -C - --progress-bar -o "$output_path")
+    if [[ -n "$HF_TOKEN" ]]; then
+        curl_args+=(-H "Authorization: Bearer ${HF_TOKEN}")
+    fi
 
-        # Verify download status
-        echo ""
-        local download_result
-        download_result=$(verify_model_download "$repo_id" "$size_gb" "$filename")
-        local download_status
-        download_status=$(echo "$download_result" | cut -d'|' -f1)
-
-        if [[ "$download_status" == "FOUND" ]]; then
-            local actual_bytes
-            actual_bytes=$(echo "$download_result" | cut -d'|' -f2)
-            local file_count
-            file_count=$(echo "$download_result" | cut -d'|' -f3)
-            local cache_path
-            cache_path=$(echo "$download_result" | cut -d'|' -f4)
-            local mod_time
-            mod_time=$(echo "$download_result" | cut -d'|' -f5)
-
-            total_size_bytes=$((total_size_bytes + actual_bytes))
-            local actual_size_human
-            actual_size_human=$(format_bytes "$actual_bytes")
-
-            print_success "Download Status: ${BOLD_GREEN}DOWNLOADED${RESET}"
-            print_detail "${DIM}Path: ${cache_path}${RESET}"
-            print_detail "${DIM}Actual Size: ${actual_size_human} (${actual_bytes} bytes)${RESET}"
-            print_detail "${DIM}Files: ${file_count} file(s)${RESET}"
-            print_detail "${DIM}Modified: ${mod_time}${RESET}"
-
-            # Check for size mismatch
-            local expected_bytes
-            expected_bytes=$(echo "$size_gb * 1073741824" | bc | cut -d'.' -f1)
-            local size_diff_pct
-            size_diff_pct=$(echo "scale=2; (($actual_bytes - $expected_bytes) / $expected_bytes) * 100" | bc | sed 's/^\./0./')
-            local abs_size_diff_pct
-            abs_size_diff_pct=${size_diff_pct#-}
-
-            if (( $(echo "$abs_size_diff_pct > 10" | bc -l) )); then
-                print_warning "Size mismatch: ${size_diff_pct}% difference from expected"
-            fi
-
-            models_downloaded=$((models_downloaded+1))
-
-            # Verify link status
-            echo ""
-            local file_mappings
-            file_mappings=$(parse_file_mappings "$CONFIG_FILE" "$category" "$repo_id")
-            local link_result
-            link_result=$(verify_model_links "$repo_id" "$filename" "$file_mappings")
-            local first_line
-            first_line=$(echo -e "$link_result" | head -n1)
-            local link_status
-            link_status=$(echo "$first_line" | cut -d'|' -f1)
-
-            if [[ "$link_status" == "CHECKED" ]]; then
-                local total_links
-                total_links=$(echo "$first_line" | cut -d'|' -f2)
-                local valid_links
-                valid_links=$(echo "$first_line" | cut -d'|' -f3)
-                local broken_links
-                broken_links=$(echo "$first_line" | cut -d'|' -f4)
-
-                if [[ $broken_links -gt 0 ]]; then
-                    print_warning "Link Status: ${BOLD_YELLOW}${broken_links} BROKEN LINK(S)${RESET}"
-                    models_broken=$((models_broken+1))
-                elif [[ $valid_links -eq $total_links ]] && [[ $total_links -gt 0 ]]; then
-                    print_success "Link Status: ${BOLD_GREEN}LINKED${RESET} (${valid_links}/${total_links})"
-                    models_linked=$((models_linked+1))
-                else
-                    print_warning "Link Status: ${BOLD_YELLOW}PARTIALLY LINKED${RESET} (${valid_links}/${total_links})"
-                    models_not_linked=$((models_not_linked+1))
-                fi
-
-                # Show link details
-                local link_details
-                link_details=$(echo -e "$link_result" | tail -n +2)
-                if [[ -n "$link_details" ]]; then
-                    while IFS='|' read -r link_state link_name link_target; do
-                        if [[ -z "$link_state" ]]; then
-                            continue
-                        fi
-
-                        case "$link_state" in
-                            VALID)
-                                print_detail "${LINK} ${BOLD_GREEN}✓${RESET} ${DIM}${link_name}${RESET}"
-                                ;;
-                            BROKEN)
-                                print_detail "${LINK} ${BOLD_RED}✗${RESET} ${DIM}${link_name}${RESET} ${BOLD_RED}(BROKEN)${RESET}"
-                                ;;
-                            MISSING)
-                                print_detail "${LINK} ${BOLD_YELLOW}○${RESET} ${DIM}${link_name}${RESET} ${BOLD_YELLOW}(NOT LINKED)${RESET}"
-                                ;;
-                        esac
-                    done <<< "$link_details"
-                fi
-            else
-                print_error "Link Status: ${BOLD_RED}NOT LINKED${RESET}"
-                models_not_linked=$((models_not_linked+1))
-            fi
-        else
-            print_error "Download Status: ${BOLD_RED}NOT DOWNLOADED${RESET}"
-            models_missing=$((models_missing+1))
-            echo ""
-            print_info "Link Status: ${DIM}N/A (model not downloaded)${RESET}"
+    if curl "${curl_args[@]}" "$url" 2>&1; then
+        if [[ -f "$output_path" ]] && [[ -s "$output_path" ]]; then
+            local size
+            size=$(du -h "$output_path" | cut -f1)
+            print_success "Downloaded ${BOLD_WHITE}${filename}${RESET} (${size})"
+            return 0
         fi
+    fi
 
-        show_progress "$current" "$total_models"
-    done <<< "$models_data"
-
-    echo -e "\n"
-
-    # Category summary
-    local total_size_human
-    total_size_human=$(format_bytes "$total_size_bytes")
-    local expected_size_human
-    expected_size_human=$(format_bytes "$expected_size_bytes")
-
-    print_info "Category Summary:"
-    echo -e "  ${BOLD_WHITE}Total Models:${RESET}       ${total_models}"
-    echo -e "  ${BOLD_GREEN}✓ Downloaded:${RESET}       ${models_downloaded} ($(( models_downloaded * 100 / total_models ))%)"
-    echo -e "  ${BOLD_RED}✗ Missing:${RESET}          ${models_missing} ($(( models_missing * 100 / total_models ))%)"
-    echo -e "  ${BOLD_GREEN}✓ Properly Linked:${RESET}  ${models_linked}"
-    echo -e "  ${BOLD_YELLOW}⚠ Broken Links:${RESET}     ${models_broken}"
-    echo -e "  ${BOLD_YELLOW}○ Not Linked:${RESET}       ${models_not_linked}"
-    echo -e "  ${BOLD_CYAN}📊 Disk Usage:${RESET}      ${total_size_human} / ${expected_size_human} expected"
-
-    # Return statistics for global summary (format: downloaded|missing|linked|broken|not_linked|total_size|expected_size)
-    echo "${models_downloaded}|${models_missing}|${models_linked}|${models_broken}|${models_not_linked}|${total_size_bytes}|${expected_size_bytes}" > /tmp/verify_stats_${category}
+    print_error "Failed to download ${source}"
+    rm -f "$output_path" 2>/dev/null || true
+    return 1
 }
 
-# Process models by category
-process_category() {
-    local category="$1"
-    local category_display="$2"
+download_model() {
+    local config="$1"
+    local index="$2"
+    local repo_id="$3"
+    local description="$4"
 
-    print_section "${category_display}"
+    print_detail "Repository: ${BOLD_WHITE}${repo_id}${RESET}"
+    [[ -n "$description" ]] && print_detail "Description: ${description}"
 
-    # Get models for this category
-    local models_data
-    models_data=$(parse_yaml "$CONFIG_FILE" "$category")
+    local files_count
+    files_count=$(get_files_count "$config" "$index")
 
-    if [[ -z "$models_data" ]]; then
-        print_warning "No models found in category: ${category}"
-        return 0
+    if [[ "$files_count" == "0" ]]; then
+        print_warning "No files defined for ${repo_id}"
+        return 1
     fi
 
-    local total_models
-    total_models=$(echo "$models_data" | wc -l)
-    local current=0
     local succeeded=0
     local failed=0
 
-    while IFS='|' read -r repo_id description size_gb essential filename; do
-        current=$((current+1))
+    for ((f=0; f<files_count; f++)); do
+        local source
+        source=$(get_file_field "$config" "$index" "$f" "source")
 
-        echo ""
-        print_step "$current" "$total_models" "${BOLD_MAGENTA}${description}${RESET}"
-
-        local success=true
-
-        # Download if command is 'download' or 'both'
-        if [[ "$COMMAND" == "download" ]] || [[ "$COMMAND" == "both" ]]; then
-            if ! download_model "$repo_id" "$description" "$size_gb"; then
-                success=false
-            fi
+        if [[ -z "$source" ]]; then
+            continue
         fi
 
-        # Link if command is 'link' or 'both'
-        if [[ "$COMMAND" == "link" ]] || [[ "$COMMAND" == "both" ]]; then
-            if $success; then
-                # Parse file mappings from YAML for this model
-                local file_mappings
-                file_mappings=$(parse_file_mappings "$CONFIG_FILE" "$category" "$repo_id")
-
-                # Pass file mappings to link_model (empty string if no mappings found)
-                if ! link_model "$repo_id" "$filename" "$file_mappings"; then
-                    success=false
-                fi
-
-                # Cleanup unused cache files if enabled
-                if [[ "$CLEANUP_MODE" == true ]] && $success; then
-                    cleanup_unused_cache_files "$repo_id" "$CACHE_DIR" "$COMFYUI_DIR" "$file_mappings"
-                fi
-            fi
-        fi
-
-        if $success; then
+        if download_file "$repo_id" "$source"; then
             succeeded=$((succeeded+1))
         else
             failed=$((failed+1))
         fi
+    done
 
-        show_progress "$current" "$total_models"
-    done <<< "$models_data"
-
-    echo -e "\n"
-    print_info "Category Summary: ${BOLD_GREEN}${succeeded} succeeded${RESET}, ${BOLD_RED}${failed} failed${RESET}"
+    if [[ $failed -eq 0 ]]; then
+        return 0
+    else
+        return 1
+    fi
 }
 
-# Display verification summary
-display_verification_summary() {
-    local start_time="$1"
-    local end_time="$2"
-    local categories="$3"
+# ============================================================================
+# LINK FUNCTIONS
+# ============================================================================
 
-    local duration=$((end_time - start_time))
-    local minutes=$((duration / 60))
-    local seconds=$((duration % 60))
+link_file() {
+    local repo_id="$1"
+    local source="$2"
+    local dest="$3"
 
-    # Aggregate statistics from all categories
-    local total_models=0
-    local total_downloaded=0
-    local total_missing=0
-    local total_linked=0
-    local total_broken=0
-    local total_not_linked=0
-    local total_size_bytes=0
-    local expected_size_bytes=0
+    local cache_repo_dir="${CACHE_DIR}/${repo_id}"
+    local source_path="${cache_repo_dir}/${source}"
+    local link_path="${OUTPUT_DIR}/${dest}"
+    local link_dir
+    link_dir=$(dirname "$link_path")
 
-    while IFS= read -r category; do
-        if [[ -f "/tmp/verify_stats_${category}" ]]; then
-            local stats
-            stats=$(cat "/tmp/verify_stats_${category}")
-            local downloaded
-            downloaded=$(echo "$stats" | cut -d'|' -f1)
-            local missing
-            missing=$(echo "$stats" | cut -d'|' -f2)
-            local linked
-            linked=$(echo "$stats" | cut -d'|' -f3)
-            local broken
-            broken=$(echo "$stats" | cut -d'|' -f4)
-            local not_linked
-            not_linked=$(echo "$stats" | cut -d'|' -f5)
-            local size_bytes
-            size_bytes=$(echo "$stats" | cut -d'|' -f6)
-            local expected_bytes
-            expected_bytes=$(echo "$stats" | cut -d'|' -f7)
+    print_detail "Source: ${CYAN}${source_path}${RESET}"
+    print_detail "Target: ${CYAN}${link_path}${RESET}"
 
-            total_models=$((total_models + downloaded + missing))
-            total_downloaded=$((total_downloaded + downloaded))
-            total_missing=$((total_missing + missing))
-            total_linked=$((total_linked + linked))
-            total_broken=$((total_broken + broken))
-            total_not_linked=$((total_not_linked + not_linked))
-            total_size_bytes=$((total_size_bytes + size_bytes))
-            expected_size_bytes=$((expected_size_bytes + expected_bytes))
+    # Check if source exists
+    if [[ ! -f "$source_path" ]]; then
+        print_warning "Source file not found: ${source}"
+        return 1
+    fi
 
-            rm -f "/tmp/verify_stats_${category}"
-        fi
-    done <<< "$categories"
+    # Dry-run mode
+    if [[ "$DRY_RUN" == true ]]; then
+        print_info "DRY-RUN: Would link ${BOLD_WHITE}${source}${RESET} → ${dest}"
+        return 0
+    fi
 
-    local total_size_human
-    total_size_human=$(format_bytes "$total_size_bytes")
-    local expected_size_human
-    expected_size_human=$(format_bytes "$expected_size_bytes")
+    # Create target directory
+    mkdir -p "$link_dir"
 
-    print_banner "VERIFICATION COMPLETE"
+    # Remove existing symlink
+    if [[ -L "$link_path" ]]; then
+        rm -f "$link_path"
+    elif [[ -e "$link_path" ]]; then
+        print_warning "File exists (not a symlink): ${dest}"
+        return 1
+    fi
 
-    echo -e "${BOLD_CYAN}${STAR} Global Summary${RESET}"
-    echo -e "${CYAN}$(printf '%.0s'"${BOX_DOUBLE}" $(seq 1 80))${RESET}"
-    echo -e "  ${BOLD_WHITE}Total Models:${RESET}              ${total_models}"
-    echo ""
-    echo -e "  ${BOLD_GREEN}✓ Downloaded:${RESET}              ${total_downloaded} ($(( total_models > 0 ? total_downloaded * 100 / total_models : 0 ))%)"
-    echo -e "  ${BOLD_RED}✗ Missing:${RESET}                 ${total_missing} ($(( total_models > 0 ? total_missing * 100 / total_models : 0 ))%)"
-    echo ""
-    echo -e "  ${BOLD_GREEN}✓ Properly Linked:${RESET}         ${total_linked} ($(( total_models > 0 ? total_linked * 100 / total_models : 0 ))%)"
-    echo -e "  ${BOLD_YELLOW}⚠  Broken Links:${RESET}            ${total_broken} ($(( total_models > 0 ? total_broken * 100 / total_models : 0 ))%)"
-    echo -e "  ${BOLD_YELLOW}○ Not Linked:${RESET}              ${total_not_linked} ($(( total_models > 0 ? total_not_linked * 100 / total_models : 0 ))%)"
-    echo ""
-    echo -e "  ${BOLD_CYAN}📊 Disk Space Used:${RESET}        ${total_size_human} / ${expected_size_human} expected"
-    echo -e "  ${BOLD_WHITE}Cache Directory:${RESET}           ${CYAN}${CACHE_DIR}${RESET}"
-    echo -e "  ${BOLD_WHITE}ComfyUI Directory:${RESET}         ${CYAN}${COMFYUI_DIR}${RESET}"
-    echo -e "  ${BOLD_WHITE}Duration:${RESET}                  ${BOLD_YELLOW}${minutes}m ${seconds}s${RESET}"
-    echo -e "${CYAN}$(printf '%.0s'"${BOX_DOUBLE}" $(seq 1 80))${RESET}"
+    # Create symlink
+    ln -s "$source_path" "$link_path"
+    print_success "Linked: ${LINK_ICON} ${dest}"
+    return 0
+}
 
-    # Provide actionable suggestions
-    if [[ $total_missing -gt 0 ]] || [[ $total_broken -gt 0 ]] || [[ $total_not_linked -gt 0 ]]; then
-        echo -e "\n${BOLD_YELLOW}${WARNING} Issues Found - Suggested Actions:${RESET}\n"
+link_model() {
+    local config="$1"
+    local index="$2"
+    local repo_id="$3"
 
-        if [[ $total_missing -gt 0 ]]; then
-            echo -e "  ${BOLD_RED}✗${RESET} ${total_missing} model(s) not downloaded"
-            echo -e "    ${DIM}Fix:${RESET} ${CYAN}$0 download -c ${CONFIG_FILE}${RESET}"
+    print_detail "Repository: ${BOLD_WHITE}${repo_id}${RESET}"
+
+    local files_count
+    files_count=$(get_files_count "$config" "$index")
+
+    if [[ "$files_count" == "0" ]]; then
+        print_warning "No files defined for ${repo_id}"
+        return 1
+    fi
+
+    local succeeded=0
+    local failed=0
+
+    for ((f=0; f<files_count; f++)); do
+        local source dest
+        source=$(get_file_field "$config" "$index" "$f" "source")
+        dest=$(get_file_field "$config" "$index" "$f" "dest")
+
+        if [[ -z "$source" ]]; then
+            continue
         fi
 
-        if [[ $total_broken -gt 0 ]] || [[ $total_not_linked -gt 0 ]]; then
-            echo -e "  ${BOLD_YELLOW}⚠${RESET}  $(( total_broken + total_not_linked )) model(s) with link issues"
-            echo -e "    ${DIM}Fix:${RESET} ${CYAN}$0 link -c ${CONFIG_FILE}${RESET}"
+        # If dest is empty, use source as dest
+        if [[ -z "$dest" ]]; then
+            dest="$source"
         fi
 
-        if [[ $total_missing -gt 0 ]] && [[ $(( total_broken + total_not_linked )) -gt 0 ]]; then
-            echo -e "  ${BOLD_CYAN}ℹ${RESET}  Fix everything in one go:"
-            echo -e "    ${CYAN}$0 both -c ${CONFIG_FILE}${RESET}"
+        if link_file "$repo_id" "$source" "$dest"; then
+            succeeded=$((succeeded+1))
+        else
+            failed=$((failed+1))
         fi
+    done
+
+    if [[ $failed -eq 0 ]]; then
+        return 0
+    else
+        return 1
+    fi
+}
+
+# ============================================================================
+# VERIFY FUNCTIONS
+# ============================================================================
+
+verify_model() {
+    local config="$1"
+    local index="$2"
+    local repo_id="$3"
+
+    echo -e "  ${BOLD_WHITE}${repo_id}${RESET}"
+
+    local files_count
+    files_count=$(get_files_count "$config" "$index")
+
+    if [[ "$files_count" == "0" ]]; then
+        echo -e "    ${BOLD_YELLOW}No files defined${RESET}"
+        return 1
+    fi
+
+    for ((f=0; f<files_count; f++)); do
+        local source dest
+        source=$(get_file_field "$config" "$index" "$f" "source")
+        dest=$(get_file_field "$config" "$index" "$f" "dest")
+
+        if [[ -z "$source" ]]; then
+            continue
+        fi
+        if [[ -z "$dest" ]]; then
+            dest="$source"
+        fi
+
+        local cache_path="${CACHE_DIR}/${repo_id}/${source}"
+        local link_path="${OUTPUT_DIR}/${dest}"
+
+        local cache_status="${BOLD_RED}${CROSS_MARK}${RESET}"
+        local link_status="${BOLD_RED}${CROSS_MARK}${RESET}"
+
+        # Check cache
+        if [[ -f "$cache_path" ]]; then
+            local size
+            size=$(du -h "$cache_path" | cut -f1)
+            cache_status="${BOLD_GREEN}${CHECK_MARK}${RESET} (${size})"
+        fi
+
+        # Check symlink
+        if [[ -L "$link_path" ]]; then
+            if [[ -e "$link_path" ]]; then
+                link_status="${BOLD_GREEN}${CHECK_MARK}${RESET}"
+            else
+                link_status="${BOLD_YELLOW}${WARNING}${RESET} (broken)"
+            fi
+        fi
+
+        local filename
+        filename=$(basename "$source")
+        echo -e "    ${DIM}${filename}${RESET}"
+        echo -e "      Cache: ${cache_status}"
+        echo -e "      Link:  ${link_status}"
+    done
+}
+
+# ============================================================================
+# MAIN WORKFLOW
+# ============================================================================
+
+process_models() {
+    local action="$1"
+
+    print_section "Processing Models (${action})"
+
+    local total
+    total=$(get_model_count "$CONFIG_FILE")
+
+    if [[ "$total" == "0" ]]; then
+        print_warning "No models found in configuration"
+        return 0
+    fi
+
+    local processed=0
+    local succeeded=0
+    local failed=0
+    local skipped=0
+
+    for ((i=0; i<total; i++)); do
+        local repo_id category description
+        repo_id=$(get_model_field "$CONFIG_FILE" "$i" "repo_id")
+        category=$(get_model_field "$CONFIG_FILE" "$i" "category")
+        description=$(get_model_field "$CONFIG_FILE" "$i" "description")
+
+        # Skip entries without repo_id (might be malformed YAML)
+        if [[ -z "$repo_id" ]]; then
+            continue
+        fi
+
+        # Apply filters
+        if ! matches_filters "$repo_id" "$category"; then
+            skipped=$((skipped+1))
+            continue
+        fi
+
+        processed=$((processed+1))
         echo ""
-    else
-        echo -e "\n${BOLD_GREEN}${SPARKLES} All models verified successfully! ${SPARKLES}${RESET}\n"
+        print_step "$processed" "$((total-skipped))" "${BOLD_CYAN}${repo_id}${RESET}"
+
+        local result=0
+        case "$action" in
+            download)
+                download_model "$CONFIG_FILE" "$i" "$repo_id" "$description" || result=1
+                ;;
+            link)
+                link_model "$CONFIG_FILE" "$i" "$repo_id" || result=1
+                ;;
+            verify)
+                verify_model "$CONFIG_FILE" "$i" "$repo_id"
+                ;;
+        esac
+
+        if [[ $result -eq 0 ]]; then
+            succeeded=$((succeeded+1))
+        else
+            failed=$((failed+1))
+        fi
+    done
+
+    echo ""
+    if [[ $skipped -gt 0 ]]; then
+        print_info "Skipped ${skipped} model(s) (filtered)"
     fi
+    print_info "Summary: ${BOLD_GREEN}${succeeded} succeeded${RESET}, ${BOLD_RED}${failed} failed${RESET}"
 }
 
-# Display summary
-display_summary() {
-    local start_time="$1"
-    local end_time="$2"
-    local total_downloaded="$3"
-    local total_failed="$4"
+show_help() {
+    cat << 'EOF'
+HuggingFace Model Downloader - A Beautiful CLI Tool
 
-    local duration=$((end_time - start_time))
-    local minutes=$((duration / 60))
-    local seconds=$((duration % 60))
+Usage: artifact_huggingface_download.sh [COMMAND] [OPTIONS]
 
-    print_banner "DOWNLOAD COMPLETE"
+Commands:
+  download    Download models to cache directory (default)
+  link        Create symlinks from cache to output directory
+  verify      Verify model status (cache and symlinks)
 
-    echo -e "${BOLD_CYAN}${STAR} Summary${RESET}"
-    echo -e "${CYAN}$(printf '%.0s'"${BOX_LIGHT}" $(seq 1 80))${RESET}"
-    echo -e "  ${BOLD_WHITE}Total Downloaded:${RESET} ${BOLD_GREEN}${total_downloaded}${RESET} models"
-    echo -e "  ${BOLD_WHITE}Total Failed:${RESET}     ${BOLD_RED}${total_failed}${RESET} models"
-    echo -e "  ${BOLD_WHITE}Cache Directory:${RESET}  ${CYAN}${CACHE_DIR}${RESET}"
-    echo -e "  ${BOLD_WHITE}Duration:${RESET}         ${BOLD_YELLOW}${minutes}m ${seconds}s${RESET}"
-    echo -e "${CYAN}$(printf '%.0s'"${BOX_LIGHT}" $(seq 1 80))${RESET}"
+Options:
+  -c, --config FILE         Configuration YAML file (required)
+  --cache-dir DIR           Cache directory
+  --output-dir DIR          Output/installation directory
+  --category CAT1,CAT2      Filter by category (comma-separated)
+  --repo-id ID1,ID2         Filter by repo_id (comma-separated)
+  --auth-token TOKEN        HuggingFace token (or set HF_TOKEN env var)
+  -n, --dry-run             Show what would be done without making changes
+  -h, --help                Show this help message
 
-    if [[ $total_failed -eq 0 ]]; then
-        echo -e "\n${BOLD_GREEN}${SPARKLES} All models downloaded successfully! ${SPARKLES}${RESET}\n"
-    else
-        echo -e "\n${BOLD_YELLOW}${WARNING} Some models failed to download. Check logs above.${RESET}\n"
-    fi
+Environment Variables:
+  HF_TOKEN                  HuggingFace API token
+  CACHE_DIR                 Override default cache directory
+  OUTPUT_DIR                Override default output directory
+
+YAML Configuration Format:
+  - repo_id: organization/model-name
+    description: Model description
+    category: image|video|llm|upscale
+    files:
+      - source: path/to/file.safetensors
+        dest: subdir/output_filename.safetensors
+
+Examples:
+  # Download all models from config
+  ./artifact_huggingface_download.sh download -c models_huggingface.yaml
+
+  # Download only image models
+  ./artifact_huggingface_download.sh download -c models_huggingface.yaml --category image
+
+  # Download specific model
+  ./artifact_huggingface_download.sh download -c models_huggingface.yaml --repo-id black-forest-labs/FLUX.1-schnell
+
+  # Create symlinks for downloaded models
+  ./artifact_huggingface_download.sh link -c models_huggingface.yaml
+
+  # Verify all models
+  ./artifact_huggingface_download.sh verify -c models_huggingface.yaml
+
+  # Dry-run to preview operations
+  ./artifact_huggingface_download.sh download -c models_huggingface.yaml --dry-run
+
+EOF
 }
 
-# ============================================================================
-# MAIN FUNCTION
-# ============================================================================
-
 main() {
-    local start_time
-    start_time=$(date +%s)
+    # Check for yq
+    check_yq
 
-    # Display beautiful banner
-    if [[ "$COMMAND" == "verify" ]]; then
-        print_banner "${SPARKLES} ComfyUI Model Verification ${SPARKLES}"
-        echo -e "${BOLD_CYAN}Comprehensive Model Status Check${RESET}"
-        echo -e "${DIM}Verify Downloads & Links ${LINK} Configuration-Driven ${STAR}${RESET}\n"
-    else
-        print_banner "${ROCKET} ComfyUI Model Downloader ${ROCKET}"
-        echo -e "${BOLD_CYAN}A Beautiful CLI Tool for Downloading AI Models${RESET}"
-        echo -e "${DIM}Powered by HuggingFace ${LINK} Configuration-Driven ${STAR}${RESET}\n"
-    fi
+    # Load token from .env files
+    load_env_token
 
-    # Show dry-run warning banner
+    # Display banner
+    print_banner "${ROCKET} HuggingFace Model Downloader ${SPARKLES}"
+
+    # Show dry-run warning
     if [[ "$DRY_RUN" == true ]]; then
         echo -e "${BOLD_YELLOW}${WARNING}  DRY-RUN MODE - No changes will be made ${WARNING}${RESET}\n"
     fi
 
-    # Check dependencies
-    check_dependencies
-
     # Validate configuration
-    validate_config
+    print_section "Configuration"
 
-    # Get all categories (or filtered categories if --category is specified)
     if [[ -z "$CONFIG_FILE" ]]; then
-        print_error "No configuration file specified. Use -c/--config to provide one."
+        print_error "Configuration file required. Use -c/--config"
         exit 1
     fi
 
-    local categories
-    categories=$(validate_and_get_categories "$CONFIG_FILE")
-
-    if [[ -z "$categories" ]]; then
-        print_error "No model categories found in configuration"
+    if [[ ! -f "$CONFIG_FILE" ]]; then
+        print_error "Configuration file not found: $CONFIG_FILE"
         exit 1
     fi
 
-    # Handle verify command separately
-    if [[ "$COMMAND" == "verify" ]]; then
-        # Process each category with verification
-        while IFS= read -r category; do
-            # Get category display name (capitalize and add spaces)
-            local category_display
-            category_display=$(echo "$category" | sed 's/_/ /g' | sed 's/\b\(.\)/\u\1/g')
+    print_success "Config: ${CYAN}${CONFIG_FILE}${RESET}"
+    print_success "Cache: ${CYAN}${CACHE_DIR}${RESET}"
+    print_success "Output: ${CYAN}${OUTPUT_DIR}${RESET}"
+    print_success "Command: ${BOLD_CYAN}${COMMAND}${RESET}"
 
-            verify_category "$category" "$category_display"
-        done <<< "$categories"
-
-        # Display verification summary
-        local end_time
-        end_time=$(date +%s)
-        display_verification_summary "$start_time" "$end_time" "$categories"
+    # Show token status (not required for public repos)
+    if [[ -n "$HF_TOKEN" ]]; then
+        print_success "HF Token: ${DIM}${HF_TOKEN:0:8}...${RESET}"
     else
-        # Original download/link logic
-        local total_succeeded=0
-        local total_failed=0
-
-        # Process each category
-        while IFS= read -r category; do
-            # Get category display name (capitalize and add spaces)
-            local category_display
-            category_display=$(echo "$category" | sed 's/_/ /g' | sed 's/\b\(.\)/\u\1/g')
-
-            process_category "$category" "$category_display"
-
-            # Update counters (this is simplified, you'd need to track actual numbers)
-            total_succeeded=$((total_succeeded+1))
-        done <<< "$categories"
-
-        # Display summary
-        local end_time
-        end_time=$(date +%s)
-        display_summary "$start_time" "$end_time" "$total_succeeded" "$total_failed"
+        print_info "HF Token: ${DIM}not set (public repos only)${RESET}"
     fi
+
+    # Show filters if set
+    [[ -n "$CATEGORY_FILTER" ]] && print_info "Category filter: ${BOLD_WHITE}${CATEGORY_FILTER}${RESET}"
+    [[ -n "$REPO_ID_FILTER" ]] && print_info "Repo ID filter: ${BOLD_WHITE}${REPO_ID_FILTER}${RESET}"
+
+    # Process based on command
+    process_models "$COMMAND"
+
+    # Final message
+    echo ""
+    print_banner "${SPARKLES} Complete ${SPARKLES}"
 }
 
 # ============================================================================
-# ENTRY POINT
+# ARGUMENT PARSING
 # ============================================================================
 
-# Parse command line arguments
-POSITIONAL_ARGS=()
 while [[ $# -gt 0 ]]; do
     case $1 in
         -c|--config)
@@ -1589,58 +719,32 @@ while [[ $# -gt 0 ]]; do
             CACHE_DIR="$2"
             shift 2
             ;;
-        --comfyui-dir)
-            COMFYUI_DIR="$2"
+        --output-dir)
+            OUTPUT_DIR="$2"
             shift 2
             ;;
-        --category|--cat)
+        --category)
             CATEGORY_FILTER="$2"
             shift 2
             ;;
-        --cleanup|--clean)
-            CLEANUP_MODE=true
-            shift
+        --repo-id)
+            REPO_ID_FILTER="$2"
+            shift 2
             ;;
-        --dry-run|-n)
+        --auth-token)
+            HF_TOKEN="$2"
+            shift 2
+            ;;
+        -n|--dry-run)
             DRY_RUN=true
             shift
             ;;
-        download|link|both|verify)
+        download|link|verify)
             COMMAND="$1"
             shift
             ;;
         -h|--help)
-            echo "Usage: $0 [COMMAND] [options]"
-            echo ""
-            echo "Commands:"
-            echo "  download              Download models only"
-            echo "  link                  Create symlinks only (models must already be downloaded)"
-            echo "  both                  Download and create symlinks (default)"
-            echo "  verify                Verify model downloads and links (read-only)"
-            echo ""
-            echo "Options:"
-            echo "  -c, --config FILE     Configuration file (required)"
-            echo "  --cache-dir DIR       Cache directory (default: auto-detect)"
-            echo "                        RunPod: /workspace/huggingface_cache"
-            echo "                        Local: ~/.cache/huggingface"
-            echo "  --comfyui-dir DIR     ComfyUI models directory (default: auto-detect)"
-            echo "                        RunPod: /workspace/ComfyUI/models"
-            echo "                        Local: ~/ComfyUI/models"
-            echo "  -h, --help            Show this help message"
-            echo ""
-            echo "Examples:"
-            echo "  $0 download -c models.yaml"
-            echo "  $0 link --comfyui-dir /opt/ComfyUI/models -c models.yaml"
-            echo "  $0 both -c models.yaml --cache-dir /data/hf-cache"
-            echo "  $0 verify -c models.yaml"
-            echo ""
-            echo "Verify Command:"
-            echo "  The verify command performs comprehensive checks on all models:"
-            echo "    - Download status (file existence, size, location)"
-            echo "    - Link status (valid, broken, or missing symlinks)"
-            echo "    - Size mismatch warnings (actual vs expected)"
-            echo "    - Disk space usage statistics"
-            echo "    - Actionable fix suggestions"
+            show_help
             exit 0
             ;;
         -*)
@@ -1649,16 +753,14 @@ while [[ $# -gt 0 ]]; do
             exit 1
             ;;
         *)
-            POSITIONAL_ARGS+=("$1")
+            # Positional argument - treat as config file
+            if [[ -z "$CONFIG_FILE" ]]; then
+                CONFIG_FILE="$1"
+            fi
             shift
             ;;
     esac
 done
 
-# Handle positional argument (config file path)
-if [[ ${#POSITIONAL_ARGS[@]} -gt 0 ]]; then
-    CONFIG_FILE="${POSITIONAL_ARGS[0]}"
-fi
-
-# Run main function
+# Run main
 main