with open('/home/crogers2287/projects/dope_wars_2026/lib/screens/market_screen.dart', 'r') as f:
    content = f.read()

# Add chapter model import
old_imports = "import '../models/reputation.dart';"
new_imports = """import '../models/reputation.dart';
import '../models/chapter.dart';"""
content = content.replace(old_imports, new_imports, 1)

# Add flutter_animate import
old_flutter = "import 'package:flutter_riverpod/flutter_riverpod.dart';"
new_flutter = """import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:flutter_animate/flutter_animate.dart';"""
content = content.replace(old_flutter, new_flutter, 1)

# Add chapter banner before the ListView.builder header item
old_header_return = """        return Column(
          children: [
            if (!canDeal)"""

new_header_return = """        final chapter = currentChapterFor(state.player);
        final chapData = chapterDataFor(chapter);
        
        return Column(
          children: [
            _ChapterBanner(chapter: chapter, chapData: chapData, player: state.player),
            if (!canDeal)"""
content = content.replace(old_header_return, new_header_return, 1)

with open('/home/crogers2287/projects/dope_wars_2026/lib/screens/market_screen.dart', 'w') as f:
    f.write(content)
print("Done")