import re

content = open('app/main.py', 'r', encoding='utf-8').read()

# Remove slowapi imports
content = re.sub(r'from slowapi import.*?\nlimiter = Limiter.*?\]\)\n', '', content, flags=re.DOTALL)

# Remove slowapi registration
content = re.sub(r'# Global Rate Limiting.*?app\.add_middleware\(SlowAPIMiddleware\)\n', '', content, flags=re.DOTALL)

# Import SimpleGlobalRateLimitMiddleware
content = content.replace('from app.api.middleware.security_middleware import EnterpriseSecurityMiddleware', 
'from app.api.middleware.security_middleware import EnterpriseSecurityMiddleware, SimpleGlobalRateLimitMiddleware')

# Add middleware registration
content = content.replace('app.add_middleware(EnterpriseSecurityMiddleware)', 
'''app.add_middleware(EnterpriseSecurityMiddleware)
    app.add_middleware(SimpleGlobalRateLimitMiddleware, max_requests=60, window_seconds=60)''')

open('app/main.py', 'w', encoding='utf-8').write(content)
print("Updated main.py")