forked from ckh08045/AutoPercenty
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
import openai
|
|
import requests
|
|
from PIL import Image
|
|
from io import BytesIO
|
|
|
|
# OpenAI API 키 설정
|
|
openai.api_key = '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'
|
|
|
|
def download_image(image_url):
|
|
"""주어진 URL에서 이미지를 다운로드하고 PIL 이미지 객체로 반환합니다."""
|
|
response = requests.get(image_url)
|
|
image = Image.open(BytesIO(response.content))
|
|
return image
|
|
|
|
def generate_product_description(image_url, product_name):
|
|
"""이미지와 상품명을 사용하여 상품 설명을 생성합니다."""
|
|
|
|
# 이미지 다운로드 (이 예시에서는 사용하지 않음, 실제 API 요청에 이미지 직접 사용 불가)
|
|
image = download_image(image_url)
|
|
|
|
# 상품명을 사용하여 챗GPT에 설명 요청
|
|
prompt = f"상품명: {product_name}\n상품의 용도, 설명, 무게, 홍보 문구를 작성해주세요."
|
|
|
|
response = openai.Completion.create(
|
|
model="text-davinci-003", # 모델 버전에 따라 변경 가능
|
|
prompt=prompt,
|
|
temperature=0.7,
|
|
max_tokens=150
|
|
)
|
|
|
|
# 생성된 텍스트 반환
|
|
return response.choices[0].text.strip()
|
|
|
|
# 예시 사용
|
|
image_url = "https://example.com/your-image.jpg"
|
|
product_name = "예시 상품명"
|
|
description = generate_product_description(image_url, product_name)
|
|
print(description)
|