Upload app.py

app.py

@@ -133,14 +133,17 @@ def process_model_response(prompt, max_retries=3):
                 
                 response = st.session_state.chat.send_message(prompt, stream=True)
                 mensaje_completo = ""
+                mensaje_actual = ""
                 
-                # Procesar la respuesta por chunks
+                # Procesar la respuesta por chunks con efecto de escritura
                 for chunk in response:
                     mensaje_completo += chunk.text
-                    typing_indicator.markdown(mensaje_completo + "▌")
-                    time.sleep(0.01)
+                    for caracter in chunk.text:
+                        mensaje_actual += caracter
+                        typing_indicator.markdown(mensaje_actual + "▌")
+                        time.sleep(0.01)  # Velocidad de escritura ajustable
                 
-                # Mostrar mensaje final y limpiar indicador
+                # Mostrar mensaje final
                 typing_indicator.markdown(mensaje_completo)
             
             add_message(MODEL_ROLE, mensaje_completo, AI_AVATAR_ICON)
@@ -270,10 +273,20 @@ with st.sidebar:
     if st.button('🗑️ Borrar Historial de Chat Actual'):
         if st.session_state.current_chat_id in st.session_state.chats_in_memory:
             del st.session_state.chats_in_memory[st.session_state.current_chat_id]
-            st.session_state.messages = []
-            st.session_state.gemini_history = []
-            st.session_state.chat_title = 'Nuevo Chat'
-            st.rerun()
+        
+        # Reiniciar completamente el estado del chat actual
+        st.session_state.update({
+            'messages': [],
+            'gemini_history': [],
+            'chat_title': 'Nuevo Chat',
+            'show_examples': True
+        })
+        
+        # Reiniciar el chat con el modelo
+        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()
 
 # === CONFIGURACIÓN DEL MODELO ===
 model = genai.GenerativeModel(model_name='gemini-2.0-flash')
@@ -375,4 +388,17 @@ with chat_container:
         }
         
         process_model_response(new_prompt)
-        update_chat_memory()
+        update_chat_memory()
+        # En la sección del SIDEBAR
+        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()