Update app.py

app.py

@@ -1,6 +1,7 @@
 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 = "Tu nombre es Xaman 3.0"
@@ -27,7 +28,7 @@ def text_to_speech(text):
     return 'output.mp3'
 
 def generate(
-    prompt, history, temperature=None, max_new_tokens=2048, top_p=0.95, repetition_penalty=1.0,
+    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
@@ -44,7 +45,7 @@ def generate(
         seed=42,
     )
 
-    formatted_prompt = format_prompt(prompt, history)
+    formatted_prompt = format_prompt(user_input, history)
     
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
     output = ""
@@ -56,19 +57,20 @@ def generate(
     text_to_speech(output)
     return output
 
+input_text = gr.Textbox(label="Usuario", placeholder="Escribe aquí tu mensaje")
 output_audio = gr.Audio(type="file", label="Respuesta del chatbot en audio", autoplay=True)
-output_text = gr.Textbox(label="Respuesta del chatbot en texto")
+output_text = gr.Textbox(label="Respuesta del chatbot en texto", readonly=True)
+
+def update_output_text(output):
+    output_text.update(output)
 
 chat_interface = gr.Interface(
     fn=generate,
-    inputs=["text"],
+    inputs=[input_text],
     outputs=[output_audio, output_text],
-    theme="soft",
     live=True,
-    retry_btn=None,
-    undo_btn=None,
-    clear_btn=None,
+    theme="soft",
     submit_btn="Enviar",
 )
 
-chat_interface.launch()
+chat_interface.launch(share=True, debug=True, capture_session=True, callback=update_output_text)