Update app.py

app.py

@@ -18,51 +18,72 @@ def search_artworks(query, is_public_domain=False):
     
     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 "검색 결과가 없습니다."
+    try:
+        response = requests.get(search_url, params=params)
+        response.raise_for_status()  # 에러 체크
+        results = response.json()
+        
+        if "data" not in results:
+            return [], "검색 결과가 없습니다."
+        
+        images = []
+        captions = []
         
-    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
+        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"
                 
-    return output if output else "검색 결과가 없습니다."
+                # 작품 정보 포맷팅
+                artwork_info = f"""제목: {artwork.get('title', 'Unknown')}\n작가: {artwork.get('artist_display', 'Unknown')}\n제작연도: {artwork.get('date_display', 'Unknown')}"""
+                
+                try:
+                    img_response = requests.get(image_url, timeout=5)  # 타임아웃 설정
+                    img_response.raise_for_status()  # 이미지 다운로드 에러 체크
+                    
+                    # 이미지 검증
+                    img = Image.open(io.BytesIO(img_response.content))
+                    img.verify()  # 이미지 파일 검증
+                    img = Image.open(io.BytesIO(img_response.content))  # 다시 열기
+                    
+                    # RGB 모드로 변환 (알파 채널 제거)
+                    if img.mode in ('RGBA', 'LA') or (img.mode == 'P' and 'transparency' in img.info):
+                        img = img.convert('RGB')
+                    
+                    images.append(img)
+                    captions.append(artwork_info)
+                except Exception as e:
+                    print(f"이미지 처리 중 에러 발생: {e}")
+                    continue
+        
+        if not images:
+            return [], "검색된 작품에 대한 이미지를 불러올 수 없습니다."
+        
+        return images, "\n\n".join(captions)
+    
+    except Exception as e:
+        print(f"API 요청 중 에러 발생: {e}")
+        return [], f"검색 중 오류가 발생했습니다: {str(e)}"
 
 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 "갤러리 정보를 찾을 수 없습니다."
+    try:
+        gallery_url = f"{BASE_URL}/galleries/{gallery_id}"
+        response = requests.get(gallery_url)
+        response.raise_for_status()
+        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')}
-    """
+        gallery_data = gallery["data"]
+        return f"""
+        갤러리 이름: {gallery_data.get('title', 'Unknown')}
+        위치: {gallery_data.get('location', 'Unknown')}
+        설명: {gallery_data.get('description', 'No description available')}
+        """
+    except Exception as e:
+        return f"갤러리 정보 조회 중 오류가 발생했습니다: {str(e)}"
 
 # Gradio 인터페이스 구성
 with gr.Blocks() as demo:
@@ -73,8 +94,18 @@ with gr.Blocks() as demo:
             search_input = gr.Textbox(label="검색어를 입력하세요")
             public_domain = gr.Checkbox(label="공개 도메인 작품만 보기")
         search_btn = gr.Button("검색")
-        gallery = gr.Gallery(label="검색 결과")
-        info = gr.Textbox(label="작품 정보", lines=3)
+        
+        with gr.Row():
+            gallery = gr.Gallery(
+                label="검색 결과",
+                show_label=True,
+                elem_id="gallery",
+                columns=[2],
+                rows=[2],
+                height="auto"
+            )
+        
+        info = gr.Textbox(label="작품 정보", lines=10)
         
         search_btn.click(
             fn=search_artworks,