import psycopg2
import os
import sys

print("\n🎯 FIXING: First Economy Mail → Browser Category\n")

# Quick connection
try:
    conn = psycopg2.connect(
        host="localhost",
        database="timesheet",
        user="postgres",
        password="asdf1234"
    )
    cur = conn.cursor()
    
    # Fix it
    cur.execute("""
        UPDATE activity_records 
        SET category = 'browser', subcategory = 'email'
        WHERE category = 'productive' 
        AND window_title LIKE '%First Economy Mail%'
    """)
    
    count = cur.rowcount
    conn.commit()
    
    print(f"✅ Fixed {count} 'First Economy Mail' activities\n")
    
    # Also fix any other mail
    cur.execute("""
        UPDATE activity_records 
        SET category = 'browser', subcategory = 'email'
        WHERE category = 'productive' 
        AND LOWER(window_title) LIKE '%mail%'
    """)
    
    count2 = cur.rowcount
    conn.commit()
    
    print(f"✅ Fixed {count2} other mail activities\n")
    
    print("⚠️  NOW CLEAR YOUR BROWSER CACHE!")
    print("   Press Ctrl+F5 in your browser")
    print("   Or open in Incognito mode\n")
    
    conn.close()
    
except Exception as e:
    print(f"Error: {e}")
    print("\nTry this SQL command manually:")
    print("UPDATE activity_records SET category='browser' WHERE window_title LIKE '%First Economy Mail%'")

input("\nPress Enter to exit...")