def pick_images_for(section: str): names = sorted(os.listdir(IMAGE_DIR)) prompt = ( f"보고서 ‘{section}’ 섹션에 적합한 이미지를 아래 목록에서 1~2개 추천해 파일명만 리턴하세요:\n" + "\n".join(names) ) resp = call_gpt("당신은 디자인 어시스턴트입니다.", prompt) picked = [] for line in resp.splitlines(): fn = line.strip() if fn in names: picked.append(fn) return picked # ─── 5) 디자인 템플릿 선택 ───────────────────────