Upload app.py

app.py

@@ -129,7 +129,7 @@ def process_model_response(prompt, max_retries=3):
             with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
                 # Añade un indicador de "escribiendo..."
                 typing_indicator = st.empty()
-                typing_indicator.markdown("*RoboCopy está escribiendo...*")
+                typing_indicator.markdown("*🤖 RoboCopy está escribiendo...*")
                 
                 response = st.session_state.chat.send_message(prompt, stream=True)
                 mensaje_completo = ""
@@ -141,9 +141,9 @@ def process_model_response(prompt, max_retries=3):
                     for caracter in chunk.text:
                         mensaje_actual += caracter
                         typing_indicator.markdown(mensaje_actual + "▌")
-                        time.sleep(0.01)  # Velocidad de escritura ajustable
+                        time.sleep(0.02)  # Velocidad de escritura ajustada
                 
-                # Mostrar mensaje final
+                # Mostrar mensaje final y limpiar indicador
                 typing_indicator.markdown(mensaje_completo)
             
             add_message(MODEL_ROLE, mensaje_completo, AI_AVATAR_ICON)
@@ -259,6 +259,20 @@ with st.sidebar:
     
     st.write('# Chats Anteriores')
     
+    # Añadir botón de nuevo chat
+    if st.button('📝 Nuevo Chat'):
+        st.session_state.update({
+            'current_chat_id': str(time.time()),
+            'messages': [],
+            'gemini_history': [],
+            'chat_title': 'Nuevo Chat',
+            'show_examples': True
+        })
+        st.session_state.chat = st.session_state.model.start_chat(history=[])
+        st.session_state.chat.send_message(SYSTEM_PROMPT)
+        st.session_state.gemini_history = st.session_state.chat.history
+        st.rerun()
+    
     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
     
@@ -283,7 +297,7 @@ with st.sidebar:
         })
         
         # Reiniciar el chat con el modelo
-        st.session_state.chat = st.session_state.model.start_chat(history=[])
+        st.session_state.chat = st.session_state.model.start_chat(history=st.session_state.gemini_history)
         st.session_state.chat.send_message(SYSTEM_PROMPT)
         st.session_state.gemini_history = st.session_state.chat.history
         st.rerun()
@@ -367,6 +381,29 @@ with chat_container:
         process_model_response(prompt)
         update_chat_memory()
     
+    # Procesar nueva entrada si existe
+    if new_prompt:
+        with st.chat_message("user", avatar=USER_AVATAR_ICON):
+            st.markdown(new_prompt)
+        add_message("user", new_prompt, USER_AVATAR_ICON)
+        
+        try:
+            title_response = st.session_state.model.generate_content(
+                f"Título para consulta: '{new_prompt}' (máximo 4 palabras)"
+            )
+            st.session_state.chat_title = title_response.text.strip()[:25]
+        except Exception as e:
+            st.session_state.chat_title = f"Chat-{time.strftime('%H:%M')}"
+        
+        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': st.session_state.chat_title
+        }
+        
+        process_model_response(new_prompt)
+        update_chat_memory()
+    
     # Procesar nueva entrada si existe
     if new_prompt:
         with st.chat_message("user", avatar=USER_AVATAR_ICON):
@@ -401,4 +438,4 @@ with chat_container:
             st.session_state.chat = st.session_state.model.start_chat(history=[])
             st.session_state.chat.send_message(SYSTEM_PROMPT)
             st.session_state.gemini_history = st.session_state.chat.history
-            st.rerun()
+            st.rerun()