import os

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

if "export const viewport" not in content:
    viewport_export = """
export const viewport = {
  width: 'device-width',
  initialScale: 1,
};
"""
    content = content.replace('export default function RootLayout', viewport_export + '\nexport default function RootLayout')
    with open(filepath, 'w') as f:
        f.write(content)
    print("Viewport restored")