import re

with open('kifi-api/src/main/java/com/kifi/api/service/ProjectService.java', 'r') as f:
    content = f.read()

# Remove Json import
content = content.replace('import io.r2dbc.postgresql.codec.Json;\n', '')

# Fix addComment logic
content = content.replace('comment.setImages(Json.of(imagesJson));', 'comment.setImages(imagesJson);')

with open('kifi-api/src/main/java/com/kifi/api/service/ProjectService.java', 'w') as f:
    f.write(content)