Update pages.py

pages.py

@@ -214,8 +214,19 @@ def audio_llm():
 
     if "messages" not in st.session_state:
         st.session_state.messages = []
-    
-    if prompt := st.chat_input(placeholder="Your Instruction"):
+
+    if 'disprompt' not in st.session_state:
+        st.session_state.disprompt = False
+
+    for message in st.session_state.messages[-2:]:
+        with st.chat_message(message["role"]):
+            st.write(message["content"])
+
+    if prompt := st.chat_input(
+        placeholder="Your Instruction", 
+        disabled=st.session_state.disprompt, 
+        on_submit=lambda: st.session_state.update(disprompt=True)
+    ):
         with st.chat_message("user"):
             st.write(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
@@ -234,3 +245,6 @@ def audio_llm():
                         st.write(response)
                         raise(e)
             st.session_state.messages.append({"role": "assistant", "content": response})
+
+        st.session_state.disprompt = False
+        st.rerun()