UI fixes and cleanup

[dnakov]

Checklist

• [ ] Closing issues: #issue
• [X] Mark this if you consider it ready to merge
• [ ] I've added tests (optional)
• [ ] I wrote some documentation

Description

Fixing modals, focus, etc. Cleaned up to use the same new auto.py

[trufae]

You are on fire 🔥