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
@@ -7,50 +6,32 @@ from dotenv import load_dotenv
 # Cargar variables de entorno
 load_dotenv()
 
-# Configurar la API de Google con la clave
+# Configurar la API de Google
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Crear la sesión de chat con el modelo de Gemini
+# Modelo de IA de Gemini
 model = genai.GenerativeModel("gemini-2.0-flash")
 
-def chat_stream(message, history):
-    """Envía el mensaje del usuario a Gemini con historial y devuelve la respuesta en streaming."""
+def chat(message, history):
+    """Envía el mensaje del usuario a Gemini con historial y devuelve la respuesta."""
     try:
-        # Crear una conversación nueva
-        chat = model.start_chat()
-        
-        # Agregar el historial previo a la conversación
-        for user_msg, assistant_msg in history:
-            chat.send_message(user_msg)
-        
-        # Enviar el mensaje actual y obtener la respuesta en streaming
-        response = chat.send_message(message, stream=True)
-        
-        # Devolver los fragmentos como un flujo
-        for chunk in response:
-            if chunk.text:
-                time.sleep(0.05)  # Añade un pequeño retraso de 50ms entre chunks
-                yield chunk.text
-        
+        # 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)
+
+        return response.text  # Devuelve solo el texto de la respuesta
     except Exception as e:
-        yield f"Error: {e}"
+        return f"Error: {e}"
 
 # Crear la interfaz de chat con historial
 demo = gr.ChatInterface(
-    fn=chat_stream,
+    fn=chat,
     examples=["Write an example Python lambda function."],
     title="Gemini Chatbot",
-    description="Chatbot interactivo con historial de conversación usando Gemini AI.",
-    height=600,  # Aumenta la altura del chat
-    fill_height=True,  # Permite que el chat ocupe todo el espacio disponible
-    bubble_full_width=False,  # Ajusta el ancho de las burbujas de chat
-    layout="panel"  # Usa un diseño de panel para mejor visualización
+    description="Chatbot interactivo con historial de conversación usando Gemini AI."
 )
 
 # Iniciar la app
-demo.launch(
-    show_api=False,
-    share=False,
-    height=800,  # Altura total de la ventana
-    allow_flagging="never"
-)
+demo.launch()