Fixing the backend

Author: Asaif Ali

agent_service/app/presentation/routes/agent_router.py

@@ -21,6 +21,7 @@
 from app.infrastructure.utils.auth_client import get_current_user_http as get_current_user
 from app.infrastructure.repositories.migration_data_repository import fetch_migration_data, delete_migration_data
 from app.infrastructure.repositories.task_repository import create_task, update_task, get_task
+from app.infrastructure.utils.task_progress import bind_task
 from app.infrastructure.utils.migration_context import (
     migration_id_ctx,
     migration_name_ctx,
@@ -179,6 +180,9 @@ async def upload_team_files(
 async def execute_agent_team(task_id: str, request: RunTeamRequest, owner_user_id: str, llm_gateway_token: str = ""):
     logger.info("[TASK %s] STARTED | source_path=%r owner=%r", task_id, request.source_path, owner_user_id)
     create_task(task_id, user_id=owner_user_id, status=AgentConstants.TASK_STATUS_RUNNING)
+    # Connect workflow progress events to this persisted task so the UI
+    # receives live updates through GET /v1/tasks/{task_id}.
+    bind_task(task_id)
     try:
         # Background task: bind the authenticated user (ignore client-supplied user_id on the body).
         try:

streamlit_ui/app.py

@@ -667,15 +667,40 @@ def count_up(label: str, target: int, duration: float = 0.5, steps: int = 12):
                         terminal_progress["message"] = str(output_message)
                     st.session_state.last_progress = terminal_progress
                     render_live_plan(terminal_progress)
-                    if failed_payload.get("kind") == "blocked" or "release gate" in str(output_message).lower():
-                        if st.session_state.active_migration_name:
-                            try:
-                                data = client.download_migration(st.session_state.active_migration_name)
-                                st.download_button("Download gated artifact", data=data, file_name=f"{st.session_state.active_migration_name}.zip", mime="application/zip", width="stretch", key="blocked_demo_download")
-                            except ApiError:
-                                st.caption("Download will be available shortly.")
-                    else:
-                        st.error(f"Migration failed: {output_message}")
+                    st.error(f"Migration failed: {output_message}")
+
+                    # Keep the download affordance visible for every terminal failure.
+                    # A real migrated ZIP is downloadable when packaging succeeded; otherwise
+                    # provide a small failure report so the user always has a retrievable artifact.
+                    if st.session_state.active_migration_name:
+                        try:
+                            data = client.download_migration(st.session_state.active_migration_name)
+                            st.download_button(
+                                "Download available artifact",
+                                data=data,
+                                file_name=f"{st.session_state.active_migration_name}.zip",
+                                mime="application/zip",
+                                width="stretch",
+                                key="failed_run_artifact_download",
+                            )
+                        except ApiError:
+                            failure_report = {
+                                "migration": st.session_state.active_migration_name,
+                                "task_id": st.session_state.active_task_id,
+                                "status": state,
+                                "message": str(output_message),
+                                "result": failed_payload or status.get("result"),
+                                "last_progress": st.session_state.get("last_progress"),
+                            }
+                            st.download_button(
+                                "Download failure report",
+                                data=json.dumps(failure_report, indent=2, default=str).encode("utf-8"),
+                                file_name=f"{st.session_state.active_migration_name}-failure-report.json",
+                                mime="application/json",
+                                width="stretch",
+                                key="failed_run_report_download",
+                            )
+                            st.caption("No migrated ZIP was created before the failure; the report preserves the run details for review.")
 
                 if status.get("result"):
                     with st.expander("Workflow payload", expanded=False):