Upload app.py

app.py

@@ -22,13 +22,19 @@ CONFIG = {
     }
 }
 
-# Usar CONFIG en lugar de constantes separadas
+# Configuración de página
 st.set_page_config(
     page_title=CONFIG['page']['title'],
     page_icon=CONFIG['page']['icon'],
     layout=CONFIG['page']['layout']
 )
 
+# Configuración de avatares e identificadores
+MODEL_ROLE = 'ai'
+AI_AVATAR_ICON = '🤖'
+USER_AVATAR_ICON = '👤'
+
+# === FUNCIONES AUXILIARES ===
 def load_css(file_path):
     try:
         with open(file_path) as f:
@@ -44,43 +50,27 @@ def load_css(file_path):
         </style>
         """, unsafe_allow_html=True)
 
-load_dotenv()
-GOOGLE_API_KEY=os.environ.get('GOOGLE_API_KEY')
-genai.configure(api_key=GOOGLE_API_KEY)
-
-new_chat_id = f'{time.time()}'
-# Configuración de avatares e identificadores
-MODEL_ROLE = 'ai'
-AI_AVATAR_ICON = '🤖'
-USER_AVATAR_ICON = '👤'  # Aseguramos que el emoji sea visible
-AI_AVATAR_ICON = '🤖'
-
-# Función para manejar títulos de chat de manera unificada
 def get_chat_title(messages):
     if not messages:
         return "Nuevo Chat"
     first_msg = messages[0]['content'] if messages else ""
-    # Limitar el título a los primeros 30 caracteres
     title = first_msg[:30] + "..." if len(first_msg) > 30 else first_msg
     return title
 
 def get_formulas_for_prompt():
-    """Genera texto formateado con las fórmulas PUV para incluir en el prompt del sistema"""
     prompt_text = "\nFÓRMULAS DE PROPUESTAS ÚNICAS DE VALOR (PUVs):\n\n"
     
     for key, formula in puv_formulas.items():
         prompt_text += f"🔹 {key}:\n"
         prompt_text += f"   - Descripción: {formula.get('description', 'Descripción no disponible').strip()}\n"
         
-        # Extraer estructura de la descripción
         if 'Structure:' in formula.get('description', ''):
             estructura = formula['description'].split('Structure:')[1].split('Key elements:')[0].strip()
             prompt_text += "   - Estructura Base:\n"
             for line in estructura.split('\n'):
-                if line.strip():  # Ignorar líneas vacías
+                if line.strip():
                     prompt_text += f"     * {line.strip()}\n"
         
-        # Manejar ejemplos
         ejemplos = formula.get('examples', [])[:2]
         if ejemplos:
             prompt_text += "   - Ejemplos destacados:\n"
@@ -93,7 +83,98 @@ def get_formulas_for_prompt():
     
     return prompt_text
 
-# Definición COMPLETA del prompt multipersona para el sistema
+def add_message(role, content, avatar):
+    message = {
+        'role': role,
+        'content': content,
+        'avatar': avatar
+    }
+    st.session_state.messages.append(message)
+    return message
+
+def update_chat_memory():
+    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
+    })
+
+def handle_model_error(error, retry_count, max_retries):
+    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.error(error_message)
+        add_message(MODEL_ROLE, error_message, AI_AVATAR_ICON)
+        update_chat_memory()
+        return True
+    return False
+
+def mostrar_con_efecto_escritura(texto, velocidad=0.05):
+    placeholder = st.empty()
+    contenido_actual = ""
+    
+    for caracter in texto:
+        contenido_actual += caracter
+        placeholder.markdown(contenido_actual + "▌")
+        time.sleep(velocidad)
+    
+    placeholder.markdown(contenido_actual)
+    return contenido_actual
+
+def process_model_response(prompt, max_retries=3):
+    retry_count = 0
+    while retry_count < max_retries:
+        try:
+            response = st.session_state.chat.send_message(prompt, stream=True)
+            full_text = "".join(chunk.text for chunk in response)
+            mensaje_mostrado = mostrar_con_efecto_escritura(full_text, velocidad=0.01)
+            
+            add_message(MODEL_ROLE, mensaje_mostrado, AI_AVATAR_ICON)
+            st.session_state.gemini_history = st.session_state.chat.history
+            update_chat_memory()
+            return True
+            
+        except Exception as e:
+            retry_count += 1
+            if handle_model_error(e, retry_count, max_retries):
+                return False
+            wait_time = (1.5 ** retry_count)
+            time.sleep(wait_time)
+    return False
+
+def handle_example_click(prompt_text):
+    st.session_state.update({
+        'show_examples': False,
+        'messages': [],
+        'current_chat_id': str(time.time()),
+        'gemini_history': [],
+        'chat_title': 'Nuevo Chat'
+    })
+    
+    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': [],
+            'gemini_history': [],
+            'title': 'Nuevo Chat'
+        }
+    
+    try:
+        title_response = st.session_state.model.generate_content(
+            f"Título para consulta de ejemplo: '{prompt_text}' (máximo 4 palabras)"
+        )
+        st.session_state.chat_title = title_response.text.strip()[:25]
+    except Exception as e:
+        st.session_state.chat_title = f"Ejemplo-{time.strftime('%H:%M')}"
+    
+    st.session_state.chats_in_memory[st.session_state.current_chat_id]['title'] = st.session_state.chat_title
+    process_model_response(prompt_text)
+
+# === CONFIGURACIÓN INICIAL ===
+load_dotenv()
+GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY')
+genai.configure(api_key=GOOGLE_API_KEY)
+
+# Definición del prompt del sistema
 SYSTEM_PROMPT = f"""
 Eres un equipo colaborativo de expertos de clase mundial trabajando juntos para crear Propuestas Únicas de Valor (PUVs) excepcionales que conviertan a la audiencia en clientes.
 
@@ -126,8 +207,9 @@ WELCOME_MESSAGE = """
 ¿En qué puedo ayudarte hoy?
 """
 
-# Inicializar el estado de la sesión
-# Inicialización unificada
+# === INICIALIZACIÓN DEL ESTADO ===
+new_chat_id = str(time.time())
+
 if 'chats_in_memory' not in st.session_state:
     st.session_state.update({
         'chats_in_memory': {},
@@ -138,54 +220,48 @@ if 'chats_in_memory' not in st.session_state:
         'gemini_history': []
     })
 
-# Sidebar allows a list of past chats
+# === SIDEBAR ===
 with st.sidebar:
-    # Centrar el logo
     col1, col2, col3 = st.columns([1, 2, 1])
     with col2:
         st.image("assets/robocopy_logo.png", width=300)
     
     st.write('# Chats Anteriores')
     
-    # Simplificar la lógica del selector de chat
     chat_options = [new_chat_id] + list(st.session_state.chats_in_memory.keys())
     current_index = chat_options.index(st.session_state.current_chat_id) if st.session_state.current_chat_id in chat_options else 0
     
-    # Modificar a:
     st.session_state.current_chat_id = st.selectbox(
-    label='Selecciona un chat anterior',
-    options=chat_options,
-    index=current_index,
-    format_func=lambda x: st.session_state.chats_in_memory.get(x, {}).get('title', 'Nuevo Chat'),
-    key='chat_selector_unique'
+        label='Selecciona un chat anterior',
+        options=chat_options,
+        index=current_index,
+        format_func=lambda x: st.session_state.chats_in_memory.get(x, {}).get('title', 'Nuevo Chat'),
+        key='chat_selector_unique'
     )
     
-    # Botón para borrar el historial
     if st.button('🗑️ Borrar Historial de Chat Actual'):
         if st.session_state.current_chat_id in st.session_state.chats_in_memory:
             del st.session_state.chats_in_memory[st.session_state.current_chat_id]
             st.session_state.messages = []
             st.session_state.gemini_history = []
             st.session_state.chat_title = 'Nuevo Chat'
-            st.rerun()  # Versión actualizada del comando
+            st.rerun()
 
-# Configuración del modelo
+# === CONFIGURACIÓN DEL MODELO ===
 model = genai.GenerativeModel(model_name='gemini-2.0-flash')
 st.session_state.model = model
 st.session_state.chat = st.session_state.model.start_chat(history=st.session_state.gemini_history)
 
-# Si es un chat nuevo, enviar el prompt del sistema como primer mensaje
 if not st.session_state.gemini_history:
-    # Enviamos el prompt del sistema como primer mensaje (invisible para el usuario)
     st.session_state.chat.send_message(SYSTEM_PROMPT)
     st.session_state.gemini_history = st.session_state.chat.history
 
-# Mostrar mensajes del historial
+# === MOSTRAR MENSAJES DEL HISTORIAL ===
 for message in st.session_state.messages:
     with st.chat_message(name=message['role'], avatar=message.get('avatar', USER_AVATAR_ICON if message['role'] == 'user' else AI_AVATAR_ICON)):
         st.markdown(message['content'])
 
-# ===== SECCIÓN DE EJEMPLOS =====
+# === SECCIÓN DE EJEMPLOS ===
 if st.session_state.show_examples and not st.session_state.messages:
     main_container = st.container()
     with main_container:
@@ -211,16 +287,14 @@ if st.session_state.show_examples and not st.session_state.messages:
     
     st.markdown("---")
 
-# Entrada del usuario
+# === ENTRADA DEL USUARIO ===
 if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
-    # Añadir el mensaje del usuario primero
     with st.chat_message("user", avatar=USER_AVATAR_ICON):
         st.markdown(prompt)
     add_message("user", prompt, USER_AVATAR_ICON)
     
     is_first_message = len(st.session_state.messages) == 0
     
-    # Añadir inicialización de chat si no existe
     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': [],
@@ -228,7 +302,6 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
             'title': 'Nuevo Chat'
         }
     
-    # Generar título basado en el primer mensaje del usuario
     try:
         title_response = st.session_state.model.generate_content(
             f"Genera un título corto (máximo 5 palabras) que describa esta consulta, sin comillas: '{prompt}'"
@@ -238,101 +311,12 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
     except Exception as e:
         st.session_state.chat_title = f"Chat-{st.session_state.current_chat_id}"
     
-    # Actualizar título en memoria
     st.session_state.chats_in_memory[st.session_state.current_chat_id]['title'] = st.session_state.chat_title
     
-    # El mensaje de bienvenida se añade SIEMPRE que hay input
-    # Debe condicionarse a is_first_message
     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()
     
-    process_model_response(prompt)  # Mantener dentro del bloque
-    
-    # En la sección de ejemplos principal agregar:
-    if 'show_examples' not in st.session_state:
-        st.session_state.show_examples = True  # Forzar muestra inicial
-    
-    if st.session_state.show_examples:
-        # Añadir mensajes de ejemplo con avatares visibles
-        add_message('ai', WELCOME_MESSAGE, AI_AVATAR_ICON)
-        add_message('ai', "¿Quieres que te muestre algunos ejemplos de PUVs?", AI_AVATAR_ICON)
-        
-        # Usar contenedor para ancho completo
-        main_container = st.container()
-        
-        with main_container:
-            st.title("💡 RoboCopy - Asistente de PUVs")
-            st.markdown("### Tu experto en crear Propuestas Únicas de Valor que convierten")
-            
-            st.markdown("### 🎯 Prueba estos ejemplos:")
-            
-            # Usar columnas con proporción ajustada para mejor distribución
-            col1, col2 = st.columns([1, 1])
-            
-            with col1:
-                if st.button("¿Cuál es la mejor fórmula para mi PUV? 🤔", use_container_width=True):
-                    handle_example_click("¿Podrías explicarme cuál es la mejor fórmula para crear una PUV efectiva para mi negocio?")
-                
-                if st.button("Necesito una PUV para mi tienda online 🛍️", use_container_width=True):
-                    handle_example_click("Quiero crear una PUV para mi tienda online de ropa sostenible dirigida a mujeres de 25-35 años")
-            
-            with col2:
-                if st.button("Ayúdame a mejorar mi PUV actual ✨", use_container_width=True):
-                    handle_example_click("¿Podrías ayudarme a mejorar mi PUV actual para hacerla más persuasiva?")
-                
-                if st.button("Crear PUV para servicios profesionales 👔", use_container_width=True):
-                    handle_example_click("Necesito una PUV para mi servicio de consultoría en marketing digital")
-        
-        st.markdown("---")
-
-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
-    })
-
-def handle_example_click(prompt_text):
-    """Función para manejar clicks en ejemplos"""
-    st.session_state.update({
-        'show_examples': False,
-        'messages': [],
-        'current_chat_id': str(time.time()),
-        'gemini_history': [],
-        'chat_title': 'Nuevo Chat'
-    })
-    
-    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': [],
-            'gemini_history': [],
-            'title': 'Nuevo Chat'
-        }
-    
-    try:
-        title_response = st.session_state.model.generate_content(
-            f"Título para consulta de ejemplo: '{prompt_text}' (máximo 4 palabras)"
-        )
-        st.session_state.chat_title = title_response.text.strip()[:25]
-    except Exception as e:
-        st.session_state.chat_title = f"Ejemplo-{time.strftime('%H:%M')}"
-    
-    st.session_state.chats_in_memory[st.session_state.current_chat_id]['title'] = st.session_state.chat_title
-    process_model_response(prompt_text)
-
-st.markdown("---")
-
+    process_model_response(prompt)