Update app.py

app.py

@@ -12,46 +12,26 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Modelo de IA de Gemini
 model = genai.GenerativeModel("gemini-2.0-flash")
 
-def chat_stream(message, history):
+def chat(message, history):
     """Envía el mensaje del usuario a Gemini con historial y devuelve la respuesta."""
     try:
-        # Crear una nueva sesión de chat
-        chat = model.start_chat()
-        
-        # Procesar el historial previo
-        for user_msg, assistant_msg in history:
-            chat.send_message(user_msg)
-        
-        # Enviar el mensaje actual y obtener respuesta en streaming
-        response = chat.send_message(message, stream=True)
-        
-        # Devolver la respuesta en fragmentos
-        for chunk in response:
-            if chunk.text:
-                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,
-    examples=[
-        "What is Python?",
-        "Write a simple function to calculate factorial",
-        "How can I read a CSV file in Python?",
-        "Explain object-oriented programming"
-    ],
-    title="Gemini AI Assistant",
-    description="Chat interactivo con Gemini AI - Pregunta lo que quieras sobre programación"
+    fn=chat,
+    examples=["Write an example Python lambda function."],
+    title="Gemini Chatbot",
+    description="Chatbot interactivo con historial de conversación usando Gemini AI."
 )
 
-# Iniciar la app con configuración mejorada
-demo.launch(
-    share=False,
-    height=600,
-    show_api=False,
-    enable_cors=True,
-    server_name="0.0.0.0",
-    server_port=7860
-)
+# Iniciar la app
+demo.launch()