Improve UI

services/training/app/services/training_service.py

@@ -364,6 +364,10 @@ class EnhancedTrainingService:
                     job_id, results=json_safe_result
                 )
 
+                # CRITICAL: Commit the session to persist the completed status to database
+                # Without this commit, the status update is lost when the session closes
+                await session.commit()
+
                 logger.info("Enhanced training job completed successfully",
                            job_id=job_id,
                            models_created=len(stored_models))
@@ -380,7 +384,10 @@ class EnhancedTrainingService:
                 await self.training_log_repo.complete_training_log(
                     job_id, error_message=str(e)
                 )
-                
+
+                # Commit the failure status to database
+                await session.commit()
+
                 error_result = {
                     "job_id": job_id,
                     "tenant_id": tenant_id,