Upload app.py

app.py

@@ -126,20 +126,16 @@ WELCOME_MESSAGE = """
 """
 
 # Inicializar el estado de la sesión
+# Inicialización unificada
 if 'chats_in_memory' not in st.session_state:
-    st.session_state.chats_in_memory = {}
-
-if 'current_chat_id' not in st.session_state:
-    st.session_state.current_chat_id = new_chat_id
-
-if 'chat_title' not in st.session_state:
-    st.session_state.chat_title = 'Nuevo Chat'
-
-if 'messages' not in st.session_state:
-    st.session_state.messages = []
-
-if 'show_examples' not in st.session_state:
-    st.session_state.show_examples = True
+    st.session_state.update({
+        'chats_in_memory': {},
+        'current_chat_id': new_chat_id,
+        'chat_title': 'Nuevo Chat',
+        'messages': [],
+        'show_examples': True,
+        'gemini_history': []
+    })
 
 # Sidebar allows a list of past chats
 with st.sidebar:
@@ -154,12 +150,13 @@ with st.sidebar:
     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') if x != new_chat_id else 'Nuevo Chat',
-        key='chat_selector'
+    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
@@ -169,22 +166,7 @@ with st.sidebar:
             st.session_state.messages = []
             st.session_state.gemini_history = []
             st.session_state.chat_title = 'Nuevo Chat'
-            st.rerun()
-
-# Inicializar mensajes y cargar historial
-if not st.session_state.get('messages'):
-    st.session_state.messages = []
-
-# Cargar historial de chat si existe en memoria
-if st.session_state.current_chat_id in st.session_state.chats_in_memory:
-    chat_data = st.session_state.chats_in_memory[st.session_state.current_chat_id]
-    st.session_state.messages = chat_data.get('messages', [])
-    st.session_state.gemini_history = chat_data.get('gemini_history', [])
-    st.session_state.chat_title = chat_data.get('title', 'Nuevo Chat')
-else:
-    st.session_state.messages = []
-    st.session_state.gemini_history = []
-    st.session_state.chat_title = 'Nuevo Chat'
+            st.experimental_rerun()  # Evitar loops
 
 # Configuración del modelo
 model = genai.GenerativeModel(model_name='gemini-2.0-flash')
@@ -256,24 +238,15 @@ def process_model_response(prompt, max_retries=3):
     return False
 
 # Modificar la función handle_example_click para usar las nuevas funciones
+# Simplificar a:
 def handle_example_click(prompt_text):
     st.session_state.update({
         'show_examples': False,
         'messages': [],
-        'current_chat_id': str(time.time())
+        'current_chat_id': str(time.time()),
+        'gemini_history': [],
+        'chat_title': get_chat_title([])
     })
-    
-    add_message('user', prompt_text, USER_AVATAR_ICON)
-    with st.chat_message('user', avatar=USER_AVATAR_ICON):
-        st.markdown(prompt_text)
-    
-    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'
-        }
-    
     process_model_response(prompt_text)
 
 # Modificar la sección principal de manejo de chat
@@ -293,10 +266,10 @@ if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
     
     if is_first_message:
         add_message(MODEL_ROLE, WELCOME_MESSAGE, AI_AVATAR_ICON)
-        with st.chat_message(name=MODEL_ROLE, avatar=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.rerun()
+        st.experimental_rerun()
     
     process_model_response(prompt)
 
@@ -330,82 +303,3 @@ if st.session_state.show_examples and not st.session_state.messages:
     
     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
-    })
-    
-    # Asegurarnos de que el chat existe en memoria antes de actualizarlo
-    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'
-        }
-    
-    # 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()
-    st.rerun()
-
-# Modificar la sección de historial de chat
-with st.sidebar:
-    # Centrar el logo y eliminar el título de RoboCopy
-    col1, col2, col3 = st.columns([1, 2, 1])
-    with col2:
-        st.image("assets/robocopy_logo.png", width=300)
-    
-    st.write('# Chats Anteriores')
-    if st.session_state.get('current_chat_id') is None:
-        st.session_state.current_chat_id = st.selectbox(
-            label='Selecciona un chat anterior',
-            options=[new_chat_id] + list(st.session_state.chats_in_memory.keys()),
-            format_func=lambda x: st.session_state.chats_in_memory.get(x, {}).get('title', 'Nuevo Chat') if isinstance(x, str) else 'Nuevo Chat',
-            placeholder='_',
-        )
-    else:
-        st.session_state.current_chat_id = st.selectbox(
-            label='Selecciona un chat anterior',
-            options=[new_chat_id, st.session_state.current_chat_id] + list(st.session_state.chats_in_memory.keys()),
-            index=1,
-            format_func=lambda x: st.session_state.chats_in_memory.get(x, {}).get('title', 'Nuevo Chat') if isinstance(x, str) else 'Nuevo Chat',
-            placeholder='_',
-        )
-    
-    # Modificar el selectbox para el historial
-    chat_options = [new_chat_id] + list(st.session_state.chats_in_memory.keys())
-    selected_chat = st.selectbox(
-        label='Selecciona un chat anterior',
-        options=chat_options,
-        format_func=lambda x: st.session_state.chats_in_memory.get(x, {}).get('title', 'Nuevo Chat'),
-        index=chat_options.index(st.session_state.current_chat_id) if st.session_state.current_chat_id in chat_options else 0
-    )
-    
-    # Actualizar el chat actual si cambia la selección
-    if selected_chat != st.session_state.current_chat_id:
-        st.session_state.current_chat_id = selected_chat
-        if selected_chat in st.session_state.chats_in_memory:
-            st.session_state.messages = st.session_state.chats_in_memory[selected_chat].get('messages', [])
-            st.session_state.gemini_history = st.session_state.chats_in_memory[selected_chat].get('gemini_history', [])
-            st.session_state.chat_title = st.session_state.chats_in_memory[selected_chat].get('title', 'Nuevo Chat')
-        else:
-            st.session_state.messages = []
-            st.session_state.gemini_history = []
-            st.session_state.chat_title = 'Nuevo Chat'
-        st.rerun()
-    st.rerun()