from openai import OpenAI
from dotenv import load_dotenv
import os
from PIL import Image
from ImageProcess import classify_image
load_dotenv()
api_key = os.environ.get("OPENAI_API_KEY")  # 실제 키 입력

def create_funny_caption(label):
    client = OpenAI(
        api_key=api_key,
    )
    print(f"label:{label}")
    prompt = f"""이 이미지는 label: "{label}"이 표시된 이미지인데,
        resnet50 모델이 분류한 결과야.
        이 Label 과 관련된 기발하고 재치있는 농담을 해 줬으면 좋겠어.
        """
    print("prompt:", prompt)
    response = client.responses.create(
        model="gpt-3.5-turbo",
        instructions="""
        너는 밈 아이디어를 생성하는 창의적 코미디 작가야.
        니가 이미지를 직접 보지 못한다는 것은 알고 있어. 
        그러니 이미지를 직접 보지 못한다는 말은 하지 말아줘.
        대신 이미지를 설명하는 prompt 를 이해한 뒤, 짧고 재밌는 문구를 만들어줘.
        문구는 100자 이내로 해줘.
        문구는 한국어로 작성해줘.
        분석라벨 : 뒤에 오는 라벨들도 한국어로 번역해서 보여주면 더 좋겠어.
        """,
        input=prompt
    )
    return response.output_text