Show max_tokens in text area label but prevent value reset

app.py

@@ -84,7 +84,11 @@ and offer some initial analyses and insights, including the potential for studyi
 dependencies.
 """.replace("\n", " ").strip()
 
-text = st.text_area("Input text", DEFAULT_TEXT)
+text = st.text_area(
+    f"Input text (≤\u2009{max_tokens} tokens)",
+    st.session_state.get("input_text", DEFAULT_TEXT),
+    key="input_text",
+)
 
 if tokenizer.eos_token:
     text += tokenizer.eos_token
@@ -92,17 +96,13 @@ inputs = tokenizer([text])
 [input_ids] = inputs["input_ids"]
 num_user_tokens = len(input_ids) - (1 if tokenizer.eos_token else 0)
 
-if num_user_tokens < 1 or num_user_tokens > max_tokens:
-    st.caption(f":red[{num_user_tokens}]/{max_tokens} tokens")
-else:
-    st.caption(f"{num_user_tokens}/{max_tokens} tokens")
-
 if num_user_tokens < 1:
     st.error("Please enter at least one token.", icon="🚨")
     st.stop()
 if num_user_tokens > max_tokens:
     st.error(
-        f"Please enter at most {max_tokens} tokens or try reducing the window size.",
+        f"Your input has {num_user_tokens} tokens. Please enter at most {max_tokens} tokens "
+        f"or try reducing the window size.",
         icon="🚨"
     )
     st.stop()