Update app.py

app.py

@@ -43,8 +43,9 @@ css_style = """
 """
 
 def eliminar_proceso_pensamiento(texto):
-    """Elimina todo contenido entre  incluyendo las etiquetas"""
-    return re.sub(r'', '', texto, flags=re.DOTALL).strip()
+    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"
 
 def get_pdf_text(pdf_docs):
     text = ""
@@ -86,9 +87,9 @@ def get_conversational_chain():
 
 def extract_metadata(vector_store):
     metadata_questions = {
-        "title": "¿Cuál es el título principal del documento?",
+        "title": "¿Cual es o podría ser el título del documento? Redacta una sola frase",
         "entity": "¿A qué entidad u organización pertenece este documento?",
-        "date": "¿Qué fecha de implantación se menciona en el documento?"
+        "date": "¿En qué fecha se implantará el contenido? Si no se detalla responde \"No se especifica\""
     }
     
     metadata = {}
@@ -106,7 +107,6 @@ 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)
 
@@ -125,13 +125,34 @@ def procesar_consulta(user_question):
         )
         
     respuesta_final = eliminar_proceso_pensamiento(response['output_text'])
-    mostrar_respuesta(respuesta_final)
+    
+    # Almacenar la pregunta y respuesta en la historia de la conversación
+    if 'conversacion' not in st.session_state:
+        st.session_state.conversacion = []
+    
+    st.session_state.conversacion.append({
+        "pregunta": user_question,
+        "respuesta": respuesta_final
+    })
+    
+    # Mostrar toda la conversación
+    mostrar_conversacion()
+
+def mostrar_conversacion():
+    if 'conversacion' in st.session_state:
+        for pregunta, respuesta in st.session_state.conversacion:
+            st.write(f"**Pregunta:** {pregunta}")
+            mostrar_respuesta(respuesta)
 
 def main():
     st.set_page_config(page_title="PDF Consultor 🔍", page_icon="🔍", layout="wide")
     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)
@@ -143,7 +164,7 @@ def main():
         )
 
     # Procesamiento automático al cargar documentos
-    if pdf_docs and not st.session_state.get('processed'):
+    if pdf_docs and not st.session_state.documento_cargado:
         with st.spinner("Analizando documento..."):
             try:
                 raw_text = get_pdf_text(pdf_docs)
@@ -152,7 +173,7 @@ def main():
                 
                 st.session_state.metadata = extract_metadata(vector_store)
                 st.session_state.vector_store = vector_store
-                st.session_state.processed = True
+                st.session_state.documento_cargado = True
                 
                 st.rerun()
                 
@@ -180,31 +201,23 @@ def main():
         st.markdown("---")
 
     # Interfaz de consultas
-    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 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.")
 
 if __name__ == "__main__":
     main()