Update app.py

app.py

@@ -43,10 +43,7 @@ css_style = """
 """
 
 def eliminar_proceso_pensamiento(texto):
-    # Eliminar todo el contenido entre las etiquetas de pensamiento
     texto_limpio = re.sub(r'', '', texto, flags=re.DOTALL)
-    
-    # Extraer la última línea no vacía del texto restante
     lineas = [line.strip() for line in texto_limpio.split('\n') if line.strip()]
     return lineas[-1] if lineas else "Respuesta no disponible"
 
@@ -90,9 +87,9 @@ def get_conversational_chain():
 
 def extract_metadata(vector_store):
     metadata_questions = {
-        "title": "¿Cual es o podría ser el título 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": "¿En qué fecha se implantará el contenido del documento? Si no se especifica ninguna, responde 'No se especifica'"
+        "date": "¿En qué fecha se implantará el contenido? Si no se detalla responde \"No se especifica\""
     }
     
     metadata = {}
@@ -110,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)
 
@@ -136,6 +132,10 @@ 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 +147,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)
@@ -156,7 +156,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()
                 
@@ -184,31 +184,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 = "¿Cual es o podría ser el título del documento?"
-        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 = "¿En qué fecha se implantará el contenido del documento? Si no se especifica ninguna, responde 'No se especifica'"
-
-    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()