Update app.py

app.py

@@ -43,9 +43,8 @@ css_style = """
 """
 
 def eliminar_proceso_pensamiento(texto):
-    texto_limpio = re.sub(r'', '', texto, flags=re.DOTALL)
-    lineas = [line.strip() for line in texto_limpio.split('\n') if line.strip()]
-    return lineas[-1] if lineas else "Respuesta no disponible"
+    """Elimina todo contenido entre  incluyendo las etiquetas"""
+    return re.sub(r'', '', texto, flags=re.DOTALL).strip()
 
 def get_pdf_text(pdf_docs):
     text = ""
@@ -87,9 +86,9 @@ def get_conversational_chain():
 
 def extract_metadata(vector_store):
     metadata_questions = {
-        "title": "¿Cual es o podría ser el título del documento? Redacta una sola frase",
+        "title": "¿Cuál es el título principal del documento?",
         "entity": "¿A qué entidad u organización pertenece este documento?",
-        "date": "¿En qué fecha se implantará el contenido? Si no se detalla responde \"No se especifica\""
+        "date": "¿Qué fecha de implantación se menciona en el documento?"
     }
     
     metadata = {}
@@ -107,6 +106,7 @@ def extract_metadata(vector_store):
     return metadata
 
 def mostrar_respuesta(texto):
+    """Muestra la respuesta formateada en un contenedor especial"""
     with st.container():
         st.markdown(f'<div class="response-box">{texto}</div>', unsafe_allow_html=True)
 
@@ -132,10 +132,6 @@ def main():
     st.title("PDF Consultor 🔍")
     st.markdown(css_style, unsafe_allow_html=True)
 
-    # Inicializa estado de sesión
-    if 'documento_cargado' not in st.session_state:
-        st.session_state.documento_cargado = False
-
     # Sidebar - Carga de documentos
     with st.sidebar:
         st.markdown('<p class="step-number">1 Subir archivos</p>', unsafe_allow_html=True)
@@ -147,7 +143,7 @@ def main():
         )
 
     # Procesamiento automático al cargar documentos
-    if pdf_docs and not st.session_state.documento_cargado:
+    if pdf_docs and not st.session_state.get('processed'):
         with st.spinner("Analizando documento..."):
             try:
                 raw_text = get_pdf_text(pdf_docs)
@@ -156,7 +152,7 @@ def main():
                 
                 st.session_state.metadata = extract_metadata(vector_store)
                 st.session_state.vector_store = vector_store
-                st.session_state.documento_cargado = True
+                st.session_state.processed = True
                 
                 st.rerun()
                 
@@ -184,23 +180,31 @@ def main():
         st.markdown("---")
 
     # Interfaz de consultas
-    if st.session_state.documento_cargado:
-        with st.form("consulta_form"):
-            col1, col2 = st.columns([5, 1])
-            with col1:
-                user_question = st.text_input(
-                    "Escribe tu pregunta:",
-                    placeholder="Ej: ¿Qué normativa regula este proceso?",
-                    label_visibility="collapsed"
-                )
-            with col2:
-                st.markdown("<br>", unsafe_allow_html=True)
-                enviar = st.form_submit_button("Enviar ▶")
-
-        if user_question and enviar:
-            procesar_consulta(user_question)
-    else:
-        st.write("Por favor, sube un documento para continuar.")
+    with st.form("consulta_form"):
+        col1, col2 = st.columns([5, 1])
+        with col1:
+            user_question = st.text_input(
+                "Escribe tu pregunta:",
+                placeholder="Ej: ¿Qué normativa regula este proceso?",
+                label_visibility="collapsed"
+            )
+        with col2:
+            st.markdown("<br>", unsafe_allow_html=True)
+            enviar = st.form_submit_button("Enviar ▶")
+
+        botones_rapidos = st.columns(3)
+        with botones_rapidos[0]:
+            if st.form_submit_button("📝 Resumen ejecutivo"):
+                user_question = "Genera un resumen ejecutivo de máximo 3 párrafos"
+        with botones_rapidos[1]:
+            if st.form_submit_button("🏛️ Entidad relacionada"):
+                user_question = "¿A qué organización o entidad pertenece este documento?"
+        with botones_rapidos[2]:
+            if st.form_submit_button("📅 Fechas clave"):
+                user_question = "Lista las fechas importantes mencionadas en orden cronológico"
+
+    if user_question and enviar:
+        procesar_consulta(user_question)
 
 if __name__ == "__main__":
     main()