Update app.py

app.py

@@ -29,9 +29,7 @@ def text_to_speech(text, speed=2.0):
     tts.save(audio_file_path)
     return audio_file_path
 
-def generate(
-    user_input, history, temperature=None, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0,
-):
+def generate(user_input, history, temperature=None, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0):
     global system_prompt_sent
     temperature = float(temperature) if temperature is not None else 0.9
     if temperature < 1e-2:
@@ -60,25 +58,25 @@ def generate(
     audio_file = open(audio_file_path, 'rb')
     audio_bytes = audio_file.read()
 
-    with st.container():
-        st.text_area("Salida del Chatbot", value=response, height=200, max_chars=500, disabled=True)
+    return response, audio_bytes
 
-        if "history" not in st.session_state:
-            st.session_state.history = []
+if "history" not in st.session_state:
+    st.session_state.history = []
 
-        user_input = st.text_input(label="Usuario", value="Escribe aquí tu mensaje")
-        output = generate(user_input, history=st.session_state.history)
-        st.session_state.history.append((user_input, output))
+user_input = st.text_input(label="Usuario", value="")
+output, audio_bytes = generate(user_input, history=st.session_state.history)
+st.session_state.history.append((user_input, output))
 
-        for user_prompt, bot_response in st.session_state.history:
-            st.write(f"Usuario: {user_prompt}")
-            st.write(f"Respuesta: {bot_response}")
-            st.markdown(
-                f"""
-                <audio autoplay="autoplay" controls="controls" src="data:audio/mp3;base64,{base64.b64encode(audio_bytes).decode()}" type="audio/mp3" speed="1.5" id="audio_player"></audio>
-                """,
-                unsafe_allow_html=True
-            )
+with st.container():
+    st.text_area("Salida del Chatbot", value=output, height=200, max_chars=500, key="output_text", disabled=True)
 
-if __name__ == "__main__":
-    generate(user_input="", history=[])
+    for i, (user_prompt, bot_response) in enumerate(st.session_state.history):
+        st.write(f"Usuario {i + 1}: {user_prompt}")
+        st.write(f"Respuesta {i + 1}: {bot_response}")
+    
+    st.markdown(
+        f"""
+        <audio autoplay="autoplay" controls="controls" src="data:audio/mp3;base64,{base64.b64encode(audio_bytes).decode()}" type="audio/mp3" speed="1.5" id="audio_player"></audio>
+        """,
+        unsafe_allow_html=True
+    )