Upload app.py

app.py

@@ -425,42 +425,71 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
             break
 
 # Título y subtítulo principales
-st.title("💡 RoboCopy - Asistente de PUVs")
-st.markdown("### Tu experto en crear Propuestas Únicas de Valor que convierten")
+# Inicializar el estado para los ejemplos
+if 'show_examples' not in st.session_state:
+    st.session_state.show_examples = True
 
-# Ejemplos de prompts
-st.markdown("### 🎯 Prueba estos ejemplos:")
-
-# Contenedor para los ejemplos
-example_container = st.container()
-col1, col2 = example_container.columns(2)
-
-with col1:
-    if st.button("¿Cuál es la mejor fórmula para mi PUV? 🤔"):
-        prompt = "¿Podrías explicarme cuál es la mejor fórmula para crear una PUV efectiva para mi negocio?"
-        st.session_state.messages = []
-        st.session_state.current_chat_id = str(time.time())
-        st.rerun()
-
-    if st.button("Necesito una PUV para mi tienda online 🛍️"):
-        prompt = "Quiero crear una PUV para mi tienda online de ropa sostenible dirigida a mujeres de 25-35 años"
-        st.session_state.messages = []
-        st.session_state.current_chat_id = str(time.time())
-        st.rerun()
-
-with col2:
-    if st.button("Ayúdame a mejorar mi PUV actual ✨"):
-        prompt = "¿Podrías ayudarme a mejorar mi PUV actual para hacerla más persuasiva?"
-        st.session_state.messages = []
-        st.session_state.current_chat_id = str(time.time())
-        st.rerun()
+# Función para manejar los clics de los botones
+def handle_example_click(prompt_text):
+    st.session_state.show_examples = False
+    st.session_state.messages = []
+    st.session_state.current_chat_id = str(time.time())
+    st.session_state.temp_prompt = prompt_text
+    st.rerun()
 
-    if st.button("Crear PUV para servicios profesionales 👔"):
-        prompt = "Necesito una PUV para mi servicio de consultoría en marketing digital"
-        st.session_state.messages = []
-        st.session_state.current_chat_id = str(time.time())
-        st.rerun()
+# Mostrar ejemplos solo si show_examples es True
+if st.session_state.show_examples:
+    st.title("💡 RoboCopy - Asistente de PUVs")
+    st.markdown("### Tu experto en crear Propuestas Únicas de Valor que convierten")
+    
+    st.markdown("### 🎯 Prueba estos ejemplos:")
+    
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        if st.button("¿Cuál es la mejor fórmula para mi PUV? 🤔"):
+            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 🛍️"):
+            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 ✨"):
+            handle_example_click("¿Podrías ayudarme a mejorar mi PUV actual para hacerla más persuasiva?")
+        
+        if st.button("Crear PUV para servicios profesionales 👔"):
+            handle_example_click("Necesito una PUV para mi servicio de consultoría en marketing digital")
+    
+    st.markdown("---")
 
-# Separador visual
-st.markdown("---")
+# Procesar el prompt temporal si existe
+if hasattr(st.session_state, 'temp_prompt') and st.session_state.temp_prompt:
+    prompt = st.session_state.temp_prompt
+    delattr(st.session_state, 'temp_prompt')
+    
+    # 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
+    })
+    
+    # Continuar con el procesamiento normal del prompt
+    st.session_state.messages.append({
+        'role': 'user',
+        'content': prompt,
+        'avatar': USER_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()