Update app.py

app.py

@@ -446,7 +446,85 @@ def create_gradio_interface():
                             param_14=enable_enhancement,
                             api_name="/check_password"
                         )
-                        return result
+                        
+                        # API 응답 처리 - 딕셔너리를 원본처럼 PIL Image로 변환
+                        if result and len(result) >= 7:
+                            original_output, original_download, enhanced_output, enhanced_download, prompt_output, info, error = result
+                            
+                            # Gallery 데이터를 PIL Image 리스트로 변환
+                            def convert_gallery_to_images(gallery_data):
+                                if not gallery_data:
+                                    return []
+                                
+                                images = []
+                                for item in gallery_data:
+                                    try:
+                                        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 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)
+                                        elif isinstance(item, str):
+                                            # 직접 URL인 경우
+                                            response = requests.get(item)
+                                            if response.status_code == 200:
+                                                pil_image = Image.open(io.BytesIO(response.content))
+                                                images.append(pil_image)
+                                    except Exception as e:
+                                        logger.warning(f"Failed to convert gallery item to PIL Image: {e}")
+                                        continue
+                                
+                                return images
+                            
+                            # Gallery 데이터 변환
+                            converted_original = convert_gallery_to_images(original_output)
+                            converted_enhanced = convert_gallery_to_images(enhanced_output)
+                            
+                            # 다운로드 파일 처리 - 원본 그대로 전달
+                            original_file = None
+                            enhanced_file = None
+                            
+                            # 원본 이미지 파일 생성
+                            if converted_original:
+                                try:
+                                    original_file_path = f"original_image.{output_format}"
+                                    converted_original[0].save(original_file_path, format=output_format.upper())
+                                    original_file = original_file_path
+                                except Exception as e:
+                                    logger.error(f"Error saving original image: {e}")
+                            
+                            # 개선된 이미지 파일 생성
+                            if converted_enhanced:
+                                try:
+                                    enhanced_file_path = f"enhanced_image.{output_format}"
+                                    converted_enhanced[0].save(enhanced_file_path, format=output_format.upper())
+                                    enhanced_file = enhanced_file_path
+                                except Exception as e:
+                                    logger.error(f"Error saving enhanced image: {e}")
+                            
+                            return (
+                                converted_original,  # PIL Image 리스트
+                                original_file,       # 다운로드 파일 경로
+                                converted_enhanced,  # PIL Image 리스트
+                                enhanced_file,       # 다운로드 파일 경로
+                                prompt_output,       # 프롬프트 텍스트
+                                info,               # 처리 정보
+                                error               # 오류 메시지
+                            )
+                        else:
+                            return ([], None, [], None, "", "", "API 응답이 올바르지 않습니다.")
+                            
                     finally:
                         # 임시 파일 정리
                         if os.path.exists(temp_path):
@@ -457,6 +535,7 @@ def create_gradio_interface():
                                 
                 except Exception as e:
                     logger.error(f"Error in check_password: {e}")
+                    logger.error(f"Full traceback: {traceback.format_exc()}")
                     return ([], None, [], None, "", "", f"오류 발생: {str(e)}")
             
             # 프롬프트 생성 버튼 클릭 이벤트