;3h<ddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZddlmZddlmZmZddlmZdd lmZddlZdd lmZmZdd lmZerdd lm Z Gd de Z!eGdde!Z"eGdde!Z#eGdde!Z$GddZ%GddZ&y)) annotationsN)ABCabstractmethod)Sequence)Lock)Path) TYPE_CHECKINGAny)utils)document)r wasm_utils)LikeData) ComponentcHeZdZdZeddZe d ddZy)FlaggingCallbackz[ An abstract class for defining the methods that any FlaggingCallback should have. cy)a This method should be overridden and ensure that everything is set up correctly for flag(). This method gets called once at the beginning of the Interface.launch() method. Parameters: components: Set of components that will provide flagged data. flagging_dir: A string, typically containing the path to the directory where the flagging file should be stored (provided as an argument to Interface.__init__()). Nself components flagging_dirs O/var/www/html/audio-gradio/venv/lib/python3.12/site-packages/gradio/flagging.pysetupzFlaggingCallback.setup s Ncy)a This method should be overridden by the FlaggingCallback subclass and may contain optional additional arguments. This gets called every time the button is pressed. Parameters: interface: The Interface object that is being used to launch the flagging interface. flag_data: The data to be flagged. flag_option (optional): In the case that flagging_options are provided, the flag option that is being used. username (optional): The username of the user that is flagging the data, if logged in. Returns: (int) The total number of samples that have been flagged. Nr)r flag_data flag_optionusernames rflagzFlaggingCallback.flag+s$ r)rSequence[Component]rstrNNrz list[Any]r str | Nonerr$returnint)__name__ __module__ __qualname____doc__rrrrrrrrs]  #'#         rrc:eZdZdZdZddZ d ddZy) SimpleCSVLoggera A simplified implementation of the FlaggingCallback abstract class provided for illustrative purposes. Each flagged sample (both the input and output data) is logged to a CSV file on the machine running the gradio app. Example: import gradio as gr def image_classifier(inp): return {'cat': 0.3, 'dog': 0.7} demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label", flagging_callback=SimpleCSVLogger()) cyNrrs r__init__zSimpleCSVLogger.__init__N rcN||_||_tj|dyNTexist_okrrosmakedirsrs rrzSimpleCSVLogger.setupQs$( L40rNc|j}t|dz }g}t|j|dD]z\}}t|t j |j t|j ndz } | jd|j|j|| |t|ddd 5} tj| } | jtj |dddt|d 5} t#t%tj&| d z } ddd| S#1swYKxYw#1swY SxYw) Nlog.csvFstrictTr4autf-8encodingnewlinerA)rrzipr client_utils!strip_invalid_filename_characterslabelr!mkdirappendropencsvwriterwriterowr sanitize_list_for_csvlenlistreader) rrrrr log_filepathcsv_data componentsamplesave_dircsvfilerM line_counts rrzSimpleCSVLogger.flagVs@ (( L)I5 !$T__i!N  Iv>>(1(CIOO$H NNDN ) OO  ,gr B CgZZ(F OOE77A B C, 1 r?rBrArCrD)rrexists enumeratergetattrrErFrGr!r is_prop_updaterJrr_simplify_file_data_in_strdatetimenowrKrLrMrNrOrPrQrR)rrrrrrSis_newidxrUheadersrTrVrWdatarXrMrYs rrzClassicCSVLogger.flagsQ (( L)I5 ,'..00#,DOO"< Y Iw - C:cU1C C   (1 5 9)  & $C$)V>>GIw5K:cU9KLH ##F+F ,)NN6HN= ** ::4@D%% &&  $H$8bAH--11345 ,R' B CgZZ(F ; ;G DE OOE77A B  C , 1 2I;>JT)r_boolrZr"r#r[rrrr]r]wsV 51'1!1#'# 22 2 2 2rr]c`eZdZdZ d ddZ d dZd d dZ d d dZy) CSVLoggera8 The default implementation of the FlaggingCallback abstract class in gradio>=5.0. Each flagged sample (both the input and output data) is logged to a CSV file with headers on the machine running the gradio app. Unlike ClassicCSVLogger, this implementation is concurrent-safe and it creates a new dataset file every time the headers of the CSV (derived from the labels of the components) change. It also only creates columns for "username" and "flag" if the flag_option and username are provided, respectively. Example: import gradio as gr def image_classifier(inp): return {'cat': 0.3, 'dog': 0.7} demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label", flagging_callback=CSVLogger()) Guides: using-flagging Nc||_||_||_tjst |_yt j|_y)a  Parameters: simplify_file_data: If True, the file data will be simplified before being written to the CSV file. If CSVLogger is being used to cache examples, this is set to False to preserve the original FileData class verbose: If True, prints messages to the console about the dataset file creation dataset_file_name: The name of the dataset file to be created (should end in ".csv"). If None, the dataset file will be named "dataset1.csv" or the next available number. N) r_verbosedataset_file_namer IS_WASMr contextlib nullcontextlock)rr_rwrxs rr0zCSVLogger.__init__sB#5 !2$,,DF 2<2H2H2J rc@||_t||_d|_y)NT)rrr first_timers rrzCSVLogger.setups % .rc  tj|jd|g}t|jDcgc]\}}t |ddxsd|c}}|zdgz}t j|}tt|jjd}|jr|j|jz |_ n|r t|d}ttj d |j"d }t%|d d 5}t'j(|} t+| d} ddd |k7r|dz} |jd| dz |_ n||_ n|jdz |_ t|jj/st%|jdd d 5}t'j0|} | j3t j|ddd|j4rt7d|jyy|j4rt7d|jyycc}}w#1swYxYw#t,$r|jdz |_ YwxYw#1swYxYw)NTr4rHrbrcz dataset*.csvcZttjd|jdS)N\d+r)r&refindallstem)fs rz0CSVLogger._create_dataset_file.. s RZZ5OPQ5R1Sr)keyrrr=r?rfrDdatasetz.csvz dataset1.csvwzCreated dataset file at:zUsing existing dataset file at:)r7r8rrhrrir rOrQrglobrxdataset_filepathmaxr&rrrrKrLrRnext ExceptionrgrMrNrwprint) radditional_headersrorUrp dataset_files latest_file latest_numrXrRexisting_headersnew_numrMs r_create_dataset_filezCSVLogger._create_dataset_filesn D%%5  %!# '0&@ "C 7D1Gz#5GG !  !    --g6T$"3"3499.IJ  ! !$($5$58N8N$ND !  K!!'S !FK4D4D!Ea!HI +rGD: ZZ0F'+FD'9$:$w.(1nG,0,=,=''RV@W,WD),7D)%)$5$5$FD !D))*113%%sB FG, ; ;G DE  F ||0$2G2GH \\ 3T5J5J KW *:: K(,(9(9N(J% K  F Fs<IAI!"I2I!:JII!!JJJ c |jrAg}||jd||jd|j|d|_g}tt |j |dD]\}\}}|j tjtt|ddxsd|z } tj|r|jt|v||j|| nd } |jrtj| } |j| ||j|||j||jtt j j#|j$5t'|j(d d d 5} t+j,| } | j/tj0|dddt'|j(d 5} t3t5t+j6| dz } dddddd S#1swY]xYw#1swYxYw#1swY SxYw)Nrr)rFr;rHrbrdr=r>r?rfrCrD)r~rJrrhrErrrFrGr!rir rjrr_rkrlrmr|rKrrLrMrNrOrPrQrR)rrrrrrTrorUrVrWrqrXrMrYs rrzCSVLogger.flag(s@ ??!# &"))&1#"))*5  % %9K % L#DO(1 5 9)  & $C$)V!!@@ 7D9Oz#=OP  ##F+F ,)NN6HN= ** ::4@D%' &*  " OOK (   OOH %H--11345 YY @%%sB GG, ; ;H EF  G d++g> @' cjj&9!:;a?  @  @ G G  @ @  @s<I"4:I .I" +I8I" I I"I I""I,)TTN)r_rsrwrsrxr$rZr.)rzlist[str] | Noner"r#)r'r(r)r*r0rrrrrrrurus$$((,     &  &'!2Ln#'# 44 4 4 4rruc.eZdZdZdZddZ ddZy) ChatCSVLoggerz Flagging callback for chat conversations. Flagged conversations and like/dislike reactions are logged to a CSV file on the machine running the gradio app. cyr.rr/s rr0zChatCSVLogger.__init__er1rc@||_tj|dyr3)rr7r8)rrs rrzChatCSVLogger.setuphs( L40rcb|j}t|dz }t|j }|jdurdn|jdurdn |j}t j ||j |ttjjg}t|ddd 5}|r(tj|} | jgd tj|} | jtj|dddy#1swYyxYw) Nr:TLikeFDisliker>r?r=r@) conversationindexvaluerrc)rrrglikedjsondumpsrr!rlrmrKrLrMrNr rO) r like_datamessagesrrSrnfeedbackrTrXrMs rrzChatCSVLogger.flagls  (( L)I5 ,'..00$& %'  JJx OO  !!%%' (  ,gr B CgG, WXZZ(F OOE77A B  C C Cs 8A$D%%D.N)rr!)rrrrQr[rrrrr_s+  1CCCrrc8eZdZdZ d ddZddZdZy) FlagMethodz Helper class that contains the flagging options and calls the flagging method. Also provides visual feedback to the user when flag is clicked. cJ||_||_||_d|_||_y)NFlag)flagging_callbackrHrr'visual_feedback)rrrHrrs rr0zFlagMethod.__init__s)"3   .rcH |jjt||j|j|jsytjd|jS#t $r)}t d||jrYd}~yYd}~_d}~wwxYw)N)rrzError while flagging: zError!g?) rrrQrrrrrtimesleepreset)rrequestres r__call__zFlagMethod.__call__s  " " ' 'YTZZ'BRBR ( ##  3zz| *1#. /##$ s;A// B!8BB!cDtj|jdS)NT)r interactive)grButtonrHr/s rrzFlagMethod.resetsyytzzt<rs"  #$ %/0$"+" s" J 3&3 3l K'K K\ U U Up)C)CX"="=r