o hU@s8UddlmZddlZddlZddlZddlmZmZmZddl m Z m Z m Z m Z mZmZddlmZddlmZmZmZmZmZddlmZmZddlmZdd lmZdd lm Z dd l!m"Z"m#Z#erkdd l$m%Z%ej&ej'ej(ej)ej*fZ+d e,d<ee-Z.de,d<dZ/de,d<Gddde0eZ1GdddZ2dS)) annotationsN)ChainMapUserDictUserList)AsyncGenerator Generator ItemsViewIterableKeysView ValuesView)StringIO) TYPE_CHECKINGAnyCallableFinalcast)dataframe_util type_util)StreamlitAPIException) get_logger)gather_metrics)is_mem_address_strmax_char_sequence)DeltaGeneratorzFinal[tuple[type[Any], ...]] HELP_TYPESr_LOGGERu ▏ _TEXT_CURSORc@s eZdZdS)StreamingOutputN)__name__ __module__ __qualname__r!r!L/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/elements/write.pyr@src@sDeZdZeddddZedd d dddZedddZdS) WriteMixin write_streamstreamXCallable[..., Any] | Generator[Any, Any, Any] | Iterable[Any] | AsyncGenerator[Any, Any]returnlist[Any] | strc st|ts t|rtdt|dddtfdd}t|s-t |r0|}t |r:t |}zt |WntyW}z tdt|d|d}~ww|D]}t |rzt|jd ksq|jd jdurtd}n |jd jjp|d}Wnty}ztd |d}~wwt |d rz|jpd}Wnty}ztd |d}~wwt|tr|sqZd }s|jd}|7|rdntqZt|r||qZ||||qZ|sdStdkrtd trd SS)a Stream a generator, iterable, or stream-like sequence to the app. ``st.write_stream`` iterates through the given sequences and writes all chunks to the app. String chunks will be written using a typewriter effect. Other data types will be written using ``st.write``. Parameters ---------- stream : Callable, Generator, Iterable, OpenAI Stream, or LangChain Stream The generator or iterable to stream. If you pass an async generator, Streamlit will internally convert it to a sync generator. .. note:: To use additional LLM libraries, you can create a wrapper to manually define a generator function and include custom output parsing. Returns ------- str or list The full response. If the streamed output only contains text, this is a string. Otherwise, this is a list of all the streamed objects. The return value is fully compatible as input for ``st.write``. Example ------- You can pass an OpenAI stream as shown in our tutorial, `Build a basic LLM chat app `_. Alternatively, you can pass a generic generator function as input: >>> import time >>> import numpy as np >>> import pandas as pd >>> import streamlit as st >>> >>> _LOREM_IPSUM = """ >>> Lorem ipsum dolor sit amet, **consectetur adipiscing** elit, sed do eiusmod tempor >>> incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis >>> nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. >>> """ >>> >>> >>> def stream_data(): >>> for word in _LOREM_IPSUM.split(" "): >>> yield word + " " >>> time.sleep(0.02) >>> >>> yield pd.DataFrame( >>> np.random.randn(5, 10), >>> columns=["a", "b", "c", "d", "e", "f", "g", "h", "i", "j"], >>> ) >>> >>> for word in _LOREM_IPSUM.split(" "): >>> yield word + " " >>> time.sleep(0.02) >>> >>> >>> if st.button("Stream data"): >>> st.write_stream(stream_data) .. output:: https://doc-write-stream-data.streamlit.app/ height: 550px zI`st.write_stream` expects a generator or stream-like object as input not z3. Please use `st.write` instead for this data type.Ncs0rrdddSdSdS)z#Write the full response to the app.Nr))markdownappendr!stream_containerstreamed_responsewritten_contentr!r"flush_stream_responses   z6WriteMixin.write_stream..flush_stream_responsezThe provided input (type: z^) cannot be iterated. Please make sure that it is a generator, generator function or iterable.ra8Failed to parse the OpenAI ChatCompletionChunk. The most likely cause is a change of the chunk object structure due to a recent OpenAI update. You might be able to fix this by downgrading the OpenAI library or upgrading Streamlit. Also, please report this issue to: https://github.com/streamlit/streamlit/issues.z)langchain_core.messages.ai.AIMessageChunka9Failed to parse the LangChain AIMessageChunk. The most likely cause is a change of the chunk object structure due to a recent LangChain update. You might be able to fix this by downgrading the OpenAI library or upgrading Streamlit. Also, please report this issue to: https://github.com/streamlit/streamlit/issues.FT) isinstancestrris_dataframe_likertyperinspectisgeneratorfunctionisasyncgenfunction isasyncgenrasync_generator_to_synciter TypeErroris_openai_chunklenchoicesdeltacontentAttributeErroris_typedgemptyr*rcallablewriter+)selfr%r0excchunkerr first_textr!r,r"r$EsO          zWriteMixin.write_streamrGFunsafe_allow_htmlargsrrNboolNonec sV|rtd|gjjst|dkrtdfdd}|D]}t|tr0|q"t|t rP||D]}t |rG||q:j |dq:q"t|t r_|j |q"t|rn|j|q"t|r}|j|q"t|r|j|q"t|dr|j|q"t|r|j|q"t|dr|j|q"t|r|j|q"t|r|j|q"t|r|j |q"t!|rd d l"m#}||$|}j|%q"t|t&t't(t)t*j+t,t-t.t/t0t1f s)t2|s)t3|s)t4|r3|j5|q"t6|rC|j7|q"t|t8rU|j9|:q"t;<|stt;=|stt;>|stt;?|stt|d r}|@|q"t|tAr|j|q"tBC|r|j|q"t;D|r|jtEd |q"rtF|d rÈjG|Hq"tF|dstF|dr|j|q"t|} tI| r|j|q"d| vrtJdtK| dd} d| } | d| d| q"tK| dd} d| } | | | q"|dS)aDisplays arguments in the app. This is the Swiss Army knife of Streamlit commands: it does different things depending on what you throw at it. Unlike other Streamlit commands, ``st.write()`` has some unique properties: - You can pass in multiple arguments, all of which will be displayed. - Its behavior depends on the input type(s). Parameters ---------- *args : any One or many objects to display in the app. .. list-table:: Each type of argument is handled as follows: :header-rows: 1 * - Type - Handling * - ``str`` - Uses ``st.markdown()``. * - dataframe-like, ``dict``, or ``list`` - Uses ``st.dataframe()``. * - ``Exception`` - Uses ``st.exception()``. * - function, module, or class - Uses ``st.help()``. * - ``DeltaGenerator`` - Uses ``st.help()``. * - Altair chart - Uses ``st.altair_chart()``. * - Bokeh figure - Uses ``st.bokeh_chart()``. * - Graphviz graph - Uses ``st.graphviz_chart()``. * - Keras model - Converts model and uses ``st.graphviz_chart()``. * - Matplotlib figure - Uses ``st.pyplot()``. * - Plotly figure - Uses ``st.plotly_chart()``. * - ``PIL.Image`` - Uses ``st.image()``. * - generator or stream (like ``openai.Stream``) - Uses ``st.write_stream()``. * - SymPy expression - Uses ``st.latex()``. * - An object with ``._repr_html()`` - Uses ``st.html()``. * - Database cursor - Displays DB API 2.0 cursor results in a table. * - Any - Displays ``str(arg)`` as inline code. unsafe_allow_html : bool Whether to render HTML within ``*args``. This only applies to strings or objects falling back on ``_repr_html_()``. If this is ``False`` (default), any HTML tags found in ``body`` will be escaped and therefore treated as raw text. If this is ``True``, any HTML expressions within ``body`` will be rendered. Adding custom HTML to your app impacts safety, styling, and maintainability. .. note:: If you only want to insert HTML or CSS without Markdown text, we recommend using ``st.html`` instead. **kwargs : any Keyword arguments. Not used. .. deprecated:: ``**kwargs`` is deprecated and will be removed in a later version. Use other, more specific Streamlit commands to pass additional keyword arguments. Returns ------- None Examples -------- Its basic use case is to draw Markdown-formatted text, whenever the input is a string: >>> import streamlit as st >>> >>> st.write("Hello, *World!* :sunglasses:") .. output:: https://doc-write1.streamlit.app/ height: 150px As mentioned earlier, ``st.write()`` also accepts other data formats, such as numbers, data frames, styled data frames, and assorted objects: >>> import streamlit as st >>> import pandas as pd >>> >>> st.write(1234) >>> st.write( ... pd.DataFrame( ... { ... "first column": [1, 2, 3, 4], ... "second column": [10, 20, 30, 40], ... } ... ) ... ) .. output:: https://doc-write2.streamlit.app/ height: 350px Finally, you can pass in multiple arguments to do things like: >>> import streamlit as st >>> >>> st.write("1 + 1 = ", 2) >>> st.write("Below is a DataFrame:", data_frame, "Above is a dataframe.") .. output:: https://doc-write3.streamlit.app/ height: 410px Oh, one more thing: ``st.write`` accepts chart objects too! For example: >>> import streamlit as st >>> import pandas as pd >>> import numpy as np >>> import altair as alt >>> >>> df = pd.DataFrame(np.random.randn(200, 3), columns=["a", "b", "c"]) >>> c = ( ... alt.Chart(df) ... .mark_circle() ... .encode(x="a", y="b", size="c", color="c", tooltip=["a", "b", "c"]) ... ) >>> >>> st.write(c) .. output:: https://doc-vega-lite-chart.streamlit.app/ height: 300px zInvalid arguments were passed to "st.write" function. Support for passing such unknown keywords arguments will be dropped in future. Invalid arguments were: %sr1zCannot replace a single element with multiple elements. The `write()` method only supports multiple elements when inserting elements rather than replacing. That is, only when called as `st.write()` or `st.sidebar.write()`.cs:rd}j}|j|dgdd<dSdS)N rM)joinrDrEr*) text_contenttext_containerrH string_bufferrNr!r" flush_buffers  z&WriteMixin.write..flush_bufferrMzmatplotlib.figure.Figurezbokeh.plotting.figure.Figurer) vis_utilsz openai.Streamr5 _repr_html_ to_pandas __dataframe__ `N)LrwarningrD _is_top_levelr>rr2r3r+rrFrG Exception exceptionris_delta_generatorhelprr4 dataframeis_altair_chart altair_chartrCpyplotis_plotly_chart plotly_chart bokeh_chartis_graphviz_chartgraphviz_chartis_sympy_expressionlatexis_pillow_imageimageis_keras_modeltensorflow.python.keras.utilsrY model_to_dot to_stringdictlistmap enumeratetypesMappingProxyTyperrrrr r is_custom_dict is_namedtupleis_pydantic_modeljson is_pydeck pydeck_chartr r*getvaluer6 isgeneratorr7r8r9r$r dataclasses is_dataclassisclassrhas_callable_attrhtmlrZrmaxr) rHrNrOkwargsrXargitemrYdotstringified_argbacktick_countbacktick_wrapperr!rVr"rGs                            zWriteMixin.writercCs td|S)zGet our DeltaGenerator.r)r)rHr!r!r"rD:s z WriteMixin.dgN)r%r&r'r()rOrrNrPr'rQ)r'r)rrr rr$rGpropertyrDr!r!r!r"r#Ds :<r#)3 __future__rrr6r{ collectionsrrrcollections.abcrrrr r r ior typingr rrrr streamlitrrstreamlit.errorsrstreamlit.loggerrstreamlit.runtime.metrics_utilrstreamlit.string_utilrrstreamlit.delta_generatorrBuiltinFunctionTypeBuiltinMethodType FunctionType MethodType ModuleTyper__annotations__rrrrxrr#r!r!r!r"s2