o 0i@sRdZddlZddlZddlZddlmZmZmZddlZddl m Z m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZee Zejdd d eDed Zd eed eefddZeddZejdgddddddddej ddf deeeej!dedfdeeeej!ddfdeeeej!ddfdeeeej!ddfd eeeej!d!dfd"ed#ed$ed%ed dfd&d'Z"ejd(gd)dd*ej ddfd+eeej#d,dfd"ed#ed$ed%ed df d-d.Z$ejd/d0gd d9d1eeej#d2dfd%ed dfd3d4Z%ejd5d6gd d9d1eeej#d2dfd%ed dfd7d8Z&dS):aContains commands to interact with papers on the Hugging Face Hub. Usage: # list daily papers (most recently submitted) hf papers ls # list trending papers hf papers ls --sort=trending # list papers from a specific date, ordered by upvotes hf papers ls --date=2025-01-23 # list today's papers, ordered by upvotes hf papers ls --date=today # list papers from a specific week hf papers ls --week=2025-W09 # list papers by a specific submitter hf papers ls --submitter=someuser # search papers hf papers search "vision language" # get info about a paper hf papers info 2502.08025 # read a paper as markdown hf papers read 2502.08025 N) AnnotatedOptionalget_args)CLIErrorHfHubHTTPError)DailyPapersSort_T) FormatOptLimitOpt OutputFormatQuietOptTokenOpt _format_cellapi_object_to_dict get_hf_apiprint_list_output typer_factory PaperSortEnumcCsi|]}||qSr).0srrT/home/kim/smarthome/.venv/lib/python3.10/site-packages/huggingface_hub/cli/papers.py Fsr)typevaluereturncCs*|durdS|dkrtjS|S)z6Parse date option, converting 'today' to current date.Ntoday)lowerdatetimedater isoformat)rrrr _parse_dateIs  r!z Interact with papers on the Hub.)helpz list | ls)z hf papers lszhf papers ls --sort trendingzhf papers ls --date 2025-01-23zhf papers ls --week 2025-W09z hf papers ls --submitter akhaliqzhf papers ls --format json)Zexamples2Frz+Date in ISO format (YYYY-MM-DD) or 'today'.)r"callbackweekz'ISO week to filter by, e.g. '2025-W09'.monthz;Month to filter by in ISO format (YYYY-MM), e.g. '2025-02'. submitterz$Filter by username of the submitter.sortz Sort results.limitformatquiettokenc  Csvt|d} |r |jnd} dd| j||||| |dD} gd} dtdttfd d } t| ||d | | d d d ddS)zList daily papers on the Hub.r,NcSg|]}t|qSrrr paper_inforrr szpapers_ls..)rr%r&r'r(r))idtitleupvotescomments published_at submitted_byitemrc Ssr|dpi}|dp|dpd}|ddt|ddddt|d dt|d dt|d d|gS) Nr8fullnameusernamer3r4<max_lenr5r6r7getrstr)r9r8Zsubmitter_namerrr _paper_rows zpapers_ls.._paper_rowr3right)r5r6r*r+Zid_keyheadersZrow_fnZ alignments)rrZlist_daily_papersdictlistrBr)rr%r&r'r(r)r*r+r,apisort_keyresults_HEADERSrCrrr papers_lsUs. )   rMsearch)z"hf papers search "vision language"z1hf papers search "attention mechanism" --limit 10z*hf papers search "diffusion" --format jsonqueryzSearch query string.c Cs^t|d}dd|j||dD}gd}dtdttfdd }t|||d ||d d id dS)zSearch papers on the Hub.r-cSr.rr/r0rrrr2sz!papers_search..)rPr))r3r4r5r7r9rcSs<|ddt|ddddt|ddt|ddgS)Nr3r<r4Fr>r5r7r@)r9rrrrCs z!papers_search.._paper_rowr3r5rDrEN)rZ list_papersrGrHrBr) rPr)r*r+r,rIrKrLrCrrr papers_searchs  rRinfozhf papers info 2601.15621paper_idz'The arXiv paper ID (e.g. '2502.08025').c Csnt|d}z|j|d}Wnty)}z|jjdkr$td|d|d}~wwttjt |dddS) zs  0       D