For me, the ‘AI Written’ category should be used for stories which are completely AI generated, or only have minor modifications to the AI output.
What constitutes ‘minor modification’? I don’t think that’s easy to define, I guess we have to leave it up to the author. Certainly what you describe I wouldn’t call ‘AI Written’, you’re really just using for inspiration / assistance, not getting it to write the story.