Upload app.py

app.py

@@ -434,11 +434,55 @@ 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()
+    
+    # Mostrar mensaje del usuario
+    with st.chat_message('user', avatar=USER_AVATAR_ICON):
+        st.markdown(prompt_text)
+    
+    # Añadir mensaje del usuario al historial
+    st.session_state.messages.append({
+        'role': 'user',
+        'content': prompt_text,
+        'avatar': USER_AVATAR_ICON
+    })
+    
+    # Asegurarnos de que el chat existe 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,
+            'gemini_history': st.session_state.gemini_history,
+            '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
+        
+        # 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)}")
 
-# Mostrar ejemplos solo si show_examples es True
-if st.session_state.show_examples:
+# 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:
     st.title("💡 RoboCopy - Asistente de PUVs")
     st.markdown("### Tu experto en crear Propuestas Únicas de Valor que convierten")