Update app.py

app.py

@@ -64,10 +64,11 @@ 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"
-
+    """Extrae el proceso de pensamiento del modelo"""
+    pensamiento = re.findall(r'<think>(.*?)</think>', texto, flags=re.DOTALL)
+    texto_limpio = re.sub(r'<think>.*?</think>', '', texto, flags=re.DOTALL).strip()
+    return texto_limpio, pensamiento[0].strip() if pensamiento else "No disponible"
+    
 def get_pdf_text(pdf_docs):
     text = ""
     for pdf in pdf_docs:
@@ -131,7 +132,11 @@ def extract_metadata(vector_store):
 def mostrar_respuesta(texto):
     with st.container():
         st.markdown(f'<div class="response-box">{texto}</div>', unsafe_allow_html=True)
-
+ 
+    # Desplegable para mostrar el proceso de pensamiento del modelo
+    with st.expander("💭 Pensamiento del modelo"):
+        st.markdown(pensamiento)
+        
 def generar_sugerencias():
     """Genera preguntas sugeridas simples y generales"""
     if 'vector_store' not in st.session_state:
@@ -176,6 +181,7 @@ def generar_sugerencias():
         return
 
 def procesar_consulta(user_question):
+    """Procesa la consulta del usuario y muestra la respuesta"""
     if 'vector_store' not in st.session_state:
         st.error("Por favor carga un documento primero")
         return
@@ -189,8 +195,10 @@ def procesar_consulta(user_question):
             return_only_outputs=True
         )
         
-    respuesta_final = eliminar_proceso_pensamiento(response['output_text'])
-    mostrar_respuesta(respuesta_final)
+    respuesta_final, pensamiento = eliminar_proceso_pensamiento(response['output_text'])
+    
+    # Mostrar respuesta y pensamiento
+    mostrar_respuesta(respuesta_final, pensamiento)
 
 def main():
     st.set_page_config(page_title="PDF Consultor 🔍", page_icon="🔍", layout="wide")