Have you tried Gemini 3.1 Pro for that yet? They put a ton of work into improving its frontend abilities.