Update app.py

app.py

@@ -3,12 +3,10 @@ from huggingface_hub import InferenceClient
 from gtts import gTTS
 import base64
 from pydub import AudioSegment
-import io
 from pydub.playback import play
 
-st.sidebar.width = 100
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
-pre_prompt = "Te llamarás Xaman 3.0 (NO LO REPITAS) tu rol como IA conductual, tus principios son: existencialismo estóico y holístico."
+pre_prompt = "Tu nombre será Chaman 3.0 una IA conductual, hablarás español tu rol es la bioética y el estoicismo holístico."
 pre_prompt_sent = False
 
 def format_prompt(message, history):
@@ -36,7 +34,7 @@ def text_to_speech(text, speed=1.3):
 
     return audio_file_path
 
-def generate(user_input, history, temperature=None, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
+def generate(user_input, history, temperature=None, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0):
     global pre_prompt_sent
     temperature = float(temperature) if temperature is not None else 0.9
     if temperature < 1e-2:
@@ -63,26 +61,21 @@ def generate(user_input, history, temperature=None, max_new_tokens=512, top_p=0.
 
     audio_file_path = text_to_speech(response, speed=1.3) 
     audio_file = open(audio_file_path, 'rb')
-    audio_bytes = io.BytesIO(open(audio_file_path, 'rb').read())
+    audio_bytes = audio_file.read()
 
     return response, audio_bytes
 
 if "history" not in st.session_state:
     st.session_state.history = []
 
-if st.session_state.history:
-    user_input = st.session_state.history[-1][0]
+with st.beta_container():
+    user_input = st.text_input(label="", value="")
     output, audio_bytes = generate(user_input, history=st.session_state.history)
-    st.session_state.history[-1] = (user_input, output)
-else:
-    output, audio_bytes = "", ""
 
-st.text_area("Respuesta de la IA", value=output, height=500, max_chars=600, key="output_text", disabled=True)
-user_input = st.text_input(label="Ingresa tu mensaje", value="")
+    if user_input:
+        st.session_state.history.append((user_input, output))
 
-if user_input:
-    output, audio_bytes = generate(user_input, history=st.session_state.history)
-    st.session_state.history.append((user_input, output))
+    st.text_area("Salida del Chatbot", value=output, height=500, max_chars=600, key="output_text", disabled=True)
 
 st.markdown(
     f"""