button not active

app.py

@@ -20,11 +20,13 @@ def remove_citation(text: str) -> str:
     pattern = r"【\d+†\w+】"
     return re.sub(pattern, "📚", text)
 
-# Initialize session state for messages and thread_id
+# Initialize session state for messages, thread_id and processing state
 if "messages" not in st.session_state:
     st.session_state["messages"] = []
 if "thread_id" not in st.session_state:
     st.session_state["thread_id"] = None
+if "is_processing" not in st.session_state:
+    st.session_state["is_processing"] = False
 
 st.title("Solution Specifier A")
 
@@ -54,23 +56,35 @@ for msg in st.session_state["messages"]:
             st.write(msg["content"])
 
 # Create the chat input widget at the bottom of the page
-user_input = st.chat_input("Type your message here...")
+# Disable it while processing a message
+user_input = st.chat_input(
+    "Type your message here...",
+    disabled=st.session_state["is_processing"]
+)
 
 # When the user hits ENTER on st.chat_input
-if user_input:
-    # Add the user message to session state
-    st.session_state["messages"].append({"role": "user", "content": user_input})
-    
-    # Display the user's message
-    with st.chat_message("user"):
-        st.write(user_input)
+if user_input and not st.session_state["is_processing"]:
+    try:
+        # Set processing state to True
+        st.session_state["is_processing"] = True
+        
+        # Add the user message to session state
+        st.session_state["messages"].append({"role": "user", "content": user_input})
+        
+        # Display the user's message
+        with st.chat_message("user"):
+            st.write(user_input)
+            
+        # Get the assistant's response
+        response_text = predict(user_input)
+        
+        # Add the assistant response to session state
+        st.session_state["messages"].append({"role": "assistant", "content": response_text})
         
-    # Get the assistant's response
-    response_text = predict(user_input)
-    
-    # Add the assistant response to session state
-    st.session_state["messages"].append({"role": "assistant", "content": response_text})
-    
-    # Display the assistant's reply
-    with st.chat_message("assistant"):
-        st.write(response_text)
+        # Display the assistant's reply
+        with st.chat_message("assistant"):
+            st.write(response_text)
+            
+    finally:
+        # Reset processing state when done
+        st.session_state["is_processing"] = False