Update app.py

app.py

@@ -1,8 +1,11 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
+from gtts import gTTS
+import IPython.display as ipd
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
-system_prompt = "Tus principios son el estoicismo, la bioética, ecología profunda, activismo social y transhumanismo."
+system_prompt = "Tus principios básicos son el estoicismo bioético, ecología y el mindfullness, la ciencias holísticas, el activismo social y transhumanismo."
+system_prompt_sent = False  # Variable para controlar si el sistema prompt ya se envió
 
 def format_prompt(message, history):
     prompt = "<s>"
@@ -12,10 +15,8 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-def generate(
-    prompt, history, temperature=0.9, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0,
-):
-    global system_prompt
+def generate(prompt, history, temperature=0.9, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0):
+    global system_prompt, system_prompt_sent
 
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -31,17 +32,24 @@ def generate(
         seed=42,
     )
 
-    if not system_prompt:
+    if not system_prompt_sent:
         formatted_prompt = format_prompt(prompt, history)
+        system_prompt_sent = True
     else:
         formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
+
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
-    output = ""
+    output_text = ""
 
     for response in stream:
-        output += response.token.text
-        yield output
-    return output
+        output_text += response.token.text
+        yield output_text
+
+    tts = gTTS(output_text, lang='es')
+    tts.save('output.mp3')
+    ipd.display(ipd.Audio('output.mp3'))
+
+    return output_text
 
 chat_interface = gr.ChatInterface(
     fn=generate,