Make seed configurable

app.py

@@ -87,12 +87,6 @@ def instantiate_models():
             p["pipeline"].load()
 
 
-def set_new_seed():
-    seed = randint(0, 2**32 - 1)
-    set_seed(seed)
-    return seed
-
-
 def main():
     st.set_page_config(  # Alternate names: setup_page, page, layout
         page_title="Netherator",  # String or None. Strings get appended with "• Streamlit".
@@ -122,9 +116,6 @@ def main():
 
     st.session_state["text"] = st.text_area("Enter text", st.session_state.prompt_box)
 
-    # min_length = st.sidebar.number_input(
-    #     "Min length", min_value=10, max_value=150, value=75
-    # )
     max_length = st.sidebar.number_input(
         "Lengte van de tekst",
         value=200,
@@ -140,8 +131,28 @@ def main():
         "Num return sequences", min_value=1, max_value=5, value=1
     )
 
+    seed_placeholder = st.sidebar.empty()
+    if "seed" not in st.session_state:
+        print(f"Session state {st.session_state} does not contain seed")
+        st.session_state["seed"] = 4162549114
+        print(f"Seed is set to: {st.session_state['seed']}")
+
+    seed = seed_placeholder.number_input(
+        "Seed", min_value=0, max_value=2 ** 32 - 1, value=st.session_state["seed"]
+    )
+
+    def set_random_seed():
+        st.session_state["seed"] = randint(0, 2 ** 32 - 1)
+        seed = seed_placeholder.number_input(
+            "Seed", min_value=0, max_value=2 ** 32 - 1, value=st.session_state["seed"]
+        )
+        print(f"New random seed set to: {seed}")
+
+    if st.button("New random seed?"):
+        set_random_seed()
+
     if sampling_mode := st.sidebar.selectbox(
-        "select a Mode", index=0, options=["Top-k Sampling", "Beam Search"]
+            "select a Mode", index=0, options=["Top-k Sampling", "Beam Search"]
     ):
         if sampling_mode == "Beam Search":
             num_beams = st.sidebar.number_input(
@@ -200,7 +211,7 @@ and the [Huggingface text generation interface doc](https://huggingface.co/trans
         estimate = int(estimate)
 
         with st.spinner(
-            text=f"Please wait ~ {estimate} second{'s' if estimate != 1 else ''} while getting results ..."
+                text=f"Please wait ~ {estimate} second{'s' if estimate != 1 else ''} while getting results ..."
         ):
             memory = psutil.virtual_memory()
             generator = next(
@@ -211,7 +222,7 @@ and the [Huggingface text generation interface doc](https://huggingface.co/trans
                 ),
                 None,
             )
-            seed = set_new_seed()
+            set_seed(seed)
             time_start = time.time()
             result = generator.get_text(text=st.session_state.text, **params)
             time_end = time.time()