i>TdZddlZddlZddlmZddlmZddlmZee jdz Z dZ dgd gd d d d dggdd dgdgdd dgdgdd dgdgdd dgdgdd dgdgdd dgd gd!d d"gd#gdd d$gd#gdd d%gd&gd'd d( Z id)d*d+id,d*d-id.d*d/id0d*d1id2d*d3id4d*d5id6d*d7id8d*d9id:d*d;idd*d?id@d*dAidBd*dCidDd*dEidFd*dGidHd*dIiZgdJgdKdLZeej#ZGdMdNZdOZdPZdQZde fdRZe fdSZy)Tz@ Design Search Core - BM25 search engine for UI/UX style guides N) defaultdict)log)Pathdataz styles.csv)Style CategoryKeywordsBest ForTypeAI Prompt Keywords)rr r zPrimary ColorszEffects & Animationr uLight Mode ✓u Dark Mode ✓ Performance AccessibilityzFramework Compatibility Complexityr zCSS/Technical KeywordszImplementation ChecklistzDesign System Variables)file search_cols output_colsz colors.csv Product TypeNotes)rPrimaryz On Primary Secondaryz On SecondaryAccentz On Accent Background ForegroundCardzCard ForegroundMutedzMuted ForegroundBorder DestructivezOn DestructiveRingrz charts.csv) Data Typer Best Chart Type When to UseWhen NOT to UseAccessibility Notes) rr r zSecondary Optionsr!r"zData Volume ThresholdzColor GuidancezAccessibility Grader#z A11y FallbackzLibrary RecommendationzInteractive Levelz landing.csv) Pattern Namer Conversion Optimization Section Order)r$r r&zPrimary CTA PlacementzColor Strategyr%z products.csv)rr Primary Style RecommendationzKey Considerations)rr r'zSecondary StyleszLanding Page PatternzDashboard Style (if applicable)zColor Palette Focuszux-guidelines.csv)CategoryIssue DescriptionPlatform) r(r)r+r*DoDon'tzCode Example GoodzCode Example BadSeverityztypography.csv)Font Pairing Namer(Mood/Style Keywordsr Heading Font Body Font) r/r(r1r2r0r Google Fonts URLz CSS ImportzTailwind Configrz icons.csv)r( Icon Namer r )r(r4r Libraryz Import CodeUsager Stylezreact-performance.csv)r(r)r r*zapp-interface.csvzgoogle-fonts.csv)Familyr(StrokeClassificationsr Subsets Designers) r8r(r9r:Stylesz Variable Axesr;r<zPopularity Rankr3) stylecolorchartlandingproductux typographyiconsreactweb google-fontsrFrzstacks/react.csvnextjszstacks/nextjs.csvvuezstacks/vue.csvsveltezstacks/svelte.csvastrozstacks/astro.csvswiftuizstacks/swiftui.csvz react-nativezstacks/react-native.csvflutterzstacks/flutter.csvnuxtjszstacks/nuxtjs.csvznuxt-uizstacks/nuxt-ui.csvz html-tailwindzstacks/html-tailwind.csvshadcnzstacks/shadcn.csvzjetpack-composezstacks/jetpack-compose.csvthreejszstacks/threejs.csvangularzstacks/angular.csvlaravelzstacks/laravel.csv)r( Guideliner*r,r-) r(rTr*r,r-z Code GoodzCode Badr.zDocs URL)rrc*eZdZdZddZdZdZdZy)BM25z&BM25 ranking algorithm for text searchc||_||_g|_g|_d|_i|_t t|_d|_ y)Nr) k1bcorpus doc_lengthsavgdlidfrint doc_freqsN)selfrXrYs I/home/jay/workspace/.worktrees/task-2116-dev1/tools/design-search/core.py__init__z BM25.__init__s@  $S)ctjddt|j}|j Dcgc]}t |dkDs|c}Scc}w)z8Lowercase, split, remove punctuation, filter short wordsz[^\w\s] )resubstrlowersplitlen)ratextws rbtokenizez BM25.tokenizesCvvj#s4y'89::<6a3q6A:666s AAc|Dcgc]}|j|c}|_t|j|_|jdk(ry|jDcgc] }t|c}|_t |j|jz |_|jD]@}t}|D]/}||vs|j|xxdz cc<|j|1B|jjD]6\}}t|j|z dz|dzz dz|j|<8ycc}wcc}w)zBuild BM25 index from documentsrNg?) rprZrmr`r[sumr\setr_additemsrr])ra documentsdocseenwordfreqs rbfitzBM25.fits5>?ct}}S)? T[[! 66Q; 04 <CH<))*TVV3 ;; #C5D #t#NN4(A-(HHTN # #....0 KJD$ $&&4-#"5$*!E!IJDHHTN K@=s EEc|j|}g}t|jD]\}}d}|j|}t t }|D]} || xxdz cc<|D]~} | |j vs|| } |j | } | |jdzz} | |jd|jz |j|z|jz zzz}|| | z|z z }|j||ft|ddS)z!Score all documents against queryrrrc |dS)Nrr)xs rbzBM25.score..0s AaDrdT)keyreverse) rp enumeraterZr[rr^r]rXrYr\appendsorted)raquery query_tokensscoresidxrxscoredoc_len term_freqsrztokentfr] numerator denominators rbrz BM25.scores }}U+ !$++. (HCE&&s+G$S)J &4 A%  && ;DHH$#E*B((5/C "dggk 2I"$tww!dff*tvv?ORVR\R\?\2\']"]KS9_{::E  ; MM3, ' ("f.$??rdN)g?g?)__name__ __module__ __qualname____doc__rcrpr|rrrdrbrVrVs07 K&@rdrVct|dd5}ttj|cdddS#1swYyxYw)z!Load CSV and return list of dictsrzutf-8)encodingN)openlistcsv DictReader)filepathfs rb _load_csvr4s6 hg .'!CNN1%&'''s 7Ac |jsgSt|}|Dcgc]djfd|D}}t}|j ||j |} g} | d|D]E\} } | dkDs || | j |D cic]} | vs| j| dc} G| Scc}wcc} w)zCore search function using BM25rfc3TK|]}tj|d!yw)N)rjget).0colrows rb z_search_csv..Bs!GC#cggc2./Gs%(Nrr)existsrjoinrVr|rrr)rrrr max_resultsrrrwbm25rankedresultsrrrs ` rb _search_csvr:s ??  X DTXXCG;GGXIX 6DHHY ZZ FG\k*Y U 19s)C NN[WcCSVJCb!11W XY NYXs!C C $C c |jgdgdgdgdgdgdgdgdgd gd gd d }|jDcic]\}}|tfd |Dc}}tfd}|dkDr|SdScc}}w)z/Auto-detect the most relevant domain from query) r?palettehex#rgbrsemanticaccent destructivemuted foreground) r@graph visualizationtrendbarpiescatterheatmapfunnel)rApagecta conversionhero testimonialpricingsection)Rsaas ecommercez e-commercefintech healthcaregaming portfoliocrypto dashboardfitness restauranthoteltravelmusic educationlearninglegal insurancemedicalbeautypharmacydentalpetdatingweddingrecipedeliveryridebookingcalendartimertrackerdiarynotechat messengercrminvoiceparkingtransitvpnalarmweathersleep meditationfastinghabitgrocerymemewardrobez plant carereading flashcardpuzzletriviaarcade photography streamingpodcast newsletter marketplace freelancer coworkingairlinemuseumtheaterchurchz non-profitcharity kindergartendaycarez senior care veterinaryfloristbakerybrewery construction automotivez real estate logistics agriculturezcoding bootcamp)r>designui minimalism glassmorphism neumorphism brutalismz dark modeflataurorapromptcssimplementationvariable checklisttailwind) rC usability accessibilitywcagtouchscroll animationkeyboard navigationmobile)z font pairingztypography pairingz heading fontz body font)z google fontz font familyz font weightz font stylez variable fontnotozfont forz find fontz font subsetz font languagezmonospace fontz serif fontzsans serif fontz display fontzhandwriting fontfontrDserifsans)iconrElucide heroiconssymbolglyph pictogramzsvg icon)rFznext.jsrIsuspensememo usecallback useeffectrerenderbundle waterfallbarrelzdynamic importrsczserver component) ariafocusoutliner virtualize autocompleteformz input type preconnect) r?r@rArBr>rCrDrHrErFrGc3K|]6}tjdtj|zdzs3d8yw)z\brrN)rhsearchescape)rkw query_lowers rbrz detect_domain..s1b"BIIebiim6Ke6SU`,aAbs4??c|S)Nr)rrs rbrzdetect_domain..s VAYrd)rrr>)rkrvrsmax)rdomain_keywordsdomainkeywordsbestrIrs @@rb detect_domainrPSs++-K  lhS h $  Z *f  StOp!0 5 5 7 FH bbbbF v. /D$rzFile not found: )errorrMrr)rMrrcountr)rP CSV_CONFIGrDATA_DIRrrrm)rrMrconfigrrs rbrFrFs ~u% ^^FJw$7 8F&.(H ?? +H:6&II(F=$96-;PRWYdeGufVnsSZ|ho pprdc|tvrdd|ddjtiStt|dz }|j sd||dSt |t dt d ||}d ||t|dt||d S) z Search stack-specific guidelinesrRzUnknown stack: z . Available: z, rzStack file not found: )rRstackrrrX)rMrXrrrSr) STACK_CONFIGrAVAILABLE_STACKSrUrr _STACK_COLSrm)rrXrrrs rb search_stackr\%s L ?5'tyyIY?Z>[\]],u-f55H ?? 1( M@Z\acnoGU#F+W  rd)rrrh collectionsrmathrpathlibr__file__parentrU MAX_RESULTSrTrYr[rkeysrZrVrrrPrFr\rrdrbrds # > 6 ) _ .&0 2  6_  i  $G    !   0Ht (G   $G    #n  W{ z f( ) v*+ F$ % v*+   f( )  ,- V67,- v*+,-f89 v*+ <=,-,- ,-! *K    ))+,;@;@~' 21D; q ,7rd