streamlit-nightly 1.37.1.dev20240731__py2.py3-none-any.whl → 1.37.1.dev20240802__py2.py3-none-any.whl

streamlit/commands/execution_control.py

@@ -67,9 +67,7 @@ def _new_fragment_id_queue(
         return []
 
     else:  # scope == "fragment"
-        curr_queue = (
-            ctx.script_requests.fragment_id_queue if ctx.script_requests else []
-        )
+        curr_queue = ctx.fragment_ids_this_run
 
         # If st.rerun(scope="fragment") is called during a full script run, we raise an
         # exception. This occurs, of course, if st.rerun(scope="fragment") is called
@@ -143,7 +141,7 @@ def rerun(  # type: ignore[misc]
                 query_string=query_string,
                 page_script_hash=page_script_hash,
                 fragment_id_queue=_new_fragment_id_queue(ctx, scope),
-                is_fragment_scoped_rerun=True,
+                is_fragment_scoped_rerun=scope == "fragment",
             )
         )
         # Force a yield point so the runner can do the rerun

streamlit/config.py

@@ -927,6 +927,21 @@ _create_option(
     """,
 )
 
+# Config Section: Secrets #
+
+_create_section("secrets", "Secrets configuration.")
+
+_create_option(
+    "secrets.files",
+    description="""List of locations where secrets are searched. Entries can be a path to toml file or directory path where Kubernetes style secrets will be scanned. Order is important, import is first to last, so secrets in later files will take precedence over earlier ones.""",
+    default_val=[
+        # NOTE: The order here is important! Project-level secrets should overwrite global
+        # secrets.
+        file_util.get_streamlit_file_path("secrets.toml"),
+        file_util.get_project_streamlit_file_path("secrets.toml"),
+    ],
+)
+
 
 def get_where_defined(key: str) -> str:
     """Indicate where (e.g. in which file) this option was defined.

streamlit/proto/ClientState_pb2.py

@@ -15,7 +15,7 @@ _sym_db = _symbol_database.Default()
 from streamlit.proto import WidgetStates_pb2 as streamlit_dot_proto_dot_WidgetStates__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!streamlit/proto/ClientState.proto\x1a\"streamlit/proto/WidgetStates.proto\"\x8b\x01\n\x0b\x43lientState\x12\x14\n\x0cquery_string\x18\x01 \x01(\t\x12$\n\rwidget_states\x18\x02 \x01(\x0b\x32\r.WidgetStates\x12\x18\n\x10page_script_hash\x18\x03 \x01(\t\x12\x11\n\tpage_name\x18\x04 \x01(\t\x12\x13\n\x0b\x66ragment_id\x18\x05 \x01(\tB0\n\x1c\x63om.snowflake.apps.streamlitB\x10\x43lientStateProtob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!streamlit/proto/ClientState.proto\x1a\"streamlit/proto/WidgetStates.proto\"\xa2\x01\n\x0b\x43lientState\x12\x14\n\x0cquery_string\x18\x01 \x01(\t\x12$\n\rwidget_states\x18\x02 \x01(\x0b\x32\r.WidgetStates\x12\x18\n\x10page_script_hash\x18\x03 \x01(\t\x12\x11\n\tpage_name\x18\x04 \x01(\t\x12\x13\n\x0b\x66ragment_id\x18\x05 \x01(\t\x12\x15\n\ris_auto_rerun\x18\x06 \x01(\x08\x42\x30\n\x1c\x63om.snowflake.apps.streamlitB\x10\x43lientStateProtob\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -24,5 +24,5 @@ if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._loaded_options = None
   _globals['DESCRIPTOR']._serialized_options = b'\n\034com.snowflake.apps.streamlitB\020ClientStateProto'
   _globals['_CLIENTSTATE']._serialized_start=74
-  _globals['_CLIENTSTATE']._serialized_end=213
+  _globals['_CLIENTSTATE']._serialized_end=236
 # @@protoc_insertion_point(module_scope)

streamlit/proto/ClientState_pb2.pyi

@@ -34,10 +34,12 @@ class ClientState(google.protobuf.message.Message):
     PAGE_SCRIPT_HASH_FIELD_NUMBER: builtins.int
     PAGE_NAME_FIELD_NUMBER: builtins.int
     FRAGMENT_ID_FIELD_NUMBER: builtins.int
+    IS_AUTO_RERUN_FIELD_NUMBER: builtins.int
     query_string: builtins.str
     page_script_hash: builtins.str
     page_name: builtins.str
     fragment_id: builtins.str
+    is_auto_rerun: builtins.bool
     @property
     def widget_states(self) -> streamlit.proto.WidgetStates_pb2.WidgetStates: ...
     def __init__(
@@ -48,8 +50,9 @@ class ClientState(google.protobuf.message.Message):
         page_script_hash: builtins.str = ...,
         page_name: builtins.str = ...,
         fragment_id: builtins.str = ...,
+        is_auto_rerun: builtins.bool = ...,
     ) -> None: ...
     def HasField(self, field_name: typing.Literal["widget_states", b"widget_states"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["fragment_id", b"fragment_id", "page_name", b"page_name", "page_script_hash", b"page_script_hash", "query_string", b"query_string", "widget_states", b"widget_states"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["fragment_id", b"fragment_id", "is_auto_rerun", b"is_auto_rerun", "page_name", b"page_name", "page_script_hash", b"page_script_hash", "query_string", b"query_string", "widget_states", b"widget_states"]) -> None: ...
 
 global___ClientState = ClientState

streamlit/runtime/app_session.py

@@ -363,6 +363,7 @@ class AppSession:
                 client_state.page_script_hash,
                 client_state.page_name,
                 fragment_id=fragment_id if fragment_id else None,
+                is_auto_rerun=client_state.is_auto_rerun,
             )
         else:
             rerun_data = RerunData()

streamlit/runtime/fragment.py

@@ -193,7 +193,7 @@ def _fragment(
             ctx = get_script_run_ctx(suppress_warning=True)
             assert ctx is not None
 
-            if ctx.script_requests and ctx.script_requests.fragment_id_queue:
+            if ctx.fragment_ids_this_run:
                 # This script run is a run of one or more fragments. We restore the
                 # state of ctx.cursors and dg_stack to the snapshots we took when this
                 # fragment was declared.

streamlit/runtime/metrics_util.py

@@ -482,8 +482,6 @@ def create_page_profile_message(
         page_profile.uncaught_exception = uncaught_exception
 
     if ctx := get_script_run_ctx():
-        page_profile.is_fragment_run = bool(
-            ctx.script_requests and ctx.script_requests.fragment_id_queue
-        )
+        page_profile.is_fragment_run = bool(ctx.fragment_ids_this_run)
 
     return msg

streamlit/runtime/scriptrunner/exec_code.py

@@ -27,7 +27,13 @@ if TYPE_CHECKING:
 
 def exec_func_with_error_handling(
     func: Callable[[], None], ctx: ScriptRunContext
-) -> tuple[Any | None, bool, RerunData | None, bool]:
+) -> tuple[
+    Any | None,
+    bool,
+    RerunData | None,
+    bool,
+    Exception | None,
+]:
     """Execute the passed function wrapped in a try/except block.
 
     This function is called by the script runner to execute the user's script or
@@ -53,6 +59,7 @@ def exec_func_with_error_handling(
             interrupted by a RerunException.
         - A boolean indicating whether the script was stopped prematurely (False for
             RerunExceptions, True for all other exceptions).
+        - The uncaught exception if one occurred, None otherwise
     """
 
     # Avoid circular imports
@@ -71,6 +78,9 @@ def exec_func_with_error_handling(
     # The result of the passed function
     result: Any | None = None
 
+    # The uncaught exception if one occurred, None otherwise
+    uncaught_exception: Exception | None = None
+
     try:
         result = func()
     except RerunException as e:
@@ -102,5 +112,12 @@ def exec_func_with_error_handling(
         run_without_errors = False
         premature_stop = True
         handle_uncaught_app_exception(ex)
-
-    return result, run_without_errors, rerun_exception_data, premature_stop
+        uncaught_exception = ex
+
+    return (
+        result,
+        run_without_errors,
+        rerun_exception_data,
+        premature_stop,
+        uncaught_exception,
+    )

streamlit/runtime/scriptrunner/script_requests.py

@@ -54,6 +54,7 @@ class RerunData:
     # The queue of fragment_ids waiting to be run.
     fragment_id_queue: list[str] = field(default_factory=list)
     is_fragment_scoped_rerun: bool = False
+    is_auto_rerun: bool = False
 
     def __repr__(self) -> str:
         return util.repr_(self)
@@ -102,13 +103,6 @@ class ScriptRequests:
         self._state = ScriptRequestType.CONTINUE
         self._rerun_data = RerunData()
 
-    @property
-    def fragment_id_queue(self) -> list[str]:
-        if not self._rerun_data:
-            return []
-
-        return self._rerun_data.fragment_id_queue
-
     def request_stop(self) -> None:
         """Request that the ScriptRunner stop running. A stopped ScriptRunner
         can't be used anymore. STOP requests succeed unconditionally.
@@ -180,6 +174,7 @@ class ScriptRequests:
                     page_name=new_data.page_name,
                     fragment_id_queue=fragment_id_queue,
                     is_fragment_scoped_rerun=new_data.is_fragment_scoped_rerun,
+                    is_auto_rerun=new_data.is_auto_rerun,
                 )
 
                 return True

streamlit/runtime/scriptrunner/script_run_context.py

@@ -77,6 +77,7 @@ class ScriptRunContext:
     cursors: dict[int, RunningCursor] = field(default_factory=dict)
     script_requests: ScriptRequests | None = None
     current_fragment_id: str | None = None
+    fragment_ids_this_run: list[str] | None = None
     new_fragment_ids: set[str] = field(default_factory=set)
     # we allow only one dialog to be open at the same time
     has_dialog_opened: bool = False
@@ -100,6 +101,7 @@ class ScriptRunContext:
         self,
         query_string: str = "",
         page_script_hash: str = "",
+        fragment_ids_this_run: list[str] | None = None,
     ) -> None:
         self.cursors = {}
         self.widget_ids_this_run = set()
@@ -115,7 +117,7 @@ class ScriptRunContext:
         self.tracked_commands_counter = collections.Counter()
         self.current_fragment_id = None
         self.current_fragment_delta_path: list[int] = []
-        self.fragment_ids_this_run = None
+        self.fragment_ids_this_run = fragment_ids_this_run
         self.new_fragment_ids = set()
         self.has_dialog_opened = False
         self.disallow_cached_widget_usage = False

streamlit/runtime/scriptrunner/script_runner.py

@@ -425,7 +425,6 @@ class ScriptRunner:
                 rerun_data.page_script_hash, rerun_data.page_name
             )
             main_page_info = self._pages_manager.get_main_page()
-            uncaught_exception = None
 
             page_script_hash = (
                 active_script["page_script_hash"]
@@ -433,6 +432,8 @@ class ScriptRunner:
                 else main_page_info["page_script_hash"]
             )
 
+            fragment_ids_this_run = list(rerun_data.fragment_id_queue)
+
             ctx = self._get_script_run_ctx()
             # Clear widget state on page change. This normally happens implicitly
             # in the script run cleanup steps, but doing it explicitly ensures
@@ -457,6 +458,7 @@ class ScriptRunner:
             ctx.reset(
                 query_string=rerun_data.query_string,
                 page_script_hash=page_script_hash,
+                fragment_ids_this_run=fragment_ids_this_run,
             )
             self._pages_manager.reset_active_script_hash()
 
@@ -555,9 +557,16 @@ class ScriptRunner:
                                 wrapped_fragment()
 
                             except FragmentStorageKeyError:
-                                raise RuntimeError(
-                                    f"Could not find fragment with id {fragment_id}"
-                                )
+                                # Only raise an error if the fragment is not an
+                                # auto_rerun. If it is an auto_rerun, we might have a
+                                # race condition where the fragment_id is removed
+                                # but the webapp sends a rerun request before the
+                                # removal information has reached the web app
+                                # (see https://github.com/streamlit/streamlit/issues/9080).
+                                if not rerun_data.is_auto_rerun:
+                                    raise RuntimeError(
+                                        f"Could not find fragment with id {fragment_id}"
+                                    )
                             except (RerunException, StopException) as e:
                                 # The wrapped_fragment function is executed
                                 # inside of a exec_func_with_error_handling call, so
@@ -586,6 +595,7 @@ class ScriptRunner:
                 run_without_errors,
                 rerun_exception_data,
                 premature_stop,
+                uncaught_exception,
             ) = exec_func_with_error_handling(code_to_exec, ctx)
             # setting the session state here triggers a yield-callback call
             # which reads self._requests and checks for rerun data
@@ -611,7 +621,7 @@ class ScriptRunner:
                     # Create and send page profile information
                     ctx.enqueue(
                         create_page_profile_message(
-                            ctx.tracked_commands,
+                            commands=ctx.tracked_commands,
                             exec_time=to_microseconds(timer() - start_time),
                             prep_time=to_microseconds(prep_time),
                             uncaught_exception=(