Upload app.py

app.py

@@ -5,34 +5,43 @@ import streamlit as st
 import google.generativeai as genai
 from dotenv import load_dotenv
 
-# Configuración inicial de la página (DEBE IR AL INICIO)
+# Configuración global
+CONFIG = {
+    'model': {
+        'role': 'ai',
+        'avatar': '🤖'
+    },
+    'user': {
+        'avatar': '👤'
+    },
+    'page': {
+        'title': "RoboCopy - Creador de PUVs",
+        'icon': "🚀",
+        'layout': "wide"
+    }
+}
+
+# Usar CONFIG en lugar de constantes separadas
 st.set_page_config(
-    page_title="RoboCopy - Creador de PUVs",
-    page_icon="🚀",
-    layout="wide"
+    page_title=CONFIG['page']['title'],
+    page_icon=CONFIG['page']['icon'],
+    layout=CONFIG['page']['layout']
 )
 
-# Función para cargar CSS personalizado
 def load_css(file_path):
-    with open(file_path) as f:
-        st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
-
-# Intentar cargar el CSS personalizado con ruta absoluta para mayor seguridad
-try:
-    css_path = os.path.join(os.path.dirname(__file__), 'static', 'css', 'style.css')
-    load_css(css_path)
-except Exception as e:
-    print(f"Error al cargar CSS: {e}")
-    # Si el archivo no existe, crear un estilo básico en línea
-    st.markdown("""
-    <style>
-    .robocopy-title {
-        color: #4ECDC4 !important;
-        font-weight: bold;
-        font-size: 2em;
-    }
-    </style>
-    """, unsafe_allow_html=True)
+    try:
+        with open(file_path) as f:
+            st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
+    except Exception as e:
+        st.markdown("""
+        <style>
+        .robocopy-title {
+            color: #4ECDC4 !important;
+            font-weight: bold;
+            font-size: 2em;
+        }
+        </style>
+        """, unsafe_allow_html=True)
 
 load_dotenv()
 GOOGLE_API_KEY=os.environ.get('GOOGLE_API_KEY')
@@ -192,26 +201,23 @@ Creo PUVs persuasivas que:
 
 # Función para mostrar texto con efecto de escritura
 def mostrar_con_efecto_escritura(mensaje, velocidad=0.05):
-    with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
-        # Contenedor para el mensaje con ancho completo
-        message_container = st.container()
-        message_placeholder = message_container.empty()
-        full_response = ''
-        
-        # Indicador de escritura en el contenedor
-        typing_indicator = message_container.empty()
-        typing_indicator.markdown("*RoboCopy está escribiendo...*")
-        
-        # Mostrar respuesta por fragmentos con formato mejorado
-        for palabra in mensaje.split(' '):
-            full_response += palabra + ' '
-            time.sleep(velocidad)
-            # Usar markdown para mejor formato
-            message_placeholder.markdown(full_response + '▌')
-        
-        # Eliminar indicador y mostrar respuesta completa con formato
-        typing_indicator.empty()
-        message_placeholder.markdown(mensaje)
+    # Crear un contenedor principal que ocupe todo el ancho
+    main_container = st.container()
+    
+    with main_container:
+        with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
+            message_placeholder = st.empty()
+            typing_indicator = st.empty()
+            typing_indicator.markdown("*RoboCopy está escribiendo...*")
+            
+            full_response = ''
+            for palabra in mensaje.split(' '):
+                full_response += palabra + ' '
+                time.sleep(velocidad)
+                message_placeholder.markdown(full_response + '▌')
+            
+            typing_indicator.empty()
+            message_placeholder.markdown(mensaje)
     
     return mensaje
 
@@ -434,22 +440,25 @@ if 'show_examples' not in st.session_state:
 
 # 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())
-    
-    # Mostrar mensaje del usuario
-    with st.chat_message('user', avatar=USER_AVATAR_ICON):
-        st.markdown(prompt_text)
+    # Reiniciar estado
+    st.session_state.update({
+        'show_examples': False,
+        'messages': [],
+        'current_chat_id': str(time.time())
+    })
     
-    # Añadir mensaje del usuario al historial
-    st.session_state.messages.append({
+    # Añadir mensaje del usuario
+    message = {
         'role': 'user',
         'content': prompt_text,
         'avatar': USER_AVATAR_ICON
-    })
+    }
+    st.session_state.messages.append(message)
+    
+    with st.chat_message('user', avatar=USER_AVATAR_ICON):
+        st.markdown(prompt_text)
     
-    # Asegurarnos de que el chat existe en memoria
+    # Actualizar chat 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,
@@ -547,3 +556,48 @@ if hasattr(st.session_state, 'temp_prompt') and st.session_state.temp_prompt:
     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()