Update app.py

app.py

@@ -64,11 +64,10 @@ css_style = """
 """
 
 def eliminar_proceso_pensamiento(texto):
-    """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"
-    
+    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 = ""
     for pdf in pdf_docs:
@@ -132,11 +131,7 @@ 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:
@@ -181,7 +176,6 @@ 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
@@ -195,10 +189,8 @@ def procesar_consulta(user_question):
             return_only_outputs=True
         )
         
-    respuesta_final, pensamiento = eliminar_proceso_pensamiento(response['output_text'])
-    
-    # Mostrar respuesta y pensamiento
-    mostrar_respuesta(respuesta_final, pensamiento)
+    respuesta_final = eliminar_proceso_pensamiento(response['output_text'])
+    mostrar_respuesta(respuesta_final)
 
 def main():
     st.set_page_config(page_title="PDF Consultor 🔍", page_icon="🔍", layout="wide")