import gradio as gr
import requests
import json
from PIL import Image
import io

# API 기본 URL
BASE_URL = "https://api.artic.edu/api/v1"

def search_artworks(query, is_public_domain=False):
    # 아트워크 검색 API 호출
    search_url = f"{BASE_URL}/artworks/search"
    params = {
        "q": query,
        "limit": 10,
        "fields": "id,title,artist_display,date_display,image_id,is_public_domain",
    }
    
    if is_public_domain:
        params["query"] = {"term": {"is_public_domain": True}}
        
    response = requests.get(search_url, params=params)
    results = response.json()
    
    if "data" not in results:
        return "검색 결과가 없습니다."
        
    output = []
    for artwork in results["data"]:
        if artwork.get("image_id"):
            # IIIF 이미지 URL 생성
            image_url = f"https://www.artic.edu/iiif/2/{artwork['image_id']}/full/843,/0/default.jpg"
            
            # 작품 정보 포맷팅
            artwork_info = f"""
            제목: {artwork.get('title', 'Unknown')}
            작가: {artwork.get('artist_display', 'Unknown')}
            제작연도: {artwork.get('date_display', 'Unknown')}
            """
            
            # 이미지 다운로드 및 PIL Image로 변환
            try:
                img_response = requests.get(image_url)
                img = Image.open(io.BytesIO(img_response.content))
                output.append((img, artwork_info))
            except:
                continue
                
    return output if output else "검색 결과가 없습니다."

def get_gallery_info(gallery_id):
    # 갤러리 정보 API 호출
    gallery_url = f"{BASE_URL}/galleries/{gallery_id}"
    response = requests.get(gallery_url)
    gallery = response.json()
    
    if "data" not in gallery:
        return "갤러리 정보를 찾을 수 없습니다."
        
    gallery_data = gallery["data"]
    return f"""
    갤러리 이름: {gallery_data.get('title', 'Unknown')}
    위치: {gallery_data.get('location', 'Unknown')}
    설명: {gallery_data.get('description', 'No description available')}
    """

# Gradio 인터페이스 구성
with gr.Blocks() as demo:
    gr.Markdown("# Art Institute of Chicago Gallery")
    
    with gr.Tab("작품 검색"):
        with gr.Row():
            search_input = gr.Textbox(label="검색어를 입력하세요")
            public_domain = gr.Checkbox(label="공개 도메인 작품만 보기")
        search_btn = gr.Button("검색")
        gallery = gr.Gallery(label="검색 결과")
        info = gr.Textbox(label="작품 정보", lines=3)
        
        search_btn.click(
            fn=search_artworks,
            inputs=[search_input, public_domain],
            outputs=[gallery, info]
        )
    
    with gr.Tab("갤러리 정보"):
        gallery_id = gr.Number(label="갤러리 ID를 입력하세요")
        gallery_btn = gr.Button("정보 조회")
        gallery_info = gr.Textbox(label="갤러리 정보", lines=5)
        
        gallery_btn.click(
            fn=get_gallery_info,
            inputs=gallery_id,
            outputs=gallery_info
        )

demo.launch()