Update app.py

Disable chat_input while generating

app.py

@@ -8,8 +8,10 @@ REPETITION_PENALTY = 1.05
 MAX_NEW_TOKENS = 500
 MODEL_NAME = "ericzzz/falcon-rw-1b-chat"
 
+# fmt: off
 st.write("**💬Tiny Chat with [Falcon-RW-1B-Chat](https://huggingface.co/ericzzz/falcon-rw-1b-chat)**" ) 
 
+# fmt: on
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
@@ -103,6 +105,8 @@ class ResponseStreamer:
         # clean up states (actually not needed as the instance will get recreated)
         self.first_call_to_put = True
         self.current_response = ""
+        # rerun to unfreeze the chat_input
+        st.rerun()
 
 
 tokenizer, model = load_model()
@@ -111,8 +115,13 @@ chat_messages_container = st.container()
 for msg in st.session_state.chat_history:
     show_chat_message(chat_messages_container, msg)
 
-user_input = st.chat_input()
+input_placeholder = st.empty()  # use placeholder as a hack to disable input
+user_input = input_placeholder.chat_input(key="user_input_original")
+
 if user_input:
+    # disable chat_input while generating
+    input_placeholder.chat_input(key="user_input_disabled", disabled=True)
+
     new_user_message = {"role": "user", "content": user_input}
     st.session_state.chat_history.append(new_user_message)
     show_chat_message(chat_messages_container, new_user_message)