streamlit-nightly 1.41.2.dev20250130__py2.py3-none-any.whl → 1.41.2.dev20250201__py2.py3-none-any.whl

streamlit/elements/lib/utils.py

@@ -30,6 +30,7 @@ from typing_extensions import TypeAlias
 
 from streamlit import config
 from streamlit.errors import StreamlitDuplicateElementId, StreamlitDuplicateElementKey
+from streamlit.proto.ChatInput_pb2 import ChatInput
 from streamlit.proto.LabelVisibilityMessage_pb2 import LabelVisibilityMessage
 from streamlit.runtime.scriptrunner_utils.script_run_context import (
     ScriptRunContext,
@@ -78,6 +79,21 @@ def get_label_visibility_proto_value(
     raise ValueError(f"Unknown label visibility value: {label_visibility_string}")
 
 
+def get_chat_input_accept_file_proto_value(
+    accept_file_value: bool | Literal["multiple"],
+) -> ChatInput.AcceptFile.ValueType:
+    """Returns one of ChatInput.AcceptFile enum value based on string value."""
+
+    if accept_file_value is False:
+        return ChatInput.AcceptFile.NONE
+    elif accept_file_value is True:
+        return ChatInput.AcceptFile.SINGLE
+    elif accept_file_value == "multiple":
+        return ChatInput.AcceptFile.MULTIPLE
+
+    raise ValueError(f"Unknown accept file value: {accept_file_value}")
+
+
 @overload
 def to_key(key: None) -> None: ...
 

streamlit/elements/widgets/chat.py

@@ -16,23 +16,35 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 from enum import Enum
-from typing import TYPE_CHECKING, Literal, cast
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Iterator,
+    Literal,
+    MutableMapping,
+    Sequence,
+    cast,
+    overload,
+)
 
 from streamlit import runtime
 from streamlit.delta_generator_singletons import get_dg_singleton_instance
+from streamlit.elements.lib.file_uploader_utils import normalize_upload_file_type
 from streamlit.elements.lib.form_utils import is_in_form
 from streamlit.elements.lib.image_utils import AtomicImage, WidthBehavior, image_to_url
 from streamlit.elements.lib.policies import check_widget_policies
 from streamlit.elements.lib.utils import (
     Key,
     compute_and_register_element_id,
+    get_chat_input_accept_file_proto_value,
     save_for_app_testing,
     to_key,
 )
 from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Block_pb2 import Block as BlockProto
 from streamlit.proto.ChatInput_pb2 import ChatInput as ChatInputProto
-from streamlit.proto.Common_pb2 import StringTriggerValue as StringTriggerValueProto
+from streamlit.proto.Common_pb2 import ChatInputValue as ChatInputValueProto
+from streamlit.proto.Common_pb2 import FileUploaderState as FileUploaderStateProto
 from streamlit.proto.RootContainer_pb2 import RootContainer
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner_utils.script_run_context import get_script_run_ctx
@@ -42,12 +54,43 @@ from streamlit.runtime.state import (
     WidgetKwargs,
     register_widget,
 )
+from streamlit.runtime.uploaded_file_manager import UploadedFile
 from streamlit.string_util import is_emoji, validate_material_icon
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
 
 
+@dataclass
+class ChatInputValue(MutableMapping[str, Any]):
+    text: str
+    files: list[UploadedFile]
+
+    def __len__(self) -> int:
+        return len(vars(self))
+
+    def __iter__(self) -> Iterator[str]:
+        return iter(vars(self))
+
+    def __getitem__(self, item: str) -> str | list[UploadedFile]:
+        try:
+            return getattr(self, item)  # type: ignore[no-any-return]
+        except AttributeError:
+            raise KeyError(f"Invalid key: {item}") from None
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        setattr(self, key, value)
+
+    def __delitem__(self, key: str) -> None:
+        try:
+            delattr(self, key)
+        except AttributeError:
+            raise KeyError(f"Invalid key: {key}") from None
+
+    def to_dict(self) -> dict[str, str | list[UploadedFile]]:
+        return vars(self)
+
+
 class PresetNames(str, Enum):
     USER = "user"
     ASSISTANT = "assistant"
@@ -108,18 +151,65 @@ def _process_avatar_input(
             ) from ex
 
 
+def _pop_upload_files(
+    files_value: FileUploaderStateProto | None,
+) -> list[UploadedFile]:
+    if files_value is None:
+        return []
+
+    ctx = get_script_run_ctx()
+    if ctx is None:
+        return []
+
+    uploaded_file_info = files_value.uploaded_file_info
+    if len(uploaded_file_info) == 0:
+        return []
+
+    file_recs_list = ctx.uploaded_file_mgr.get_files(
+        session_id=ctx.session_id,
+        file_ids=[f.file_id for f in uploaded_file_info],
+    )
+
+    file_recs = {f.file_id: f for f in file_recs_list}
+
+    collected_files: list[UploadedFile] = []
+
+    for f in uploaded_file_info:
+        maybe_file_rec = file_recs.get(f.file_id)
+        if maybe_file_rec is not None:
+            uploaded_file = UploadedFile(maybe_file_rec, f.file_urls)
+            collected_files.append(uploaded_file)
+
+            if hasattr(ctx.uploaded_file_mgr, "remove_file"):
+                ctx.uploaded_file_mgr.remove_file(
+                    session_id=ctx.session_id,
+                    file_id=f.file_id,
+                )
+
+    return collected_files
+
+
 @dataclass
 class ChatInputSerde:
+    accept_files: bool = False
+
     def deserialize(
-        self, ui_value: StringTriggerValueProto | None, widget_id: str = ""
-    ) -> str | None:
+        self,
+        ui_value: ChatInputValueProto | None,
+        widget_id: str = "",
+    ) -> str | ChatInputValue | None:
         if ui_value is None or not ui_value.HasField("data"):
             return None
+        if not self.accept_files:
+            return ui_value.data
+        else:
+            return ChatInputValue(
+                text=ui_value.data,
+                files=_pop_upload_files(ui_value.file_uploader_state),
+            )
 
-        return ui_value.data
-
-    def serialize(self, v: str | None) -> StringTriggerValueProto:
-        return StringTriggerValueProto(data=v)
+    def serialize(self, v: str | None) -> ChatInputValueProto:
+        return ChatInputValueProto(data=v)
 
 
 class ChatMixin:
@@ -237,6 +327,36 @@ class ChatMixin:
 
         return self.dg._block(block_proto=block_proto)
 
+    @overload
+    def chat_input(
+        self,
+        placeholder: str = "Your message",
+        *,
+        key: Key | None = None,
+        max_chars: int | None = None,
+        accept_file: Literal[False] = False,
+        file_type: str | Sequence[str] | None = None,
+        disabled: bool = False,
+        on_submit: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
+    ) -> str | None: ...
+
+    @overload
+    def chat_input(
+        self,
+        placeholder: str = "Your message",
+        *,
+        key: Key | None = None,
+        max_chars: int | None = None,
+        accept_file: Literal[True, "multiple"],
+        file_type: str | Sequence[str] | None = None,
+        disabled: bool = False,
+        on_submit: WidgetCallback | None = None,
+        args: WidgetArgs | None = None,
+        kwargs: WidgetKwargs | None = None,
+    ) -> ChatInputValue | None: ...
+
     @gather_metrics("chat_input")
     def chat_input(
         self,
@@ -244,11 +364,13 @@ class ChatMixin:
         *,
         key: Key | None = None,
         max_chars: int | None = None,
+        accept_file: bool | Literal["multiple"] = False,
+        file_type: str | Sequence[str] | None = None,
         disabled: bool = False,
         on_submit: WidgetCallback | None = None,
         args: WidgetArgs | None = None,
         kwargs: WidgetKwargs | None = None,
-    ) -> str | None:
+    ) -> str | ChatInputValue | None:
         """Display a chat input widget.
 
         Parameters
@@ -267,9 +389,17 @@ class ChatMixin:
             The maximum number of characters that can be entered. If ``None``
             (default), there will be no maximum.
 
+        accept_file : bool | str
+            Whether the chat input should accept files. ``True`` to accept a single
+            file, ``"multiple"`` to accept multiple files.
+
         disabled : bool
             Whether the chat input should be disabled. Defaults to ``False``.
 
+        file_type : str or list[str] or None
+            Array of allowed extensions. ['png', 'jpg']
+            The default is None, which means all extensions are allowed.
+
         on_submit : callable
             An optional callback invoked when the chat input's value is submitted.
 
@@ -329,7 +459,13 @@ class ChatMixin:
             writes_allowed=False,
         )
 
+        if accept_file not in {True, False, "multiple"}:
+            raise StreamlitAPIException(
+                "The `accept_file` parameter must be a boolean or 'multiple'."
+            )
+
         ctx = get_script_run_ctx()
+
         element_id = compute_and_register_element_id(
             "chat_input",
             user_key=key,
@@ -337,8 +473,13 @@ class ChatMixin:
             form_id=None,
             placeholder=placeholder,
             max_chars=max_chars,
+            accept_file=accept_file,
+            file_type=file_type,
         )
 
+        if file_type:
+            file_type = normalize_upload_file_type(file_type)
+
         # It doesn't make sense to create a chat input inside a form.
         # We throw an error to warn the user about this.
         # We omit this check for scripts running outside streamlit, because
@@ -371,8 +512,14 @@ class ChatMixin:
 
         chat_input_proto.default = default
 
-        serde = ChatInputSerde()
-        widget_state = register_widget(
+        chat_input_proto.accept_file = get_chat_input_accept_file_proto_value(
+            accept_file
+        )
+
+        chat_input_proto.file_type[:] = file_type if file_type is not None else []
+
+        serde = ChatInputSerde(accept_files=bool(accept_file))
+        widget_state = register_widget(  # type: ignore[misc]
             chat_input_proto.id,
             on_change_handler=on_submit,
             args=args,
@@ -380,7 +527,7 @@ class ChatMixin:
             deserializer=serde.deserialize,
             serializer=serde.serialize,
             ctx=ctx,
-            value_type="string_trigger_value",
+            value_type="chat_input_value",
         )
 
         chat_input_proto.disabled = disabled

streamlit/proto/ChatInput_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fstreamlit/proto/ChatInput.proto\"\xc3\x01\n\tChatInput\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bplaceholder\x18\x02 \x01(\t\x12\x11\n\tmax_chars\x18\x03 \x01(\r\x12\x10\n\x08\x64isabled\x18\x04 \x01(\x08\x12\r\n\x05value\x18\x05 \x01(\t\x12\x11\n\tset_value\x18\x06 \x01(\x08\x12\x0f\n\x07\x64\x65\x66\x61ult\x18\x07 \x01(\t\x12%\n\x08position\x18\x08 \x01(\x0e\x32\x13.ChatInput.Position\"\x16\n\x08Position\x12\n\n\x06\x42OTTOM\x10\x00\x42.\n\x1c\x63om.snowflake.apps.streamlitB\x0e\x43hatInputProtob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fstreamlit/proto/ChatInput.proto\"\xb4\x02\n\tChatInput\x12\n\n\x02id\x18\x01 \x01(\t\x12\x13\n\x0bplaceholder\x18\x02 \x01(\t\x12\x11\n\tmax_chars\x18\x03 \x01(\r\x12\x10\n\x08\x64isabled\x18\x04 \x01(\x08\x12\r\n\x05value\x18\x05 \x01(\t\x12\x11\n\tset_value\x18\x06 \x01(\x08\x12\x0f\n\x07\x64\x65\x66\x61ult\x18\x07 \x01(\t\x12%\n\x08position\x18\x08 \x01(\x0e\x32\x13.ChatInput.Position\x12*\n\x0b\x61\x63\x63\x65pt_file\x18\t \x01(\x0e\x32\x15.ChatInput.AcceptFile\x12\x11\n\tfile_type\x18\n \x03(\t\"\x16\n\x08Position\x12\n\n\x06\x42OTTOM\x10\x00\"0\n\nAcceptFile\x12\x08\n\x04NONE\x10\x00\x12\n\n\x06SINGLE\x10\x01\x12\x0c\n\x08MULTIPLE\x10\x02\x42.\n\x1c\x63om.snowflake.apps.streamlitB\x0e\x43hatInputProtob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -23,7 +23,9 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._loaded_options = None
   _globals['DESCRIPTOR']._serialized_options = b'\n\034com.snowflake.apps.streamlitB\016ChatInputProto'
   _globals['_CHATINPUT']._serialized_start=36
-  _globals['_CHATINPUT']._serialized_end=231
-  _globals['_CHATINPUT_POSITION']._serialized_start=209
-  _globals['_CHATINPUT_POSITION']._serialized_end=231
+  _globals['_CHATINPUT']._serialized_end=344
+  _globals['_CHATINPUT_POSITION']._serialized_start=272
+  _globals['_CHATINPUT_POSITION']._serialized_end=294
+  _globals['_CHATINPUT_ACCEPTFILE']._serialized_start=296
+  _globals['_CHATINPUT_ACCEPTFILE']._serialized_end=344
 # @@protoc_insertion_point(module_scope)

streamlit/proto/ChatInput_pb2.pyi

@@ -18,7 +18,9 @@ limitations under the License.
 """
 
 import builtins
+import collections.abc
 import google.protobuf.descriptor
+import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
 import sys
@@ -48,6 +50,21 @@ class ChatInput(google.protobuf.message.Message):
 
     BOTTOM: ChatInput.Position.ValueType  # 0
 
+    class _AcceptFile:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _AcceptFileEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ChatInput._AcceptFile.ValueType], builtins.type):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        NONE: ChatInput._AcceptFile.ValueType  # 0
+        SINGLE: ChatInput._AcceptFile.ValueType  # 1
+        MULTIPLE: ChatInput._AcceptFile.ValueType  # 2
+
+    class AcceptFile(_AcceptFile, metaclass=_AcceptFileEnumTypeWrapper): ...
+    NONE: ChatInput.AcceptFile.ValueType  # 0
+    SINGLE: ChatInput.AcceptFile.ValueType  # 1
+    MULTIPLE: ChatInput.AcceptFile.ValueType  # 2
+
     ID_FIELD_NUMBER: builtins.int
     PLACEHOLDER_FIELD_NUMBER: builtins.int
     MAX_CHARS_FIELD_NUMBER: builtins.int
@@ -56,6 +73,8 @@ class ChatInput(google.protobuf.message.Message):
     SET_VALUE_FIELD_NUMBER: builtins.int
     DEFAULT_FIELD_NUMBER: builtins.int
     POSITION_FIELD_NUMBER: builtins.int
+    ACCEPT_FILE_FIELD_NUMBER: builtins.int
+    FILE_TYPE_FIELD_NUMBER: builtins.int
     id: builtins.str
     placeholder: builtins.str
     max_chars: builtins.int
@@ -64,6 +83,11 @@ class ChatInput(google.protobuf.message.Message):
     set_value: builtins.bool
     default: builtins.str
     position: global___ChatInput.Position.ValueType
+    accept_file: global___ChatInput.AcceptFile.ValueType
+    @property
+    def file_type(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Supported file types: For example: ["png","jpg","img"]"""
+
     def __init__(
         self,
         *,
@@ -75,7 +99,9 @@ class ChatInput(google.protobuf.message.Message):
         set_value: builtins.bool = ...,
         default: builtins.str = ...,
         position: global___ChatInput.Position.ValueType = ...,
+        accept_file: global___ChatInput.AcceptFile.ValueType = ...,
+        file_type: collections.abc.Iterable[builtins.str] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["default", b"default", "disabled", b"disabled", "id", b"id", "max_chars", b"max_chars", "placeholder", b"placeholder", "position", b"position", "set_value", b"set_value", "value", b"value"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["accept_file", b"accept_file", "default", b"default", "disabled", b"disabled", "file_type", b"file_type", "id", b"id", "max_chars", b"max_chars", "placeholder", b"placeholder", "position", b"position", "set_value", b"set_value", "value", b"value"]) -> None: ...
 
 global___ChatInput = ChatInput

streamlit/proto/Common_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cstreamlit/proto/Common.proto\"\x1b\n\x0bStringArray\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\t\"\x1b\n\x0b\x44oubleArray\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x01\"\x1a\n\nInt32Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x05\"\x1a\n\nInt64Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x03\"\x1b\n\x0bSInt64Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x12\"\x1b\n\x0bUInt32Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\r\"0\n\x12StringTriggerValue\x12\x11\n\x04\x64\x61ta\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\x07\n\x05_data\"M\n\x0f\x46ileURLsRequest\x12\x12\n\nrequest_id\x18\x01 \x01(\t\x12\x12\n\nfile_names\x18\x02 \x03(\t\x12\x12\n\nsession_id\x18\x03 \x01(\t\"C\n\x08\x46ileURLs\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\x12\x12\n\nupload_url\x18\x02 \x01(\t\x12\x12\n\ndelete_url\x18\x03 \x01(\t\"X\n\x10\x46ileURLsResponse\x12\x13\n\x0bresponse_id\x18\x01 \x01(\t\x12\x1c\n\tfile_urls\x18\x02 \x03(\x0b\x32\t.FileURLs\x12\x11\n\terror_msg\x18\x03 \x01(\t\"i\n\x10UploadedFileInfo\x12\n\n\x02id\x18\x01 \x01(\x12\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0c\n\x04size\x18\x03 \x01(\x04\x12\x0f\n\x07\x66ile_id\x18\x04 \x01(\t\x12\x1c\n\tfile_urls\x18\x05 \x01(\x0b\x32\t.FileURLs\"W\n\x11\x46ileUploaderState\x12\x13\n\x0bmax_file_id\x18\x01 \x01(\x12\x12-\n\x12uploaded_file_info\x18\x02 \x03(\x0b\x32\x11.UploadedFileInfoB+\n\x1c\x63om.snowflake.apps.streamlitB\x0b\x43ommonProtob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cstreamlit/proto/Common.proto\"\x1b\n\x0bStringArray\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\t\"\x1b\n\x0b\x44oubleArray\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x01\"\x1a\n\nInt32Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x05\"\x1a\n\nInt64Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x03\"\x1b\n\x0bSInt64Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x12\"\x1b\n\x0bUInt32Array\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\r\"0\n\x12StringTriggerValue\x12\x11\n\x04\x64\x61ta\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\x07\n\x05_data\"M\n\x0f\x46ileURLsRequest\x12\x12\n\nrequest_id\x18\x01 \x01(\t\x12\x12\n\nfile_names\x18\x02 \x03(\t\x12\x12\n\nsession_id\x18\x03 \x01(\t\"C\n\x08\x46ileURLs\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\x12\x12\n\nupload_url\x18\x02 \x01(\t\x12\x12\n\ndelete_url\x18\x03 \x01(\t\"X\n\x10\x46ileURLsResponse\x12\x13\n\x0bresponse_id\x18\x01 \x01(\t\x12\x1c\n\tfile_urls\x18\x02 \x03(\x0b\x32\t.FileURLs\x12\x11\n\terror_msg\x18\x03 \x01(\t\"i\n\x10UploadedFileInfo\x12\n\n\x02id\x18\x01 \x01(\x12\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x0c\n\x04size\x18\x03 \x01(\x04\x12\x0f\n\x07\x66ile_id\x18\x04 \x01(\t\x12\x1c\n\tfile_urls\x18\x05 \x01(\x0b\x32\t.FileURLs\"W\n\x11\x46ileUploaderState\x12\x13\n\x0bmax_file_id\x18\x01 \x01(\x12\x12-\n\x12uploaded_file_info\x18\x02 \x03(\x0b\x32\x11.UploadedFileInfo\"z\n\x0e\x43hatInputValue\x12\x11\n\x04\x64\x61ta\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x34\n\x13\x66ile_uploader_state\x18\x02 \x01(\x0b\x32\x12.FileUploaderStateH\x01\x88\x01\x01\x42\x07\n\x05_dataB\x16\n\x14_file_uploader_stateB+\n\x1c\x63om.snowflake.apps.streamlitB\x0b\x43ommonProtob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -46,4 +46,6 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_UPLOADEDFILEINFO']._serialized_end=597
   _globals['_FILEUPLOADERSTATE']._serialized_start=599
   _globals['_FILEUPLOADERSTATE']._serialized_end=686
+  _globals['_CHATINPUTVALUE']._serialized_start=688
+  _globals['_CHATINPUTVALUE']._serialized_end=810
 # @@protoc_insertion_point(module_scope)

streamlit/proto/Common_pb2.pyi

@@ -267,3 +267,27 @@ class FileUploaderState(google.protobuf.message.Message):
     def ClearField(self, field_name: typing.Literal["max_file_id", b"max_file_id", "uploaded_file_info", b"uploaded_file_info"]) -> None: ...
 
 global___FileUploaderState = FileUploaderState
+
+@typing.final
+class ChatInputValue(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_FIELD_NUMBER: builtins.int
+    FILE_UPLOADER_STATE_FIELD_NUMBER: builtins.int
+    data: builtins.str
+    @property
+    def file_uploader_state(self) -> global___FileUploaderState: ...
+    def __init__(
+        self,
+        *,
+        data: builtins.str | None = ...,
+        file_uploader_state: global___FileUploaderState | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["_data", b"_data", "_file_uploader_state", b"_file_uploader_state", "data", b"data", "file_uploader_state", b"file_uploader_state"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["_data", b"_data", "_file_uploader_state", b"_file_uploader_state", "data", b"data", "file_uploader_state", b"file_uploader_state"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_data", b"_data"]) -> typing.Literal["data"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_file_uploader_state", b"_file_uploader_state"]) -> typing.Literal["file_uploader_state"] | None: ...
+
+global___ChatInputValue = ChatInputValue

streamlit/proto/WidgetStates_pb2.py

@@ -16,7 +16,7 @@ from streamlit.proto import Common_pb2 as streamlit_dot_proto_dot_Common__pb2
 from streamlit.proto import Components_pb2 as streamlit_dot_proto_dot_Components__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"streamlit/proto/WidgetStates.proto\x1a\x1cstreamlit/proto/Common.proto\x1a streamlit/proto/Components.proto\"-\n\x0cWidgetStates\x12\x1d\n\x07widgets\x18\x01 \x03(\x0b\x32\x0c.WidgetState\"\xd6\x03\n\x0bWidgetState\x12\n\n\x02id\x18\x01 \x01(\t\x12\x17\n\rtrigger_value\x18\x02 \x01(\x08H\x00\x12\x14\n\nbool_value\x18\x03 \x01(\x08H\x00\x12\x16\n\x0c\x64ouble_value\x18\x04 \x01(\x01H\x00\x12\x13\n\tint_value\x18\x05 \x01(\x12H\x00\x12\x16\n\x0cstring_value\x18\x06 \x01(\tH\x00\x12*\n\x12\x64ouble_array_value\x18\x07 \x01(\x0b\x32\x0c.DoubleArrayH\x00\x12\'\n\x0fint_array_value\x18\x08 \x01(\x0b\x32\x0c.SInt64ArrayH\x00\x12*\n\x12string_array_value\x18\t \x01(\x0b\x32\x0c.StringArrayH\x00\x12\x14\n\njson_value\x18\n \x01(\tH\x00\x12\"\n\x0b\x61rrow_value\x18\x0b \x01(\x0b\x32\x0b.ArrowTableH\x00\x12\x15\n\x0b\x62ytes_value\x18\x0c \x01(\x0cH\x00\x12\x37\n\x19\x66ile_uploader_state_value\x18\r \x01(\x0b\x32\x12.FileUploaderStateH\x00\x12\x33\n\x14string_trigger_value\x18\x0e \x01(\x0b\x32\x13.StringTriggerValueH\x00\x42\x07\n\x05valueB1\n\x1c\x63om.snowflake.apps.streamlitB\x11WidgetStatesProtob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"streamlit/proto/WidgetStates.proto\x1a\x1cstreamlit/proto/Common.proto\x1a streamlit/proto/Components.proto\"-\n\x0cWidgetStates\x12\x1d\n\x07widgets\x18\x01 \x03(\x0b\x32\x0c.WidgetState\"\x83\x04\n\x0bWidgetState\x12\n\n\x02id\x18\x01 \x01(\t\x12\x17\n\rtrigger_value\x18\x02 \x01(\x08H\x00\x12\x14\n\nbool_value\x18\x03 \x01(\x08H\x00\x12\x16\n\x0c\x64ouble_value\x18\x04 \x01(\x01H\x00\x12\x13\n\tint_value\x18\x05 \x01(\x12H\x00\x12\x16\n\x0cstring_value\x18\x06 \x01(\tH\x00\x12*\n\x12\x64ouble_array_value\x18\x07 \x01(\x0b\x32\x0c.DoubleArrayH\x00\x12\'\n\x0fint_array_value\x18\x08 \x01(\x0b\x32\x0c.SInt64ArrayH\x00\x12*\n\x12string_array_value\x18\t \x01(\x0b\x32\x0c.StringArrayH\x00\x12\x14\n\njson_value\x18\n \x01(\tH\x00\x12\"\n\x0b\x61rrow_value\x18\x0b \x01(\x0b\x32\x0b.ArrowTableH\x00\x12\x15\n\x0b\x62ytes_value\x18\x0c \x01(\x0cH\x00\x12\x37\n\x19\x66ile_uploader_state_value\x18\r \x01(\x0b\x32\x12.FileUploaderStateH\x00\x12\x33\n\x14string_trigger_value\x18\x0e \x01(\x0b\x32\x13.StringTriggerValueH\x00\x12+\n\x10\x63hat_input_value\x18\x0f \x01(\x0b\x32\x0f.ChatInputValueH\x00\x42\x07\n\x05valueB1\n\x1c\x63om.snowflake.apps.streamlitB\x11WidgetStatesProtob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -27,5 +27,5 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['_WIDGETSTATES']._serialized_start=102
   _globals['_WIDGETSTATES']._serialized_end=147
   _globals['_WIDGETSTATE']._serialized_start=150
-  _globals['_WIDGETSTATE']._serialized_end=620
+  _globals['_WIDGETSTATE']._serialized_end=665
 # @@protoc_insertion_point(module_scope)

streamlit/proto/WidgetStates_pb2.pyi

@@ -66,6 +66,7 @@ class WidgetState(google.protobuf.message.Message):
     BYTES_VALUE_FIELD_NUMBER: builtins.int
     FILE_UPLOADER_STATE_VALUE_FIELD_NUMBER: builtins.int
     STRING_TRIGGER_VALUE_FIELD_NUMBER: builtins.int
+    CHAT_INPUT_VALUE_FIELD_NUMBER: builtins.int
     id: builtins.str
     trigger_value: builtins.bool
     """trigger_value is for buttons. A button's value needs to
@@ -97,6 +98,8 @@ class WidgetState(google.protobuf.message.Message):
         This is used for the chat_input widget.
         """
 
+    @property
+    def chat_input_value(self) -> streamlit.proto.Common_pb2.ChatInputValue: ...
     def __init__(
         self,
         *,
@@ -114,9 +117,10 @@ class WidgetState(google.protobuf.message.Message):
         bytes_value: builtins.bytes = ...,
         file_uploader_state_value: streamlit.proto.Common_pb2.FileUploaderState | None = ...,
         string_trigger_value: streamlit.proto.Common_pb2.StringTriggerValue | None = ...,
+        chat_input_value: streamlit.proto.Common_pb2.ChatInputValue | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing.Literal["arrow_value", b"arrow_value", "bool_value", b"bool_value", "bytes_value", b"bytes_value", "double_array_value", b"double_array_value", "double_value", b"double_value", "file_uploader_state_value", b"file_uploader_state_value", "int_array_value", b"int_array_value", "int_value", b"int_value", "json_value", b"json_value", "string_array_value", b"string_array_value", "string_trigger_value", b"string_trigger_value", "string_value", b"string_value", "trigger_value", b"trigger_value", "value", b"value"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["arrow_value", b"arrow_value", "bool_value", b"bool_value", "bytes_value", b"bytes_value", "double_array_value", b"double_array_value", "double_value", b"double_value", "file_uploader_state_value", b"file_uploader_state_value", "id", b"id", "int_array_value", b"int_array_value", "int_value", b"int_value", "json_value", b"json_value", "string_array_value", b"string_array_value", "string_trigger_value", b"string_trigger_value", "string_value", b"string_value", "trigger_value", b"trigger_value", "value", b"value"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["trigger_value", "bool_value", "double_value", "int_value", "string_value", "double_array_value", "int_array_value", "string_array_value", "json_value", "arrow_value", "bytes_value", "file_uploader_state_value", "string_trigger_value"] | None: ...
+    def HasField(self, field_name: typing.Literal["arrow_value", b"arrow_value", "bool_value", b"bool_value", "bytes_value", b"bytes_value", "chat_input_value", b"chat_input_value", "double_array_value", b"double_array_value", "double_value", b"double_value", "file_uploader_state_value", b"file_uploader_state_value", "int_array_value", b"int_array_value", "int_value", b"int_value", "json_value", b"json_value", "string_array_value", b"string_array_value", "string_trigger_value", b"string_trigger_value", "string_value", b"string_value", "trigger_value", b"trigger_value", "value", b"value"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["arrow_value", b"arrow_value", "bool_value", b"bool_value", "bytes_value", b"bytes_value", "chat_input_value", b"chat_input_value", "double_array_value", b"double_array_value", "double_value", b"double_value", "file_uploader_state_value", b"file_uploader_state_value", "id", b"id", "int_array_value", b"int_array_value", "int_value", b"int_value", "json_value", b"json_value", "string_array_value", b"string_array_value", "string_trigger_value", b"string_trigger_value", "string_value", b"string_value", "trigger_value", b"trigger_value", "value", b"value"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["trigger_value", "bool_value", "double_value", "int_value", "string_value", "double_array_value", "int_array_value", "string_array_value", "json_value", "arrow_value", "bytes_value", "file_uploader_state_value", "string_trigger_value", "chat_input_value"] | None: ...
 
 global___WidgetState = WidgetState

streamlit/runtime/scriptrunner_utils/script_requests.py

@@ -20,6 +20,7 @@ from enum import Enum
 from typing import cast
 
 from streamlit import util
+from streamlit.proto.Common_pb2 import ChatInputValue as ChatInputValueProto
 from streamlit.proto.Common_pb2 import StringTriggerValue as StringTriggerValueProto
 from streamlit.proto.WidgetStates_pb2 import WidgetState, WidgetStates
 
@@ -116,6 +117,7 @@ def _coalesce_widget_states(
     trigger_value_types = [
         ("trigger_value", False),
         ("string_trigger_value", StringTriggerValueProto(data=None)),
+        ("chat_input_value", ChatInputValueProto(data=None)),
     ]
     for old_state in old_states.widgets:
         for trigger_value_type, unset_value in trigger_value_types:

streamlit/runtime/state/common.py

@@ -97,6 +97,7 @@ ValueFieldName: TypeAlias = Literal[
     "string_value",
     "trigger_value",
     "string_trigger_value",
+    "chat_input_value",
 ]
 
 

streamlit/runtime/state/session_state.py

@@ -237,6 +237,8 @@ class WStates(MutableMapping[str, Any]):
             widget.file_uploader_state_value.CopyFrom(serialized)
         elif field == "string_trigger_value":
             widget.string_trigger_value.CopyFrom(serialized)
+        elif field == "chat_input_value":
+            widget.chat_input_value.CopyFrom(serialized)
         elif field is not None and serialized is not None:
             # If the field is None, the widget value was cleared
             # by the user and therefore is None. But we cannot
@@ -605,6 +607,8 @@ class SessionState:
                     self._new_widget_state[state_id] = Value(False)
                 elif metadata.value_type == "string_trigger_value":
                     self._new_widget_state[state_id] = Value(None)
+                elif metadata.value_type == "chat_input_value":
+                    self._new_widget_state[state_id] = Value(None)
 
         for state_id in self._old_state:
             metadata = self._new_widget_state.widget_metadata.get(state_id)
@@ -613,6 +617,8 @@ class SessionState:
                     self._old_state[state_id] = False
                 elif metadata.value_type == "string_trigger_value":
                     self._old_state[state_id] = None
+                elif metadata.value_type == "chat_input_value":
+                    self._old_state[state_id] = None
 
     def _remove_stale_widgets(self, active_widget_ids: set[str]) -> None:
         """Remove widget state for widgets whose ids aren't in `active_widget_ids`."""

streamlit/static/index.html

@@ -51,7 +51,7 @@
     <script>
       window.prerenderReady = false
     </script>
-    <script type="module" crossorigin src="./static/js/index.7ACWibTY.js"></script>
+    <script type="module" crossorigin src="./static/js/index.B7J6suR-.js"></script>
     <link rel="stylesheet" crossorigin href="./static/css/index.mUTQuMqR.css">
   </head>
   <body>

streamlit/static/static/js/{FileDownload.esm.BW2GD_iE.js → FileDownload.esm.Cf3bITtP.js}

@@ -1 +1 @@
-import{r as e,E as n,_ as a}from"./index.7ACWibTY.js";var o=e.forwardRef(function(t,r){var l={fill:"currentColor",xmlns:"http://www.w3.org/2000/svg"};return e.createElement(n,a({iconAttrs:l,iconVerticalAlign:"middle",iconViewBox:"0 0 24 24"},t,{ref:r}),e.createElement("path",{fill:"none",d:"M0 0h24v24H0V0z"}),e.createElement("path",{d:"M16 9v10H8V9h8m-1.5-6h-5l-1 1H5v2h14V4h-3.5l-1-1zM18 7H6v12c0 1.1.9 2 2 2h8c1.1 0 2-.9 2-2V7z"}))});o.displayName="Delete";var i=e.forwardRef(function(t,r){var l={fill:"currentColor",xmlns:"http://www.w3.org/2000/svg"};return e.createElement(n,a({iconAttrs:l,iconVerticalAlign:"middle",iconViewBox:"0 0 24 24"},t,{ref:r}),e.createElement("rect",{width:24,height:24,fill:"none"}),e.createElement("path",{d:"M18 15v3H6v-3H4v3c0 1.1.9 2 2 2h12c1.1 0 2-.9 2-2v-3h-2zm-1-4l-1.41-1.41L13 12.17V4h-2v8.17L8.41 9.59 7 11l5 5 5-5z"}))});i.displayName="FileDownload";export{o as D,i as F};
+import{r as e,E as n,_ as a}from"./index.B7J6suR-.js";var o=e.forwardRef(function(t,r){var l={fill:"currentColor",xmlns:"http://www.w3.org/2000/svg"};return e.createElement(n,a({iconAttrs:l,iconVerticalAlign:"middle",iconViewBox:"0 0 24 24"},t,{ref:r}),e.createElement("path",{fill:"none",d:"M0 0h24v24H0V0z"}),e.createElement("path",{d:"M16 9v10H8V9h8m-1.5-6h-5l-1 1H5v2h14V4h-3.5l-1-1zM18 7H6v12c0 1.1.9 2 2 2h8c1.1 0 2-.9 2-2V7z"}))});o.displayName="Delete";var i=e.forwardRef(function(t,r){var l={fill:"currentColor",xmlns:"http://www.w3.org/2000/svg"};return e.createElement(n,a({iconAttrs:l,iconVerticalAlign:"middle",iconViewBox:"0 0 24 24"},t,{ref:r}),e.createElement("rect",{width:24,height:24,fill:"none"}),e.createElement("path",{d:"M18 15v3H6v-3H4v3c0 1.1.9 2 2 2h12c1.1 0 2-.9 2-2v-3h-2zm-1-4l-1.41-1.41L13 12.17V4h-2v8.17L8.41 9.59 7 11l5 5 5-5z"}))});i.displayName="FileDownload";export{o as D,i as F};