Upload app.py

app.py

@@ -254,9 +254,17 @@ def process_model_response(prompt, max_retries=3):
             time.sleep(wait_time)
     return False
 
-# Modificar la sección principal de manejo de chat (línea ~286)
+
 if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
-    is_first_message = len(st.session_state.messages) == 0  # Definir aquí
+    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': [],
+            'gemini_history': [],
+            'title': 'Nuevo Chat'
+        }
     
     # Generar título basado en el primer mensaje del usuario
     try:
@@ -271,48 +279,17 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
     # Actualizar título en memoria
     st.session_state.chats_in_memory[st.session_state.current_chat_id]['title'] = st.session_state.chat_title
     
-    add_message(MODEL_ROLE, WELCOME_MESSAGE, AI_AVATAR_ICON)
-    with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON, key="welcome_msg"):
+    # 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, key="welcome_msg"):
         st.markdown(WELCOME_MESSAGE)
     update_chat_memory()
-    st.experimental_rerun()
     
     process_model_response(prompt)  # Mantener dentro del bloque
 
-# Eliminar la referencia duplicada fuera del bloque if 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:
-    # 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("---")
-    
-    # Actualizar la función handle_example_click (línea ~250)
-    def handle_example_click(prompt_text):
+def handle_example_click(prompt_text):
         st.session_state.update({
             'show_examples': False,
             'messages': [],
@@ -339,3 +316,33 @@ if st.session_state.show_examples and not st.session_state.messages:
         st.session_state.chats_in_memory[st.session_state.current_chat_id]['title'] = st.session_state.chat_title
         process_model_response(prompt_text)
 
+        # 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:
+            # 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("---")
+