Update app.py

app.py

@@ -58,6 +58,18 @@ def predict(message, system_prompt='', temperature=0.7, max_new_tokens=4096,Topp
 st.title(TITLE)
 st.write(DESCRIPTION)
 
+
+
+
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"], avatar=("🧑‍💻" if message["role"] == 'human' else '🦙')):
+        st.markdown(message["content"])
+
 textinput = st.chat_input("Ask LLama-2-70b anything...")
 wav_audio_data = st_audiorec()
 
@@ -66,23 +78,13 @@ if wav_audio_data != None:
     with open("audio.wav", "wb") as f:
         f.write(wav_audio_data)
     # transcribe audio
-    response = transcribe("audio.wav")
-    
+    response = predict(message=transcribe("audio.wav")) 
+
     with st.chat_message("assistant", avatar='🦙'):
         st.markdown(response)
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
 
-
-
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# Display chat messages from history on app rerun
-for message in st.session_state.messages:
-    with st.chat_message(message["role"], avatar=("🧑‍💻" if message["role"] == 'human' else '🦙')):
-        st.markdown(message["content"])
-        
 # React to user input
 if prompt := textinput:
     # Display user message in chat message container
@@ -95,4 +97,8 @@ if prompt := textinput:
     with st.chat_message("assistant", avatar='🦙'):
         st.markdown(response)
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})
+    
+
+    
+