import gradio as gr
from PIL import Image
from ImageProcess import classify_image
from generateText import create_funny_caption

def process_image(image):
    print(f"image:{image}")
    # 1) 이미지 분류
    label = classify_image(image)

    # 2) chatGPT에게 만우절 밈 문구 요청
    caption = create_funny_caption(label)

    return f"{caption}\n\n분석 라벨: {label}"

iface = gr.Interface(
    fn=process_image,
    inputs=gr.Image(type="pil"),
    outputs="text",
    title="AI 만우절 밈 생성기",
    description="이미지를 업로드하면 AI가 분석하고, ChatGPT로부터 밈 문구를 생성합니다."
)

iface.launch()