j<dZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZddlmZd ZGd d e ZGd d ej ZdZdZdZdZdS)zFWake Gradio queues on new work and reuse SSE across chained callbacks.N) TimeoutError)wraps)Queue) FunctionTypeSimpleNamespace)queueingroutes)iter_app_routesc t|ji|jdt|tji|j|j|jS)Nasyncio)sleepCancelledError) r__code__ __globals__rr r__name__ __defaults__ __closure__)fnr s OC:\Users\johnn\OneDrive\Documents\AI\shared\gradio\gradio_queue_wakeup_patch.py _with_sleeprs  &H&H ?Y^ovpFDGDGDG&H&HJLJUWYWfhjhv w wwc.eZdZfdZfdZdZxZS) MessageQueuecttj|_tj|_dSN)super__init__r get_running_loop_loopEvent _availableself __class__s rrzMessageQueue.__init__s: -// !-//rctj|i||js&|j|jjdSdSr)rputr is_closedcall_soon_threadsafer!setr#argskwargsr$s rr&zMessageQueue.putsb T$V$$$z##%% A J + +DO,? @ @ @ @ @ A ArcK|j|sdS tj|j|d{VdS#t $rYdSwxYw)N)timeout)r!clearemptyr wait_forwaitrr#delays rr2zMessageQueue.waits zz||  F "4?#7#7#9#95III I I I I I I I I I    DD s3A(( A65A6)r __module__ __qualname__rr&r2 __classcell__r$s@rrrse***** AAAAA       rrcReZdZfdZdZfdZdZfdZfdZfdZ xZ S) WakeupQueuecxtj|i|tj|_d|_dSr)rrr r _work_ready _work_loopr*s rrzWakeupQueue.__init__,s8$)&)))"=??rczK|jd{V|jdSr)r<r2r/r3s r_wait_for_workzWakeupQueue._wait_for_work1sH##%%%%%%%%%      rcKtj|_tj|_t j}t||j}||d{VdSr) r rr=r r<rstart_processingrr?)r#original schedulerr$s rrAzWakeupQueue.start_processing5sk!244"=??77+ $*=>> ioorc|j|jdS|js&|j|jjdSdSr)r=r<r)r'r()r#s r _notify_workzWakeupQueue._notify_work?sj ? "   " " " " "**,, G O 0 01A1E F F F F F G GrcKtj|i|d{V}|dr||S)Nr)rpushrE)r#r+r,resultr$s rrGzWakeupQueue.pushEsV#uww|T4V44444444 !9       rcK tj|i|d{V |S#|wxYwr)rprocess_eventsrEr*s rrJzWakeupQueue.process_eventsKsg //@@@@@@@@@ @       D      s 9Acpt|dSr)rcloserEr"s rrLzWakeupQueue.closeQs,   r) rr5r6rr?rArErGrJrLr7r8s@rr:r:+s !!!GGG       rr:crK|jjdr|jddkrdS|j}|jrdS|j| dd{Vt|j |o|j S)Nz /queue/data stream_reuse1Fg?) urlpathendswith query_paramsget get_blocks_queuestoppedpending_messages_per_sessionr2boolpending_event_ids_session)apprequest session_hashqueues r_reuse_idle_streamr_Vs ;  $ $] 3 3w7K7O7OP^7_7_cf7f7fu NN   #E }u  ,\ : ? ? E EEEEEEEE /33LAA B B X5=GXXrc.tjtjt j|}dGfddtj}||dkrtd|j j }tjdd |dj d}g|j tj tj|jtj  |_ tjtj|gg }i|jd t(i}t+t-||j jd ||dd|jpdDS)NrceZdZfdZdS)'_with_stream_reuse..ReuseStreamct|jtjr[t|jjtjr7|jjjdkr"dz gtjdj|S|S)NCloseStreamMessagezEif await _reuse_idle_stream(app, request, session_hash): continue) isinstancevalueastCallfuncNameidparsebody)r#nodematchess r visit_Assignz4_with_stream_reuse..ReuseStream.visit_Assignjs$*ch// yJtzPSPX4Y4Y y^b^h^m^puI_I_I1 x#kllqxswxxKrN)rr5r6rq)rpsr ReuseStreamrbis.       rrrrez>Gradio stream reuse requires the pinned Gradio 5.29 SSE helperz def factory(z, z ): pass)rlctx)rg)rn type_ignoresr_execfactoryc3$K|] }|jV dSr) cell_contents).0cells r z%_with_stream_reuse..zs%!V!V$"4!V!V!V!V!V!Vr)rhrmtextwrapdedentinspect getsourceNodeTransformervisit RuntimeErrorr co_freevarsjoinrnReturnrkrLoadfix_missing_locationsModulerr_rucompile co_filenamer)rtreerrnamesrvmodule namespacerps @r_with_stream_reuserbs 9X_W%6r%:%:;; < L#79KLLI0& 9 99EEE 9Y !V!VAUSU!V!V!V WWrcztD]}t|dd}t|ts)t |jj|jpdD]^\}}|dkr |j}t|ddr%t|}t||ffd }d|_ ||__dS)Nendpointr|queue_data_helper_wangp_message_wakeupFcXKfd}t||||d{VS)NcKjj}||d{VdSr)rUrVrXr2)r4messagesr[r]s rwait_for_messagezL_patch_message_delivery..queue_data_helper..wait_for_messagesL"~~//6ST`aH"--...........r)r)r\r] process_msg _originalrr[s ` rrz2_patch_message_delivery..queue_data_helpers[//////F[4DEEg|]hiiiiiiiiirT) r getattrrfrziprrrrxrrr)r[routernamerzrBrs` r_patch_message_deliveryr}s!%%335*d33(L11  h/;X=Q=WUWXX 3 3JD$***)Hx!8%@@ )(33H 8__V^ j j j j j_ j 7;  3!2D  ! 3 33rctjturdStt_tt_t jjtfd}t|t j_dS)Nc6|i|}t||Sr)r)r+r,r[original_create_apps r create_appzinstall..create_apps+!!42622$$$ r) rrr:r ThreadQueuer Apprr staticmethod)rrs @rinstallrsv~$$ HN'H */   )44FJr)__doc__rhr rr}r functoolsrr^rtypesrrgradiorr shared.utils.fastapi_routesr rrr:r_rrrr|rrrs\LL  ////////########777777www5.((((((.(((V Y Y YXXX63334 5 5 5 5 5r