EirdUddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0m1Z1ddl2m3Z3e r*ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?ddl@mAZAdZBdeCd<ejdk(rd ndZEdeCd!<eeFZGdeCd"<Gd#d$eHZIed%&Gd'd(ZJGd)d*e ZKGd+d,eZLGd-d.ZMy)/) annotationsN) dataclassfield)Enum) TYPE_CHECKINGFinal NamedTuple)LocalComponentRegistry)BidiComponentManager) get_logger) ForwardMsg) AppSession)get_data_cache_stats_provider!get_resource_cache_stats_provider)LocalDiskCacheStorageManager)MediaFileManager)MemorySessionStorage) ScriptData) ScriptCache)ActiveSessionInfo SessionClientSessionClientDisconnectedErrorSessionManagerSessionStorage)SCRIPT_RUN_WITHOUT_ERRORS_KEYSessionStateStatProvider) StatsManager StatsProvider)WebsocketSessionManager) Awaitable)BaseComponentRegistry)BackMsg)CacheStorageManager)MediaFileStorage) UserInfoType)UploadedFileManager<rSCRIPT_RUN_CHECK_TIMEOUTwin32g?_SIGNAL_CHECK_INTERVAL_LOGGERceZdZdZy)RuntimeStoppedErrorz;Raised by operations on a Runtime instance that is stopped.N)__name__ __module__ __qualname____doc__[/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/streamlit/runtime/runtime.pyr-r-NsEr3r-T)frozenceZdZUdZded<ded<ded<eeZd ed <eeZ d ed <ee Z d ed<e Z ded<eeZded<dZded<y) RuntimeConfigz$Config options for StreamlitRuntime.str script_pathr$media_file_storager&uploaded_file_manager)default_factoryr#cache_storage_managerr!component_registryr bidi_component_registryztype[SessionManager]session_manager_classrsession_storageFboolis_helloN)r.r/r0r1__annotations__rrr=r r>r r?rr@rrArCr2r3r4r7r7Rs.)(/.2742. 16.1- 5:,51 3J/I', >}}r3c|jduS)a2True if the singleton Runtime instance has been created. When a Streamlit app is running in "raw mode" - that is, when the app is run via `python app.py` instead of `streamlit run app.py` - the Runtime will not exist, and various Streamlit functions need to adapt. N)rVrYs r4existszRuntime.existss}}D((r3ctj td|t_d|_d|_|j |_|j|_tj|_ |j|_ |j|_|j |_t%|j&|_|j*|_t/|_|jj3|j5|j6|j"|j0|j8|_t=|_|j>jAtC|j>jAtE|j"%|j>jA|j"tG|j&tHr%|j>jA|j&|j>jAtK|j:tG|j:tHr&|j>jA|j:yy)a Create a Runtime instance. It won't be started yet. Runtime is *not* thread-safe. Its public methods are generally safe to call only on the same thread that its event loop runs on. Parameters ---------- config Config options. Nz Runtime instance already exists!)storage)rAr; script_cachemessage_enqueued_callback)&rUrVrX _async_objs_loop_coroutine_taskr9_main_script_pathrC _is_hellorFrG_stater>_component_registryr?_bidi_component_registryr;_uploaded_file_mgrrr:_media_file_mgrr=_cache_storage_managerr _script_cache discover_and_register_componentsr@rA_enqueued_some_message _session_mgrr _stats_mgrregister_providerrr isinstancerr)selfconfigs r4__init__zRuntime.__init__s    (AB B 15@D!!'!3!3"** $*#<#< (.(F(F%"(">">/8Q8QR&,&B&B#(] %%FFH"88"22"&"9"9++&*&A&A 9 '. ))*G*IJ ))*K*MN  " " . OO - -d.E.E F f// ? OO - -f.G.G H ))*B4CTCT*UV d'' 7 OO - -d.?.? @ 8r3c|jSN)rfrss r4statez Runtime.states {{r3c|jSrw)rgrxs r4r>zRuntime.component_registrys'''r3c|jSrw)rhrxs r4r?zRuntime.bidi_component_registrys,,,r3c|jSrw)rirxs r4uploaded_file_mgrzRuntime.uploaded_file_mgrs&&&r3c|jSrw)rkrxs r4r=zRuntime.cache_storage_managers***r3c|jSrw)rjrxs r4media_file_mgrzRuntime.media_file_mgrs###r3c|jSrw)rprxs r4 stats_mgrzRuntime.stats_mgrs r3c6|jjS)z?A Future that completes when the Runtime's run loop has exited.)_get_async_objsrSrxs r4rSzRuntime.stopped s##%---r3cV|jj|}|y|jS)zGet the SessionClient for the given session_id, or None if no such session exists. Notes ----- Threading: SAFE. May be called on any thread. N)roget_active_session_infoclientrs session_id session_infos r4 get_clientzRuntime.get_clients/((@@L  """r3ct|jj|}||jjyy)zClear the user_info for the given session_id. Notes ----- Threading: SAFE. May be called on any thread. N)roget_session_infosessionclear_user_infors r4clear_user_info_for_sessionz#Runtime.clear_user_info_for_session$s7((99*E  #  0 0 2 $r3c Kttjtjtjtjtjtj}||_tj |jd|_|jd{y7w)aStart the runtime. This must be called only once, before any other functions are called. When this coroutine returns, Streamlit is ready to accept new sessions. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. )rNrOrPrQrRrSzRuntime.loop_coroutine)nameN) rMasyncioget_running_loopEventFuturerb create_task_loop_coroutinercrRrs async_objss r4startz Runtime.start/s"..0mmo"==?"==?NN$NN$  &$+$7$7  ")A% !    sB>CCCcljdfd }jj|y)zRequest that Streamlit close all sessions and stop running. Note that Streamlit won't stop running immediately. Notes ----- Threading: SAFE. May be called from any thread. cjtjtjhvrytj dj tjjjy)NzRuntime stopping...) rfrFrJrKr+debug _set_staterOset)rrssr4stop_on_eventloopz'Runtime.stop..stop_on_eventloopWsT{{|44l6J6JKK MM/ 0 OOL11 2  $ $ &r3NreturnNone)rrNcall_soon_threadsafe)rsrrs` @r4stopz Runtime.stopLs/))+  ' 112CDr3c8|jj|S)zTrue if the session_id belongs to an active session. Notes ----- Threading: SAFE. May be called on any thread. )rois_active_session)rsrs r4rzRuntime.is_active_sessionas  22:>>r3c|r |r td|jtjtjhvrt d|jd|j j|t|j|j|||}|jtj|jjj|S)aCreate a new session (or connect to an existing one) and return its unique ID. Parameters ---------- client A concrete SessionClient implementation for communicating with the session's client. user_info A dict that contains information about the session's user. For now, it only (optionally) contains the user's email address. { "email": "example@example.com" } existing_session_id The ID of an existing session to reconnect to. If one is not provided, a new session is created. Note that whether the Runtime's SessionManager supports reconnecting to an existing session depends on the SessionManager that this runtime is configured with. session_id_override The ID to assign to a new session being created with this method. Setting this can be useful when the service that a Streamlit Runtime is running in wants to tie the lifecycle of a Streamlit session to some other session-like object that it manages. Only one of existing_session_id and session_id_override should be set. Returns ------- str The session's unique string ID. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. z`Only one of existing_session_id and session_id_override should be set. This should never happen.zCan't connect_session (state=))r script_data user_infoexisting_session_idsession_id_override)rXrfrFrJrKr-roconnect_sessionrrdrerrIrrPr)rsrrrrrs r4rzRuntime.connect_sessionjsT #6,  ;;<00,2F2FG G%(Edkk]RS&TU U&&66"4#9#94>>J 3 3 7   CCD --113r3cVtjd|j||||S)zCreate a new session (or connect to an existing one) and return its unique ID. Notes ----- This method is simply an alias for connect_session added for backwards compatibility. z:create_session is deprecated! Use connect_session instead.)rrrr)r+warningr)rsrrrrs r4create_sessionzRuntime.create_sessions5 TU## 3 3 $  r3c|jj|}|r|jj||jy)a9Close and completely shut down a session. This differs from disconnect_session in that it always completely shuts down a session, permanently losing any associated state (session state, uploaded files, etc.). This function may be called multiple times for the same session, which is not an error. (Subsequent calls just no-op.) Parameters ---------- session_id The session's unique ID. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. N)ror close_session_on_session_disconnectedrs r4rzRuntime.close_sessions=&((99*E     + +J 7 %%'r3c|jj|}|r|jj||jy)aZDisconnect a session. It will stop producing ForwardMsgs. Differs from close_session because disconnected sessions can be reconnected to for a brief window (depending on the SessionManager/SessionStorage implementations used by the runtime). This function may be called multiple times for the same session, which is not an error. (Subsequent calls just no-op.) Parameters ---------- session_id The session's unique ID. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. N)rordisconnect_sessionrrs r4rzRuntime.disconnect_sessions=&((@@L     0 0 < %%'r3c,|jtjtjhvrt d|jd|j j |}|tjd|y|jj|y)a+Send a BackMsg to an active session. Parameters ---------- session_id The session's unique ID. msg The BackMsg to deliver to the session. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. zCan't handle_backmsg (state=rNz3Discarding BackMsg for disconnected session (id=%s)) rfrFrJrKr-rorr+rrhandle_backmsg)rsrmsgrs r4rzRuntime.handle_backmsgs ;;<00,2F2FG G%(DT[[MQR&ST T((@@L   MMEz  ++C0r3c,|jtjtjhvrt d|jd|j j |}|tjd|y|jj|y)a.Handle an Exception raised during deserialization of a BackMsg. Parameters ---------- session_id The session's unique ID. exc The Exception. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. z6Can't handle_backmsg_deserialization_exception (state=rNz=Discarding BackMsg Exception for disconnected session (id=%s)) rfrFrJrKr-rorr+rrhandle_backmsg_exception)rsrexcrs r4(handle_backmsg_deserialization_exceptionz0Runtime.handle_backmsg_deserialization_exception s ;;<00,2F2FG G%H UVW ((@@L   MMO  55c:r3cK|jtjtjtjhvryyw)N)Tok)F unavailable)rfrFrGrJrKrxs r4is_ready_for_browser_connectionz'Runtime.is_ready_for_browser_connection(s: ;;   ! !    #s>AcKtt|j|j|j|j |j ddi} |jdtj}t|jvrltj|z tkrNtjdd{t|jvrtj|z tkrNt|jvr |jy|jt}|rdnd}||f|jS7#|jwxYww) aJLoad and execute the app's script to verify it runs without an error. Returns ------- (True, "ok") if the script completes without error, or (False, err_msg) if the script raises an exception. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. emailztest@example.com)rr;r`rarNg?)Ftimeoutrerror)rrrdrerirlrn request_reruntime perf_counterr session_stater(rsleepshutdown)rsrnowrrs r4does_script_run_without_errorz%Runtime.does_script_run_without_error3s-"4#9#94>>J"&"9"9++&*&A&A 23     ! !$ '##%C-W5J5JJ&&(3.2JJmmC(((.W5J5JJ&&(3.2JJ-G4I4II'     &&'DEB$'Cs7    )    sCAEA-E?E4E5EEE5EEEEcTtjd|j|||_y)NzRuntime state: %s -> %s)r+rrf)rs new_states r4rzRuntime._set_state^s /iH r3cK|j} |jtjk(r |j tj n6|jtj k(rntd|j|jjd|jjsZ|jtj k(rtjtj|jjtj|jjftj t"d{\}}|D]}|j%|st"nj|jtj k(r|j&j)|j*j-D]S}|j.j1}|D]2} |j3||tj:dd{4Utj:dd{nntjtj|jjtj|j&jftj t"d{\}}|D]}|j%|st"Y|jjsZ|j*j=D]1}|j*j?|j.j83|j tj@|jBjdy7A#t4$r3|j*j7|j.j8YwxYw777#tD$rN} |jBjG| tIjJtLjOdYd} ~ yd} ~ wwxYww)zThe main Runtime loop. This function won't exit until `stop` is called. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. zBad Runtime state at start: N) return_whenrrg{Gz?zJ Please report this bug at https://github.com/streamlit/streamlit/issues. )(rrfrFrGrrHrIrXrR set_resultrOis_setrwaitrrPFIRST_COMPLETEDr*cancelrQclearrolist_active_sessionsrflush_browser_queue _send_messagerridr list_sessionsrrKrS Exception set_exception traceback print_excr+info) rsr done_tasks pending_taskstaskactive_session_infomsg_listrres r4rzRuntime._loop_coroutinebs))+ ^ {{l222 B BC K KK"%A$++#OPP    ) )$ / **113;;,"D"DD 7>ll#// 0D0D0I0I0KL#// 0I0I0N0N0PQ%,$;$; 671-J !.& &&*@*L [[L$O$OO--335/3/@/@/U/U/W 3+#6#>#>#R#R#T#+ 3C" $ 2 23F L#*--"222 3 3"-----3:,,++J,@,@,E,E,GH++J,E,E,J,J,LM!( 7 723-) M*"DKKM""&<&HA!**113F!% 1 1 ? ? A I !!// 0D0D0G0GH  I OOL00 1    ) )$ /I10$B" $ 1 1 D D$7$?$?$B$B!"" 3. -<     , ,Q /    ! LL   sQE O+N  BO+;N# O+%O"& O+O%B O+O(AO+BO+Q O+#8OO+OO+%O+(O++ Q4AP=8Q=QQc|jddk(r2|jtjk(r|xjdz c_|j j |y)aSend a message to a client. If the client is likely to have already cached the message, we may instead send a "reference" message that contains only the hash of the message. Parameters ---------- session_info : ActiveSessionInfo The ActiveSessionInfo associated with websocket msg : ForwardMsg The message to send to the client Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. typescript_finishedN) WhichOneofrr FINISHED_SUCCESSFULLYscript_run_countrwrite_forward_msg)rsrrs r4rzRuntime._send_messagesQ( >>& !%6 6   :#C#C C  ) )Q . ) --c2r3c|j}|jj|jjy)aCallback called by AppSession after the AppSession has enqueued a message. Sets the "needs_send_data" event, which causes our core loop to wake up and flush client message queues. Notes ----- Threading: SAFE. May be called on any thread. N)rrNrrQrrs r4rnzRuntime._enqueued_some_messages2))+ 11*2K2K2O2OPr3cH|j td|jS)zpReturn our AsyncObjects instance. If the Runtime hasn't been started, this will raise an error. zRuntime hasn't started yet!)rbrXrxs r4rzRuntime._get_async_objss(    #<= =r3c |jtjk(rf|jj dk(rH|j j j|jtjyyy)zlSet the runtime state to NO_SESSIONS_CONNECTED if the last active session was disconnected. rN) rfrFrIronum_active_sessionsrrPrrrHrxs r4rz Runtime._on_session_disconnectedse KK<FF F!!5571<  " 1 1 7 7 9 OOL>> ?= Gr3)rrU)rrB)rtr7rr)rrF)rr!)rr )rr&)rr#)rr)rr)rzAwaitable[None])rr8rzSessionClient | None)rr8rrr)rr8rrB)NN) rrrr%r str | Nonerrrr8)rr8rr"rr)rr8r BaseExceptionrr)rztuple[bool, str])rrFrr)rrrr rr)rrM)%r.r/r0rVrD classmethodr[r]rupropertyryr>r?r}r=rrrSrrrrrrrrrrrrrrrrrnrrr2r3r4rUrUs $I~$))9Av((--''++$$.. # 3!:E*?+/*. == =( = ( = =F+/*.     (  (   ,(0(014;;$1; ;>$$)V jX38 Q  @r3rU)N __future__rrsysrr dataclassesrrenumrtypingrrr 1streamlit.components.lib.local_component_registryr )streamlit.components.v2.component_managerr streamlit.loggerr streamlit.proto.ForwardMsg_pb2r streamlit.runtime.app_sessionrstreamlit.runtime.cachingrr:streamlit.runtime.caching.storage.local_disk_cache_storager$streamlit.runtime.media_file_managerr(streamlit.runtime.memory_session_storagerstreamlit.runtime.script_datar+streamlit.runtime.scriptrunner.script_cacher!streamlit.runtime.session_managerrrrrrstreamlit.runtime.staterrstreamlit.runtime.statsrr+streamlit.runtime.websocket_session_managerrcollections.abcr 2streamlit.components.types.base_component_registryr!streamlit.proto.BackMsg_pb2r"!streamlit.runtime.caching.storager#$streamlit.runtime.media_file_storager$7streamlit.runtime.scriptrunner_utils.script_run_contextr%'streamlit.runtime.uploaded_file_managerr&r(rDplatformr*r.r+rr-r7rFrMrUr2r3r4rs# (33TJ'54BI4CP)X3EETK#%%$(+||w'>DHH%%F)F $"""V4":"0i @i @r3