# Quick fix - add this import to the top of your main.py

# Find this line in your main.py:
from sqlalchemy.orm import Session

# And replace it with:
from sqlalchemy.orm import Session
from sqlalchemy import text

# Or simply add this new line after your existing SQLAlchemy imports:
from sqlalchemy import text

# You also need to add the requests import for ActivityWatch testing:
import requests