from app.database import engine
from sqlalchemy import text

def migrate():
    with engine.connect() as conn:
        try:
            conn.execute(text("ALTER TABLE video_files ADD COLUMN contact_sheet_path VARCHAR"))
            print("Added contact_sheet_path")
        except Exception as e:
            print(f"Skipped contact_sheet_path (probably exists): {e}")
            
        try:
            conn.execute(text("ALTER TABLE video_files ADD COLUMN scene_phash VARCHAR"))
            print("Added scene_phash")
        except Exception as e:
            print(f"Skipped scene_phash (probably exists): {e}")

if __name__ == "__main__":
    migrate()