A good idea for compact water distillation would be creating a sort of upside-down floodgate to a top chamber for the steam to gather in, that has Freezers surrounding it so it's forced back into water and down into a third chamber, which would hold the "distilled" water.