Update app.py

app.py

@@ -2,10 +2,14 @@ import streamlit as st
 from huggingface_hub import InferenceClient
 from gtts import gTTS
 
+# Inicializar el cliente de inferencia
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
+
+# Definir el prompt del sistema
 system_prompt = "Tu nombre es Xaman 3.0"
 system_prompt_sent = False
 
+# Función para formatear el prompt
 def format_prompt(message, history):
     global system_prompt_sent
     prompt = "<s>"
@@ -22,11 +26,13 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
+# Función para convertir texto a audio
 def text_to_speech(text, speed=1.5):
     tts = gTTS(text=text, lang='es', speed=speed)
     tts.save('output.mp3')
     return 'output.mp3'
 
+# Función para generar respuesta
 def generate(
     user_input, history, temperature=None, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0,
 ):
@@ -49,26 +55,34 @@ def generate(
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
     response = " ".join(response.token.text for response in stream)
 
-    st.text(response)  
+    # Mostrar respuesta en un área de texto
+    st.text_area("Bot:", value=response, height=200, key='response_area')  
+
+    # Convertir respuesta a audio y reproducirlo
     audio_file = text_to_speech(response, speed=1.2) 
     st.audio(audio_file, format="audio/mp3", start_time=0, key='audio')
 
     return response
 
+# Inicializar historial si no existe
 if "history" not in st.session_state:
     st.session_state.history = []
 
+# Interfaz de usuario con Streamlit
 st.title("Chatbot Interactivo")
 user_input = st.text_area(label="Usuario", value="Escribe aquí tu mensaje", height=200)
 
+# Mostrar historial de conversación
 st.subheader("Historial de Conversación")
 for user_prompt, bot_response in st.session_state.history:
     st.write(f"Usuario: {user_prompt}")
     st.write(f"Bot: {bot_response}")
     st.markdown("---")
 
+# Generar respuesta y actualizar historial
 output = generate(user_input, history=st.session_state.history)
 st.session_state.history.append((user_input, output))
 
+# Reproducir respuesta en formato de audio
 audio_file = text_to_speech(output, speed=1.2)
-st.audio(audio_file, format="audio/mp3", start_time=0, key='audio')
+st.audio(audio_file, format="audio/mp3", start_time=0, key='audio')