import gradio as gr
import os
from PIL import Image
import imageio

# 저장 경로
UPLOAD_DIR = "uploads"
os.makedirs(UPLOAD_DIR, exist_ok=True)

def save_images(image1, image2):
    path1 = os.path.join(UPLOAD_DIR, "cut1.png")
    path2 = os.path.join(UPLOAD_DIR, "cut2.png")

    image1.save(path1)
    image2.save(path2)

    # 여기서는 예시로 두 이미지 그대로 이어붙인 gif를 리턴
    gif_path = os.path.join(UPLOAD_DIR, "output.gif")
    imageio.mimsave(gif_path, [image1, image2], fps=1)

    return gif_path

iface = gr.Interface(
    fn=save_images,
    inputs=[
        gr.Image(label="컷1 업로드", type="pil"),
        gr.Image(label="컷2 업로드", type="pil")
    ],
    outputs=gr.Video(label="결과 애니메이션"),
    title="🌀 컷 기반 애니메이션 생성기 (예시)",
    description="두 장의 컷을 업로드하면, 애니메이션으로 이어주는 결과를 보여줍니다."
)

iface.launch()