Update app.py

app.py

@@ -34,6 +34,7 @@ def take_user_input():
         query = 'None'
 
     return query
+
 def format_prompt(message, history):
     global pre_prompt_sent
     prompt = "<s>"
@@ -85,21 +86,20 @@ def text_to_speech(text):
     return audio_stream.read()
 
 def speak(text):
-    audio_file_path = text_to_speech(text)
-    audio_file = open(audio_file_path, 'rb')
-    audio_bytes = audio_file.read()
+    audio_bytes = text_to_speech(text)
     st.audio(audio_bytes, format="audio/mp3", start_time=0, key="audio_player")
 
 if "history" not in st.session_state:
     st.session_state.history = []
 
-user_input = take_user_input()
-output, audio_bytes = generate(user_input, history=st.session_state.history)
-
 if user_input != 'None':
     st.session_state.history.append((user_input, output))
 
-st.text_area("Respuesta", value=output, height=400, key="output_text", disabled=True)
+with st.container(weight=900, height=400):
+    user_input_container = st.text_input("Tu entrada de usuario", value=user_input)
+    st.text_area("Respuesta", value=output, key="output_text", disabled=True)
+    user_input = take_user_input()
+    output, audio_bytes = generate(user_input, history=st.session_state.history)
 
 if audio_bytes is not None:
     st.markdown(