with open('backend/app/main.py', 'r', encoding='utf-8') as f:
    content = f.read()

new_func = '''@app.get("/api/videos/{video_id}/segment-edits", response_model=dict[str, list[schemas.SegmentEditOut]])
def list_segment_edits(video_id: str) -> dict:
    _get_video_or_404(video_id)
    return {"segments": db.list_segment_edits(video_id)}

@app.put("/api/videos/{video_id}/segment-edits"'''

content = content.replace('@app.put("/api/videos/{video_id}/segment-edits"', new_func)

with open('backend/app/main.py', 'w', encoding='utf-8') as f:
    f.write(content)
print('main.py patched')