Upload 12 files

app.py

@@ -71,12 +71,8 @@ def process_message(prompt, is_example=False):
             typing_indicator = st.empty()
             typing_indicator.markdown("*Generando respuesta...*")
             
-            # Asegurar que siempre usamos streaming
-            response = state.chat.send_message(
-                enhanced_prompt,
-                stream=True,
-                generation_config={"temperature": 0.9}
-            )
+            # Usar el nuevo método send_message
+            response = state.send_message(enhanced_prompt)
             
             full_response = ''
             
@@ -107,7 +103,7 @@ def process_message(prompt, is_example=False):
                 avatar=AI_AVATAR_ICON
             )
             
-            # Guardar el historial actualizado (primera llamada)
+            # Guardar el historial actualizado
             state.save_chat_history()
             
         except Exception as e:

session_state.py

@@ -110,10 +110,51 @@ class SessionState:
         self.model = genai.GenerativeModel(model_name)
     
     def initialize_chat(self, history=None):
-        """Inicializa el chat con el modelo"""
+        """Inicializa el chat con el modelo y asegura la persistencia del streaming"""
         if history is None:
             history = self.gemini_history
-        self.chat = self.model.start_chat(history=history)
+        
+        # Asegurar que el modelo está inicializado
+        if self.model is None:
+            self.initialize_model()
+            
+        # Configurar el chat con streaming activado
+        self.chat = self.model.start_chat(
+            history=history,
+            generation_config={
+                "temperature": 0.9,
+                "stream": True  # Forzar streaming siempre
+            }
+        )
+        
+        # Verificar que el chat se inicializó correctamente
+        if self.chat is None:
+            raise ValueError("Error al inicializar el chat")
+    
+    def send_message(self, prompt, stream=True):
+        """Método unificado para enviar mensajes y mantener el streaming"""
+        try:
+            if self.chat is None:
+                self.initialize_chat()
+                
+            return self.chat.send_message(
+                prompt,
+                stream=stream,  # Respetar el parámetro de streaming
+                generation_config={
+                    "temperature": 0.9
+                }
+            )
+        except Exception as e:
+            print(f"Error al enviar mensaje: {e}")
+            # Reintentar una vez si hay error
+            self.initialize_chat()
+            return self.chat.send_message(
+                prompt,
+                stream=stream,
+                generation_config={
+                    "temperature": 0.9
+                }
+            )
     
     def generate_chat_title(self, prompt, model_name='gemini-2.0-flash'):
         """Genera un título para el chat basado en el primer mensaje"""