import os

filepath = 'src/app/layout.tsx'
with open(filepath, 'r') as f:
    content = f.read()

# Remove ThemeProvider import and wrapper
content = content.replace('import { ThemeProvider } from "@/components/ThemeProvider";\n', '')
content = content.replace('<ThemeProvider\n          attribute="class"\n          defaultTheme="system"\n          enableSystem\n          disableTransitionOnChange\n        >', '')
content = content.replace('</ThemeProvider>', '')
content = content.replace('suppressHydrationWarning', '')

with open(filepath, 'w') as f:
    f.write(content)
print("Updated layout.tsx")