Update app.py

app.py

@@ -5,92 +5,197 @@ from PIL import Image
 import io
 
 # API 기본 URL
-BASE_URL = "https://api.artic.edu/api/v1"
+BASE_URL = "https://collectionapi.metmuseum.org/public/collection/v1"
 
-def search_artworks(query, is_public_domain=False):
-    # 아트워크 검색 API 호출
-    search_url = f"{BASE_URL}/artworks/search"
+def get_departments():
+    """Get all departments from the Met API"""
+    response = requests.get(f"{BASE_URL}/departments")
+    return response.json()['departments']
+
+def search_artworks(query, department_id=None, is_highlight=False, has_images=True, 
+                   is_on_view=False, medium=None, geo_location=None):
+    """Search artworks with various filters"""
+    search_url = f"{BASE_URL}/search"
     params = {
-        "q": query,
-        "limit": 10,
-        "fields": "id,title,artist_display,date_display,image_id,is_public_domain",
+        'q': query,
+        'hasImages': has_images,
+        'isHighlight': is_highlight,
+        'isOnView': is_on_view
     }
     
-    if is_public_domain:
-        params["query"] = {"term": {"is_public_domain": True}}
+    if department_id:
+        params['departmentId'] = department_id
+    if medium:
+        params['medium'] = medium
+    if geo_location:
+        params['geoLocation'] = geo_location
         
-    response = requests.get(search_url, params=params)
-    results = response.json()
-    
-    if "data" not in results:
-        return "검색 결과가 없습니다."
+    try:
+        # Get object IDs from search
+        response = requests.get(search_url, params=params)
+        results = response.json()
         
-    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"
+        if not results.get('objectIDs'):
+            return [], "No results found."
             
-            # 작품 정보 포맷팅
-            artwork_info = f"""
-            제목: {artwork.get('title', 'Unknown')}
-            작가: {artwork.get('artist_display', 'Unknown')}
-            제작연도: {artwork.get('date_display', 'Unknown')}
-            """
+        # Limit to first 12 objects for performance
+        object_ids = results['objectIDs'][:12]
+        
+        images = []
+        captions = []
+        
+        # Get details for each object
+        for object_id in object_ids:
+            object_url = f"{BASE_URL}/objects/{object_id}"
+            object_response = requests.get(object_url)
+            artwork = object_response.json()
             
-            # 이미지 다운로드 및 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 "갤러리 정보를 찾을 수 없습니다."
+            if artwork.get('primaryImage'):
+                try:
+                    img_response = requests.get(artwork['primaryImage'], timeout=10)
+                    img = Image.open(io.BytesIO(img_response.content))
+                    
+                    if img.mode in ('RGBA', 'LA') or (img.mode == 'P' and 'transparency' in img.info):
+                        img = img.convert('RGB')
+                    
+                    artwork_info = f"""
+                    Title: {artwork.get('title', 'Unknown')}
+                    Artist: {artwork.get('artistDisplayName', 'Unknown')}
+                    Date: {artwork.get('objectDate', 'Unknown')}
+                    Medium: {artwork.get('medium', 'Unknown')}
+                    Department: {artwork.get('department', 'Unknown')}
+                    """
+                    
+                    images.append(img)
+                    captions.append(artwork_info)
+                except Exception as e:
+                    print(f"Error processing image for object {object_id}: {e}")
+                    continue
         
-    gallery_data = gallery["data"]
-    return f"""
-    갤러리 이름: {gallery_data.get('title', 'Unknown')}
-    위치: {gallery_data.get('location', 'Unknown')}
-    설명: {gallery_data.get('description', 'No description available')}
-    """
+        return images, "\n\n".join(captions)
+        
+    except Exception as e:
+        return [], f"An error occurred: {str(e)}"
+
+# Custom CSS
+custom_css = """
+.gradio-container {
+    background: linear-gradient(135deg, #1a1a1a, #2d2d2d) !important;
+    color: #ffffff !important;
+}
+.gr-button {
+    background: linear-gradient(135deg, #8e2de2, #4a00e0) !important;
+    border: none !important;
+    color: white !important;
+    font-weight: bold !important;
+    padding: 10px 20px !important;
+    font-size: 1.1em !important;
+    box-shadow: 0 4px 15px rgba(0,0,0,0.2) !important;
+}
+.gr-button:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 6px 20px rgba(0,0,0,0.3) !important;
+    transition: all 0.3s ease;
+}
+.gr-input, .gr-select {
+    border: 2px solid #4a00e0 !important;
+    background: rgba(255, 255, 255, 0.1) !important;
+    color: white !important;
+    font-size: 1.1em !important;
+    border-radius: 8px !important;
+}
+.gr-gallery {
+    background: rgba(0, 0, 0, 0.3) !important;
+    border-radius: 15px !important;
+    padding: 20px !important;
+    min-height: 800px !important;
+    box-shadow: 0 8px 32px rgba(0,0,0,0.3) !important;
+}
+.title-text {
+    text-align: center !important;
+    color: #ffffff !important;
+    font-size: 2.8em !important;
+    margin-bottom: 0.3em !important;
+    text-shadow: 2px 2px 4px rgba(0,0,0,0.5) !important;
+    font-weight: bold !important;
+}
+.subtitle-text {
+    text-align: center !important;
+    color: #cccccc !important;
+    font-size: 1.3em !important;
+    margin-bottom: 2em !important;
+    font-style: italic !important;
+    text-shadow: 1px 1px 2px rgba(0,0,0,0.3) !important;
+}
+"""
 
-# Gradio 인터페이스 구성
-with gr.Blocks() as demo:
-    gr.Markdown("# Art Institute of Chicago Gallery")
+# Get departments for dropdown
+departments = get_departments()
+department_choices = {dept['displayName']: dept['departmentId'] for dept in departments}
+
+# Gradio interface
+with gr.Blocks(css=custom_css) as demo:
+    gr.HTML(
+        """
+        <div class="title-text">🎨 The Met Art Explorer</div>
+        <div class="subtitle-text">Explore over 5,000 years of art from The Metropolitan Museum of Art's collection</div>
+        """
+    )
     
-    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.Row():
+        with gr.Column(scale=3):
+            search_input = gr.Textbox(
+                label="Search artwork",
+                placeholder="Enter keywords (e.g., sunflowers, portrait, landscape...)"
+            )
+        with gr.Column(scale=2):
+            department_dropdown = gr.Dropdown(
+                choices=list(department_choices.keys()),
+                label="Department",
+                value=None
+            )
     
-    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
-        )
+    with gr.Row():
+        with gr.Column():
+            highlight_checkbox = gr.Checkbox(label="Show Highlights Only", value=False)
+            on_view_checkbox = gr.Checkbox(label="Currently On View", value=False)
+        with gr.Column():
+            medium_input = gr.Textbox(
+                label="Medium",
+                placeholder="e.g., Paintings, Sculpture, Ceramics"
+            )
+            location_input = gr.Textbox(
+                label="Geographic Location",
+                placeholder="e.g., France, Japan, Egypt"
+            )
+    
+    search_btn = gr.Button("🔍 Search Collection", variant="primary")
+    
+    gallery = gr.Gallery(
+        label="Search Results",
+        show_label=True,
+        elem_id="gallery",
+        columns=3,
+        rows=4,
+        height="800px",
+        object_fit="contain"
+    )
+    
+    info = gr.Textbox(
+        label="Artwork Details",
+        lines=10,
+        show_label=True
+    )
+    
+    def handle_search(query, dept, highlight, on_view, medium, location):
+        dept_id = department_choices.get(dept) if dept else None
+        return search_artworks(query, dept_id, highlight, True, on_view, medium, location)
+    
+    search_btn.click(
+        fn=handle_search,
+        inputs=[search_input, department_dropdown, highlight_checkbox, 
+                on_view_checkbox, medium_input, location_input],
+        outputs=[gallery, info]
+    )
 
 demo.launch()