Update src/streamlit_app.py

src/streamlit_app.py

@@ -86,6 +86,15 @@ def _liberar_memoria() -> None:
         torch.cuda.empty_cache()
 
 
+def _reset_estado_inicial() -> None:
+    _liberar_memoria()
+    try:
+        cargar_modelo_transformers.clear()
+    except Exception:
+        pass
+    st.session_state.clear()
+
+
 def _render_reloj_tiempo_real(inicio_epoch: int, placeholder) -> None:
     with placeholder:
         components.html(
@@ -214,6 +223,20 @@ _init_state()
 if st.session_state.get("eval_error"):
     st.error(st.session_state["eval_error"])
 
+if st.session_state.get("eval_running"):
+    st.warning("Evaluación en curso. Los controles están bloqueados hasta finalizar o cancelar.")
+    confirmar_cancelacion = st.checkbox(
+        "Confirmo que quiero cancelar la evaluación en curso",
+        key="confirmar_cancelacion_en_curso",
+        value=False,
+    )
+    if st.button("Cancelar evaluación", key="cancelar_eval_en_curso"):
+        if confirmar_cancelacion:
+            _reset_estado_inicial()
+            st.rerun()
+        else:
+            st.error("Debes confirmar la cancelación para detener la evaluación.")
+
 modo = st.radio(
     "Selecciona el modo",
     options=[ModoEvaluacion.POR_DEFECTO.value, ModoEvaluacion.PERSONALIZADA.value],
@@ -499,6 +522,9 @@ if st.session_state.get("eval_running") and st.session_state.get("eval_requested
 if st.session_state.get("eval_success") and st.session_state.get("last_result"):
     resultado = st.session_state["last_result"]
 
+    if resultado.get("elapsed_total"):
+        st.info(f"Tiempo total del proceso de evaluación: {resultado['elapsed_total']}")
+
     if resultado.get("resumen") is not None:
         st.success("Resumen de evaluación")
         st.json(resultado["resumen"])
@@ -533,11 +559,5 @@ if st.session_state.get("eval_success") and st.session_state.get("last_result"):
         st.rerun()
 
     if st.button("Nueva evaluación", disabled=st.session_state["eval_running"]):
-        st.session_state["eval_success"] = False
-        st.session_state["last_result"] = None
-        st.session_state["modelo_eval_validado"] = False
-        st.session_state["modelo_eval_confirmado"] = ""
-        st.session_state["modelo_gen_validado"] = False
-        st.session_state["modelo_gen_confirmado"] = ""
-        _liberar_memoria()
+        _reset_estado_inicial()
         st.rerun()