Update app.py

app.py

@@ -4,7 +4,7 @@ from gtts import gTTS
 import base64
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
-system_prompt = "Tu nombre será Chaman 3.0 una IA conductual."
+system_prompt = "Tu nombre es Chaman 3.0 una IA conductual"
 system_prompt_sent = False
 
 def format_prompt(message, history):
@@ -46,10 +46,10 @@ def generate(
         do_sample=True,
         seed=42,
     )
-   
+
     formatted_prompt = format_prompt(user_input, history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
-    
+
     response = ""
     for response_token in stream:
         response += response_token.token.text
@@ -59,25 +59,26 @@ def generate(
     audio_file_path = text_to_speech(response)
     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, key="output_text", disabled=True)
-    
-        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
-        )
 
-        for user_prompt, bot_response in 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))
+
+        for user_prompt, bot_response in st.session_state.history:
             st.write(f"Usuario: {user_prompt}")
             st.write(f"Respuesta: {bot_response}")
-            st.markdown("---")
-
-if "history" not in st.session_state:
-    st.session_state.history = []
+            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
+            )
 
-user_input = st.text_input(label="Usuario", value="")
-output = generate(user_input, history=st.session_state.history)
-st.session_state.history.append((user_input, output))
+if __name__ == "__main__":
+    generate(user_input="", history=[])