Upload app.py

app.py

@@ -309,156 +309,70 @@ for message in st.session_state.messages:
         st.markdown(message['content'])
 
 # React to user input
-if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
-    # Verificar si es el primer mensaje del usuario
-    is_first_message = len(st.session_state.messages) == 0
-    
-    # Guardar información del chat
-    if st.session_state.current_chat_id not in st.session_state.chats_in_memory:
-        # Es una nueva conversación, generamos un título basado en el primer mensaje
-        temp_title = f'SesiónChat-{st.session_state.current_chat_id}'
-        
-        # Generamos un título basado en el contenido del mensaje
-        try:
-            title_generator = genai.GenerativeModel('gemini-2.0-flash')
-            title_response = title_generator.generate_content(
-                f"Genera un título corto (máximo 5 palabras) que describa de qué trata esta consulta, sin usar comillas ni puntuación: '{prompt}'")
-            
-            generated_title = title_response.text.strip()
-            
-            if generated_title:
-                st.session_state.chat_title = generated_title
-            else:
-                st.session_state.chat_title = temp_title
-        except Exception as e:
-            print(f"Error al generar título: {e}")
-            st.session_state.chat_title = temp_title
-        
-        # Guardar en memoria
-        st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
-            'messages': st.session_state.messages,
-            'gemini_history': st.session_state.gemini_history,
-            'title': st.session_state.chat_title
-        }
-    
-    # Mostrar mensaje del usuario
-    with st.chat_message('user', avatar=USER_AVATAR_ICON):
-        st.markdown(prompt)
-    
-    # Añadir mensaje del usuario al historial
-    st.session_state.messages.append({
-        'role': 'user',
-        'content': prompt,
-        'avatar': USER_AVATAR_ICON
+# Funciones auxiliares para manejo de mensajes y errores
+def add_message(role, content, avatar):
+    """Función centralizada para añadir mensajes al historial"""
+    message = {
+        'role': role,
+        'content': content,
+        'avatar': avatar
+    }
+    st.session_state.messages.append(message)
+    return message
+
+def update_chat_memory():
+    """Función centralizada para actualizar el chat en memoria"""
+    st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
+        'messages': st.session_state.messages,
+        'gemini_history': st.session_state.gemini_history,
+        'title': st.session_state.chat_title
     })
-    
-    # Si es el primer mensaje, mostrar el mensaje de bienvenida
-    if is_first_message:
+
+def handle_model_error(error, retry_count, max_retries):
+    """Función centralizada para manejar errores del modelo"""
+    if retry_count >= max_retries:
+        error_message = f"Lo siento, estoy experimentando problemas para procesar tu solicitud. Por favor, intenta de nuevo más tarde. Error: {str(error)}"
         with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
-            st.markdown(WELCOME_MESSAGE)
-        
-        st.session_state.messages.append({
-            'role': MODEL_ROLE,
-            'content': WELCOME_MESSAGE,
-            'avatar': AI_AVATAR_ICON,
-        })
-        
-        # Actualizar chat en memoria
-        st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
-            'messages': st.session_state.messages,
-            'gemini_history': st.session_state.gemini_history,
-            'title': st.session_state.chat_title
-        })
-        st.rerun()
-    
-    # Implementación de reintentos con retroceso exponencial
-    max_retries = 3
+            st.error(error_message)
+        add_message(MODEL_ROLE, error_message, AI_AVATAR_ICON)
+        update_chat_memory()
+        return True
+    return False
+
+def process_model_response(prompt, max_retries=3):
+    """Función centralizada para procesar respuestas del modelo"""
     retry_count = 0
     while retry_count < max_retries:
         try:
-            # Enviar mensaje al modelo
             response = st.session_state.chat.send_message(prompt, stream=True)
-            
-            # Procesar la respuesta completa
-            full_text = ""
-            for chunk in response:
-                full_text += chunk.text
-            
-            # Mostrar respuesta del asistente con efecto de escritura
+            full_text = "".join(chunk.text for chunk in response)
             mensaje_mostrado = mostrar_con_efecto_escritura(full_text, velocidad=0.05)
             
-            # Añadir respuesta al historial
-            st.session_state.messages.append({
-                'role': MODEL_ROLE,
-                'content': mensaje_mostrado,
-                'avatar': AI_AVATAR_ICON,
-            })
-            
-            # Actualizar historial
+            add_message(MODEL_ROLE, mensaje_mostrado, AI_AVATAR_ICON)
             st.session_state.gemini_history = st.session_state.chat.history
-            
-            # Actualizar chat en memoria
-            st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
-                'messages': st.session_state.messages,
-                'gemini_history': st.session_state.gemini_history,
-                'title': st.session_state.chat_title
-            })
-            
-            break
+            update_chat_memory()
+            return True
             
         except Exception as e:
             retry_count += 1
-            if retry_count >= max_retries:
-                error_message = f"Lo siento, estoy experimentando problemas para procesar tu solicitud. Por favor, intenta de nuevo más tarde. Error: {str(e)}"
-                
-                with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
-                    st.error(error_message)
-                
-                st.session_state.messages.append({
-                    'role': MODEL_ROLE,
-                    'content': error_message,
-                    'avatar': AI_AVATAR_ICON,
-                })
-                
-                # Actualizar chat en memoria con el error
-                st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
-                    'messages': st.session_state.messages,
-                    'gemini_history': st.session_state.gemini_history,
-                    'title': st.session_state.chat_title
-                })
-            else:
-                wait_time = (2 ** retry_count) + (time.time() % 1)
-                time.sleep(wait_time)
-            
-            # Salir del bucle si la solicitud fue exitosa
-            break
+            if handle_model_error(e, retry_count, max_retries):
+                return False
+            wait_time = (2 ** retry_count) + (time.time() % 1)
+            time.sleep(wait_time)
+    return False
 
-# Título y subtítulo principales
-# Inicializar el estado para los ejemplos
-if 'show_examples' not in st.session_state:
-    st.session_state.show_examples = True
-
-# Función para manejar los clics de los botones
+# Modificar la función handle_example_click para usar las nuevas funciones
 def handle_example_click(prompt_text):
-    # Reiniciar estado
     st.session_state.update({
         'show_examples': False,
         'messages': [],
         'current_chat_id': str(time.time())
     })
     
-    # Añadir mensaje del usuario
-    message = {
-        'role': 'user',
-        'content': prompt_text,
-        'avatar': USER_AVATAR_ICON
-    }
-    st.session_state.messages.append(message)
-    
+    add_message('user', prompt_text, USER_AVATAR_ICON)
     with st.chat_message('user', avatar=USER_AVATAR_ICON):
         st.markdown(prompt_text)
     
-    # Actualizar chat en memoria
     if st.session_state.current_chat_id not in st.session_state.chats_in_memory:
         st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
             'messages': st.session_state.messages,
@@ -466,32 +380,27 @@ def handle_example_click(prompt_text):
             'title': 'Nuevo Chat'
         }
     
-    # Procesar la respuesta del modelo directamente
-    try:
-        response = st.session_state.chat.send_message(prompt_text, stream=True)
-        full_text = ""
-        for chunk in response:
-            full_text += chunk.text
-        
-        # Mostrar respuesta del asistente
-        mensaje_mostrado = mostrar_con_efecto_escritura(full_text, velocidad=0.05)
-        
-        # Actualizar historial
-        st.session_state.messages.append({
-            'role': MODEL_ROLE,
-            'content': mensaje_mostrado,
-            'avatar': AI_AVATAR_ICON,
-        })
-        st.session_state.gemini_history = st.session_state.chat.history
+    process_model_response(prompt_text)
+
+# Modificar la sección principal de manejo de chat
+if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
+    is_first_message = len(st.session_state.messages) == 0
+    
+    if st.session_state.current_chat_id not in st.session_state.chats_in_memory:
+        # ... existing code for title generation ...
         
-        # Actualizar chat en memoria
-        st.session_state.chats_in_memory[st.session_state.current_chat_id].update({
-            'messages': st.session_state.messages,
-            'gemini_history': st.session_state.gemini_history,
-            'title': st.session_state.chat_title
-        })
-    except Exception as e:
-        st.error(f"Error al procesar la solicitud: {str(e)}")
+    add_message('user', prompt, USER_AVATAR_ICON)
+    with st.chat_message('user', avatar=USER_AVATAR_ICON):
+        st.markdown(prompt)
+    
+    if is_first_message:
+        add_message(MODEL_ROLE, WELCOME_MESSAGE, AI_AVATAR_ICON)
+        with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
+            st.markdown(WELCOME_MESSAGE)
+        update_chat_memory()
+        st.rerun()
+    
+    process_model_response(prompt)
 
 # Mostrar ejemplos solo si show_examples es True y no hay mensajes previos
 if st.session_state.show_examples and not st.session_state.messages:
@@ -601,3 +510,4 @@ with st.sidebar:
             st.session_state.gemini_history = []
             st.session_state.chat_title = 'Nuevo Chat'
         st.rerun()
+    st.rerun()