Update app.py

app.py

@@ -17,7 +17,7 @@ def get_chatmodel_response(question):
             st.session_state['flowmessages'].append(HumanMessage(content=question))
             answer = chat(st.session_state['flowmessages'])
             st.session_state['flowmessages'].append(AIMessage(content=answer.content))
-            return answer.content
+            return answer.content, answer.audio
         except Exception as e:
             print(f"Error: {e}")
             if "Rate limit" in str(e):
@@ -69,6 +69,7 @@ with st.form(key='my_form',clear_on_submit=True):
     )
 
     input_question = st.text_input("Type here.", key="input")
+    response_text, response_audio = get_chatmodel_response(input_text)
 
     submit = st.form_submit_button("Submit")
 
@@ -91,11 +92,12 @@ if submit:
         input=response
         )
         # Display audio on the webpage
-        st.audio(response_audio['url'], format='audio/mp3')
+        st.audio(response_audio, format='audio/mp3')
 
         # Add a download button for the audio
-        audio_data = base64.b64decode(response_audio['url'].split(',')[1])
-        st.markdown(f"### [Download Audio](data:audio/mp3;base64,{base64.b64encode(audio_data).decode()})", unsafe_allow_html=True)
+        # Add a download button for the audio
+    audio_data = base64.b64decode(response_audio.split(',')[1])
+    st.markdown(f"### [Download Audio](data:audio/mp3;base64,{base64.b64encode(audio_data).decode()})", unsafe_allow_html=True)
 
     if response is not None: