Update app.py

app.py

@@ -3,7 +3,7 @@ SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision</h2>
 
 import os
 import time
-from typing import Optional
+from typing import Optional, Tuple
 
 import google.generativeai as genai
 import gradio as gr
@@ -35,12 +35,13 @@ def bot_response(
     model_choice: str,
     system_instruction: Optional[str],
     text_prompt: str,
-):
+    chatbot: list,
+) -> Tuple[list, str]:
     """
-    Envía el mensaje al modelo y obtiene la respuesta.
+    Envía el mensaje al modelo y obtiene la respuesta, actualizando el historial del chatbot.
     """
     if not text_prompt.strip():
-        return None, "Por favor, escribe un mensaje válido."
+        return chatbot, "Por favor, escribe un mensaje válido."
 
     model = genai.GenerativeModel(
         model_name=model_choice,
@@ -49,14 +50,20 @@ def bot_response(
     )
 
     response = model.generate_content([text_prompt], stream=True, generation_config=generation_config)
-    generated_text = ""
 
+    # Preparar el texto generado
+    generated_text = ""
     for chunk in response:
         for i in range(0, len(chunk.text), 10):  # Mostrar texto en partes
             section = chunk.text[i:i + 10]
             generated_text += section
             time.sleep(0.01)
-            yield text_prompt, generated_text
+            chatbot.append((text_prompt, generated_text))
+            yield chatbot, ""
+
+    # Devolver el historial actualizado
+    chatbot.append((text_prompt, generated_text))
+    return chatbot, ""
 
 # Componentes de la interfaz
 chatbot_component = gr.Chatbot(label="Gemini", scale=2, height=300)
@@ -91,14 +98,14 @@ with gr.Blocks() as demo:
     # Configurar eventos
     run_button_component.click(
         fn=bot_response,
-        inputs=[model_dropdown_component, system_instruction_component, text_input_component],
-        outputs=[chatbot_component],
+        inputs=[model_dropdown_component, system_instruction_component, text_input_component, chatbot_component],
+        outputs=[chatbot_component, text_input_component],
     )
 
     text_input_component.submit(
         fn=bot_response,
-        inputs=[model_dropdown_component, system_instruction_component, text_input_component],
-        outputs=[chatbot_component],
+        inputs=[model_dropdown_component, system_instruction_component, text_input_component, chatbot_component],
+        outputs=[chatbot_component, text_input_component],
     )
 
 # Lanzar la aplicación