36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
import requests
|
|
import google.generativeai as genai
|
|
import logging
|
|
|
|
# 로거 인스턴스 가져오기
|
|
logger = logging.getLogger('default_logger')
|
|
# API 키 설정
|
|
GOOGLE_API_KEY = 'AIzaSyCER9mD617P5OGaoHCK7drsTkmXUIzFn4U'
|
|
genai.configure(api_key=GOOGLE_API_KEY)
|
|
|
|
def download_image(image_url):
|
|
"""이미지 URL에서 이미지를 다운로드하고 바이트 데이터를 반환합니다."""
|
|
response = requests.get(image_url)
|
|
response.raise_for_status() # 에러가 발생했을 경우 예외를 발생시킵니다.
|
|
return response.content
|
|
|
|
def generate_image_description(image_data):
|
|
"""다운로드한 이미지 데이터로부터 설명을 생성합니다."""
|
|
# 'gemini-pro-vision' 모델을 사용하여 이미지 설명 생성
|
|
model = genai.GenerativeModel('gemini-pro-vision')
|
|
# prompt = "이 이미지를 설명하는 홍보문구를 3줄 만들어줘."
|
|
prompt = "주어진 상품이미지에서 1. 어떤상품인지 자세히 설명해 주고 2.어떤용도인지 자세히 설명하고 3. 이 상품의 무게를 대략적으로 추정해줘."
|
|
|
|
# 이미지 데이터와 텍스트 프롬프트를 함께 제출
|
|
response = model.generate_content([prompt, {"mime_type": "image/jpeg", "data": image_data}])
|
|
return response.text
|
|
|
|
# 이미지 URL
|
|
# image_url = "https://i.pinimg.com/736x/87/c4/e0/87c4e0b82e5b712497a400c84f9a992c.jpg"
|
|
|
|
# 이미지 다운로드 및 설명 생성
|
|
# image_data = download_image(image_url)
|
|
# description = generate_image_description(image_data)
|
|
|
|
# logger.debug(description)
|