16 lines
603 B
Python
16 lines
603 B
Python
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) 디자인 템플릿 선택 ─────────────────────── |