Update app.py

app.py

@@ -8,7 +8,6 @@ import io
 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,
@@ -21,107 +20,136 @@ def search_artworks(query, is_public_domain=False):
     
     try:
         response = requests.get(search_url, params=params)
-        response.raise_for_status()  # 에러 체크
+        response.raise_for_status()
         results = response.json()
         
         if "data" not in results:
-            return [], "검색 결과가 없습니다."
+            return [], "No search results found."
         
         images = []
         captions = []
         
         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')}\n작가: {artwork.get('artist_display', 'Unknown')}\n제작연도: {artwork.get('date_display', 'Unknown')}"""
+                artwork_info = f"""Title: {artwork.get('title', 'Unknown')}\nArtist: {artwork.get('artist_display', 'Unknown')}\nDate: {artwork.get('date_display', 'Unknown')}"""
                 
                 try:
-                    img_response = requests.get(image_url, timeout=5)  # 타임아웃 설정
-                    img_response.raise_for_status()  # 이미지 다운로드 에러 체크
+                    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))  # 다시 열기
+                    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}")
+                    print(f"Error processing image: {e}")
                     continue
         
         if not images:
-            return [], "검색된 작품에 대한 이미지를 불러올 수 없습니다."
+            return [], "Unable to load images for the searched artworks."
         
         return images, "\n\n".join(captions)
     
     except Exception as e:
-        print(f"API 요청 중 에러 발생: {e}")
-        return [], f"검색 중 오류가 발생했습니다: {str(e)}"
+        print(f"API request error: {e}")
+        return [], f"An error occurred during search: {str(e)}"
 
-def get_gallery_info(gallery_id):
-    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')}
-        """
-    except Exception as e:
-        return f"갤러리 정보 조회 중 오류가 발생했습니다: {str(e)}"
+# Custom CSS for styling
+custom_css = """
+.gradio-container {
+    background: linear-gradient(to right, #1a1a1a, #2d2d2d);
+    color: #ffffff;
+}
+.gr-button {
+    background: linear-gradient(to right, #c94b4b, #4b134f);
+    border: none;
+    color: white;
+    font-weight: bold;
+}
+.gr-button:hover {
+    background: linear-gradient(to right, #4b134f, #c94b4b);
+    transform: scale(1.05);
+    transition: all 0.3s ease;
+}
+.gr-input {
+    border: 2px solid #4b134f;
+    background: rgba(255, 255, 255, 0.1);
+    color: white;
+}
+.gr-form {
+    background: rgba(0, 0, 0, 0.2);
+    border-radius: 15px;
+    padding: 20px;
+}
+.gr-box {
+    border-radius: 15px;
+    border: 2px solid #4b134f;
+}
+.gr-gallery {
+    background: rgba(0, 0, 0, 0.3);
+    border-radius: 15px;
+    padding: 10px;
+}
+"""
 
-# Gradio 인터페이스 구성
-with gr.Blocks() as demo:
-    gr.Markdown("# Art Institute of Chicago Gallery")
+# Gradio interface
+with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown(
+        """
+        # 🎨 Art Institute of Chicago Explorer
+        *Discover masterpieces from one of the world's premier art collections through an elegant and intuitive interface.*
+        """
+    )
     
-    with gr.Tab("작품 검색"):
-        with gr.Row():
-            search_input = gr.Textbox(label="검색어를 입력하세요")
-            public_domain = gr.Checkbox(label="공개 도메인 작품만 보기")
-        search_btn = gr.Button("검색")
-        
-        with gr.Row():
-            gallery = gr.Gallery(
-                label="검색 결과",
-                show_label=True,
-                elem_id="gallery",
-                columns=[2],
-                rows=[2],
-                height="auto"
+    with gr.Row():
+        with gr.Column():
+            search_input = gr.Textbox(
+                label="Enter your search term",
+                placeholder="e.g., Monet, Impressionism, landscape...",
+            )
+            public_domain = gr.Checkbox(
+                label="Show only public domain artworks",
+                value=False
+            )
+            search_btn = gr.Button(
+                "🔍 Search",
+                variant="primary"
             )
-        
-        info = gr.Textbox(label="작품 정보", lines=10)
-        
-        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
+    with gr.Row():
+        gallery = gr.Gallery(
+            label="Search Results",
+            show_label=True,
+            elem_id="gallery",
+            columns=[2],
+            rows=[2],
+            height="500px",
+            object_fit="contain"
         )
+    
+    info = gr.Textbox(
+        label="Artwork Details",
+        lines=10,
+        show_label=True
+    )
+    
+    search_btn.click(
+        fn=search_artworks,
+        inputs=[search_input, public_domain],
+        outputs=[gallery, info]
+    )
 
-demo.launch()
+# Launch with a dark theme
+demo.launch(theme=gr.themes.Monochrome(
+    primary_hue="purple",
+    secondary_hue="red",
+    neutral_hue="slate",
+    font=["Helvetica", "ui-sans-serif"]
+))