Update app.py

app.py

@@ -594,6 +594,25 @@ def process_uploaded_files(files):
     return result
 
 # ──────────────────────────────── Image & Utility ─────────────────────────
+
+def load_and_show_image(img_url: str, caption: str = "Image"):
+    """
+    이미지 URL이 깨져있거나 직접 표시가 안 되는 경우를 대비해
+    requests로 다운로드 후 표시하는 함수.
+    """
+    try:
+        # 이미지 다운로드
+        response = requests.get(img_url, timeout=10)
+        response.raise_for_status()
+        
+        # 메모리에서 이미지 열기
+        image = Image.open(BytesIO(response.content))
+        
+        # Streamlit으로 표시
+        st.image(image, caption=caption, use_column_width=True)
+    except Exception as e:
+        st.warning(f"이미지 로딩 실패: {e}")
+
 def generate_image(prompt, w=768, h=768, g=3.5, steps=30, seed=3):
     """Image generation function (via Gradio endpoint)."""
     if not prompt:
@@ -815,7 +834,8 @@ def perplexity_app():
                             img_url = img_data.get('url', '')
                             caption = img_data.get('title', 'Related image')
                             if img_url:
-                                st.image(img_url, caption=caption, use_column_width=True)
+                                # ▶ 여기서 load_and_show_image 함수를 사용하여 다운로드 후 표시
+                                load_and_show_image(img_url, caption=caption)
                                 if img_data.get('source'):
                                     st.markdown(f"[Source]({img_data['source']})")
                         except Exception as img_err:
@@ -834,7 +854,8 @@ def perplexity_app():
                         col1, col2 = st.columns([1, 3])
                         with col1:
                             try:
-                                st.image(thumbnail, width=120)
+                                # 썸네일도 다운로드 후 표시
+                                load_and_show_image(thumbnail, caption="Video Thumbnail")
                             except:
                                 st.write("🎬")
                         with col2:
@@ -908,7 +929,6 @@ def process_input(query: str, uploaded_files):
                     file_content = process_uploaded_files(uploaded_files)
             
             # 최종적으로 사용할 이미지/비디오 목록 구성
-            # (이번에는 "fallback" 없이 Brave 검색 결과만 사용)
             valid_images = []
             for img in image_results:
                 url = img.get('image_url')
@@ -991,7 +1011,7 @@ def process_input(query: str, uploaded_files):
                 # 최종 응답 표시
                 message_placeholder.markdown(full_response, unsafe_allow_html=True)
                 
-                # 실제 검색된 이미지를 UI에 표시
+                # 실제 검색된 이미지를 UI에 표시 (load_and_show_image 사용)
                 if valid_images:
                     st.subheader("Related Images")
                     image_cols = st.columns(min(3, len(valid_images)))
@@ -1002,13 +1022,13 @@ def process_input(query: str, uploaded_files):
                             with image_cols[col_idx]:
                                 img_url = img_data['url']
                                 caption = img_data['title']
-                                st.image(img_url, caption=caption, use_column_width=True)
+                                load_and_show_image(img_url, caption=caption)
                                 if img_data.get('source'):
                                     st.markdown(f"[Source]({img_data['source']})")
                         except Exception as img_err:
                             logging.warning(f"Error displaying image: {img_err}")
                 
-                # 실제 검색된 비디오를 UI에 표시
+                # 실제 검색된 비디오를 UI에 표시 (load_and_show_image로 썸네일 처리)
                 if valid_videos:
                     st.subheader("Related Videos")
                     for video in valid_videos:
@@ -1021,7 +1041,7 @@ def process_input(query: str, uploaded_files):
                                 col1, col2 = st.columns([1, 3])
                                 with col1:
                                     try:
-                                        st.image(thumbnail, width=120)
+                                        load_and_show_image(thumbnail, caption="Video Thumbnail")
                                     except:
                                         st.write("🎬")
                                 with col2: