Update app.py

app.py

@@ -451,46 +451,86 @@ def create_gradio_interface():
                         if result and len(result) >= 7:
                             original_output, original_download, enhanced_output, enhanced_download, prompt_output, info, error = result
                             
+                            # 디버깅: API 응답 구조 로깅
+                            logger.info(f"Original output type: {type(original_output)}")
+                            logger.info(f"Enhanced output type: {type(enhanced_output)}")
+                            if original_output:
+                                logger.info(f"Original output first item: {type(original_output[0]) if original_output else 'Empty'}")
+                                if original_output and isinstance(original_output[0], dict):
+                                    logger.info(f"Original output keys: {list(original_output[0].keys())}")
+                            
                             # Gallery 데이터를 PIL Image 리스트로 변환
                             def convert_gallery_to_images(gallery_data):
                                 if not gallery_data:
                                     return []
                                 
                                 images = []
-                                for item in gallery_data:
+                                logger.info(f"Converting gallery data, length: {len(gallery_data)}")
+                                
+                                for i, item in enumerate(gallery_data):
                                     try:
+                                        logger.info(f"Processing gallery item {i}: {type(item)}")
+                                        
+                                        img_url = None
                                         if isinstance(item, dict):
-                                            # 딕셔너리에서 이미지 URL/경로 추출
-                                            img_info = None
-                                            if 'image' in item and isinstance(item['image'], dict):
-                                                img_info = item['image']
-                                            elif 'path' in item or 'url' in item:
-                                                img_info = item
+                                            # 여러 가지 가능한 구조 시도
+                                            if 'image' in item:
+                                                if isinstance(item['image'], dict):
+                                                    img_url = item['image'].get('url') or item['image'].get('path')
+                                                else:
+                                                    img_url = item['image']
+                                            elif 'url' in item:
+                                                img_url = item['url']
+                                            elif 'path' in item:
+                                                img_url = item['path']
                                             
-                                            if img_info:
-                                                img_url = img_info.get('url') or img_info.get('path')
-                                                if img_url:
-                                                    # URL에서 이미지 다운로드하여 PIL Image로 변환
-                                                    response = requests.get(img_url)
-                                                    if response.status_code == 200:
-                                                        pil_image = Image.open(io.BytesIO(response.content))
-                                                        images.append(pil_image)
+                                            logger.info(f"Extracted URL from dict: {img_url}")
                                         elif isinstance(item, str):
                                             # 직접 URL인 경우
-                                            response = requests.get(item)
+                                            img_url = item
+                                            logger.info(f"Direct URL: {img_url}")
+                                        
+                                        if img_url:
+                                            # URL이 상대 경로인 경우 절대 경로로 변환
+                                            if img_url.startswith('/'):
+                                                img_url = f"https://happydoggg-49493h.hf.space/gradio_api/file={img_url}"
+                                            elif not img_url.startswith('http'):
+                                                img_url = f"https://happydoggg-49493h.hf.space/gradio_api/file={img_url}"
+                                            
+                                            logger.info(f"Final URL: {img_url}")
+                                            
+                                            # URL에서 이���지 다운로드하여 PIL Image로 변환
+                                            response = requests.get(img_url)
                                             if response.status_code == 200:
                                                 pil_image = Image.open(io.BytesIO(response.content))
                                                 images.append(pil_image)
+                                                logger.info(f"Successfully converted image {i}")
+                                            else:
+                                                logger.warning(f"Failed to download image from {img_url}, status: {response.status_code}")
+                                        else:
+                                            logger.warning(f"No URL found in item {i}: {item}")
+                                            
                                     except Exception as e:
-                                        logger.warning(f"Failed to convert gallery item to PIL Image: {e}")
+                                        logger.error(f"Failed to convert gallery item {i}: {e}")
+                                        logger.error(f"Item content: {item}")
                                         continue
                                 
+                                logger.info(f"Converted {len(images)} images from gallery data")
                                 return images
                             
                             # Gallery 데이터 변환
                             converted_original = convert_gallery_to_images(original_output)
                             converted_enhanced = convert_gallery_to_images(enhanced_output)
                             
+                            # 변환된 이미지가 없는 경우, 원본 데이터를 그대로 반환 시도
+                            if not converted_original and original_output:
+                                logger.info("No converted original images, trying to use original data directly")
+                                converted_original = original_output
+                            
+                            if not converted_enhanced and enhanced_output:
+                                logger.info("No converted enhanced images, trying to use enhanced data directly")
+                                converted_enhanced = enhanced_output
+                            
                             # 다운로드 파일 처리 - 원본 그대로 전달
                             original_file = None
                             enhanced_file = None