import openai
import os

# OpenAI API 키 설정
openai.api_key = os.getenv("OPENAI_API_KEY")  # 환경 변수에서 API 키를 불러옵니다.

def generate_keywords_with_gpt(scene_descriptions):
    responses = []
    for description in scene_descriptions:
        response = openai.Completion.create(
            engine="text-davinci-003",  # 혹은 다른 GPT 모델을 선택할 수 있습니다.
            prompt=f"Generate a representative keyword for the following scene description: {description}",
            max_tokens=60,  # 필요한 토큰 수를 조절할 수 있습니다.
            temperature=0.7
        )
        responses.append(response.choices[0].text.strip())
    return responses

# 사용 예시
scene_descriptions = ["Enter description for Scene 1", "Enter description for Scene 2"]
keywords = generate_keywords_with_gpt(scene_descriptions)
print(keywords)