Update app.py

app.py

@@ -12,26 +12,47 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Modelo de IA de Gemini
 model = genai.GenerativeModel("gemini-2.0-flash")
 
-def chat(message, history):
+def chat_stream(message, history):
     """Envía el mensaje del usuario a Gemini con historial y devuelve la respuesta."""
     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)
-
-        return response.text  # Devuelve solo el texto de la respuesta
+        # 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
+                
     except Exception as e:
-        return f"Error: {e}"
+        yield f"Error: {e}"
 
 # Crear la interfaz de chat con historial
 demo = gr.ChatInterface(
-    fn=chat,
-    examples=["Write an example Python lambda function."],
-    title="Gemini Chatbot",
-    description="Chatbot interactivo con historial de conversación usando Gemini AI."
+    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"
 )
 
-# Iniciar la app
-demo.launch()
+# Iniciar la app con configuración mejorada
+demo.launch(
+    share=False,
+    height=600,
+    show_api=False,
+    allow_flagging="never",
+    enable_cors=True,
+    server_name="0.0.0.0",
+    server_port=7860
+)