gddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z dZ dZ d ddZej dd Zej dd Zdd Zdd Zdd ZdZddZddZddZdZdZy)) annotationsN)Path)env_utilerrors)is_binary_stringz .streamlitstaticc|dk(rt|rdnd}n|}|r$tj|j|Stj|S)a<Coerce bytes to a BytesIO or a StringIO. Parameters ---------- data : bytes encoding : str Returns ------- BytesIO or StringIO If the file's data is in a well-known textual format (or if the encoding parameter is set), return a StringIO. Otherwise, return BytesIO. autoNzutf-8)rioStringIOdecodeBytesIO)dataencoding data_encodings H/var/www/openai/venv/lib/python3.12/site-packages/streamlit/file_util.pyget_encoded_file_datar!sK"6!1 6G  {{4;;}566 ::d c#8Kt|}tj|jdk(rt j d|dd}|r|dz }t tjjt||5}|dddy#1swYyxYww)a$Opens a context to read this file relative to the streamlit path. For example: with streamlit_read('foo.txt') as foo: ... opens the file `.streamlit/foo.txt` path - the path to write to (within the streamlit directory) binary - set to True for binary IO rzRead zero byte file: ""rbN) get_streamlit_file_pathosstatst_sizerErroropenpathjoinCONFIG_FOLDER_NAME)rbinaryfilenamemodehandles rstreamlit_readr&@s't,H wwx  A%ll3H:Q?@@ D    bggll-t4d ;v  < ; ;sA>BB BBBc#Kd}|r|dz }t|}tjtjj |d t ||5}|dddy#1swYyxYw#t $r}dtjj|g}|jtjk(r!tjr|jdtjdj|d}~wwxYww) aOpens a file for writing within the streamlit path, and ensuring that the path exists. For example: with streamlit_write('foo/bar.txt') as bar: ... opens the file .streamlit/foo/bar.txt for writing, creating any necessary directories along the way. path - the path to write to (within the streamlit directory) binary - set to True for binary IO wrT)exist_okNzUnable to write file: zSPython is limited to files below 2GB on OSX. See https://bugs.python.org/issue24658 )rrmakedirsrdirnamerOSErrorabspatherrnoEINVALr IS_DARWINappendrrr )rr"r$r%emsgs rstreamlit_writer5Ys D    "4 (DKK%5 + $ L   +'(='>?@ 77ell "x'9'9 JJ9 ll499S>**+sOA D A2A&A2%D&A/+A2.D/A22 D;BDDDctjjtjjt}tjjtjj |dS)z3Get the folder where static HTML/JS/CSS files live.r)rrr,normpath__file__r )r,s rget_static_dirr9ysGggoobgg..x89G 77  BGGLL(; <z%file_in_pythonpath..s/"D rww//94@"s36) renvirongetlensplitpathseprr.any)rD pythonpathrabsolute_pathss` rfile_in_pythonpathrisz b1J :!8B8H8H8TU8Tbggood+8TNU " Vs&Bcttjjtjj|S)zReturn the normalized path of the joined path. Parameters ---------- *args : str The path components to join. Returns ------- str The normalized path of the joined path. )rrr7r )argss rnormalize_path_joinrls& 77  BGGLL$/ 00rc|ttj|}tjj |S)aReturn the full path to the main script directory. Parameters ---------- main_script : str The main script path. The path can be an absolute path or a relative path. Returns ------- str The full path to the main script directory. )rlrgetcwdrr,) main_scriptr=s rget_main_script_directoryrps++299; D 77??+ ,,r)r )rbytesrrCreturnzio.StringIO | io.BytesIO)F)rrrC)r=rCrrrC)rDrCrMrCrrbool)rSrCrrint)rDrCrrrs) __future__r contextlibr/r rpathlibr streamlitrrstreamlit.string_utilrr!r<rcontextmanagerr&r5r9r?rrGrOrZrirlrprrr|s# &2"""( > 0 ++>= ' <B624 1 -r