I have roborock q revo and find the roborock app much cleaner to use than the mi home app.
If I am understanding what you want correctly (and I might not be!), you should be able to get this functionality using the standard integration.
Take a look at this thread: