Update pages.py

pages.py

@@ -49,7 +49,7 @@ def sidebar_fragment():
 
 @st.fragment
 def specify_audio_fragment():
-    col1, col2, col3 = st.columns([3.5, 4, 1.5])
+    col1, col2, col3 = st.columns([4, 2, 2])
 
     with col1:        
         audio_sample_names = [audio_sample_name for audio_sample_name in AUDIO_SAMPLES_W_INSTRUCT.keys()]
@@ -121,7 +121,7 @@ def specify_audio_fragment():
                     f"**Example Instruction {i+1}**: {inst}", 
                     args=(inst,),
                     disabled=st.session_state.disprompt, 
-                    on_click=lambda p: st.session_state.update(disprompt=True, new_prompt=p, on_click_button=True)
+                    on_click=lambda p: st.session_state.update(disprompt=True, new_prompt=p, on_click_button=True, messages=[])
                 )
 
     if st.session_state.on_click_button:
@@ -142,7 +142,7 @@ def dialogue_section():
     if chat_input := st.chat_input(
         placeholder="Type Your Instruction Here", 
         disabled=st.session_state.disprompt, 
-        on_submit=lambda: st.session_state.update(disprompt=True)
+        on_submit=lambda: st.session_state.update(disprompt=True, messages=[])
     ):
         st.session_state.new_prompt = chat_input
 
@@ -200,4 +200,4 @@ def audio_llm():
     )
 
     specify_audio_fragment()
-    dialogue_section()
+    dialogue_section()