Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import time
 import gradio as gr
 import google.generativeai as genai
 from dotenv import load_dotenv
@@ -14,28 +13,24 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 model = genai.GenerativeModel("gemini-2.0-flash")
 
 def chat(message, history):
-    """Envía el mensaje del usuario a Gemini con historial y devuelve la respuesta progresiva."""
+    """Envía el mensaje del usuario a Gemini con historial y devuelve la respuesta en streaming."""
     try:
         # Convertir historial a formato adecuado para Gemini
         chat_history = [{"role": "user", "parts": [msg[0]]} for msg in history] + [{"role": "user", "parts": [message]}]
 
-        # Enviar el historial y el mensaje actual a Gemini
-        response = model.generate_content(chat_history)
-
-        # Simular respuesta progresiva caracter por caracter
-        full_response = ""
-        for i in range(len(response.text)):
-            time.sleep(0.05)
-            full_response = response.text[: i + 1]
-            yield history + [(message, full_response)]  # Retorna el historial actualizado progresivamente
-
+        # Usar streaming para obtener la respuesta en fragmentos
+        response_stream = model.generate_content_stream(chat_history)
+        
+        # Devolver los fragmentos como un flujo (streaming) en el chat
+        for chunk in response_stream:
+            yield chunk.text  # Genera cada fragmento uno por uno
+        
     except Exception as e:
-        yield history + [(message, f"Error: {e}")]
+        yield f"Error: {e}"
 
 # Crear la interfaz de chat con historial
 demo = gr.ChatInterface(
     fn=chat,
-    type="messages",
     examples=["Write an example Python lambda function."],
     title="Gemini Chatbot",
     description="Chatbot interactivo con historial de conversación usando Gemini AI."