Update app.py

app.py

@@ -188,9 +188,18 @@ def api_generate_prompt_with_password_check(password, bg_type, simple, studio, n
 def api_check_password(password, image_file, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, request_text, quality_level, aspect_ratio, output_format, enable_enhancement):
     """이미지 처리 API 호출"""
     try:
+        # 이미지 파일을 gradio_client의 handle_file로 처리
+        from gradio_client import handle_file
+        
+        # 파일이 경로인 경우 handle_file로 래핑
+        if isinstance(image_file, str):
+            image_handle = handle_file(image_file)
+        else:
+            image_handle = image_file
+            
         result = client.predict(
             password=password,
-            param_1=image_file,
+            param_1=image_handle,  # handle_file로 처리된 이미지
             param_2=bg_type,
             param_3=simple,
             param_4=studio,
@@ -209,6 +218,8 @@ def api_check_password(password, image_file, bg_type, simple, studio, nature, in
         return result
     except Exception as e:
         logger.error(f"Error calling check_password API: {e}")
+        # 더 자세한 오류 정보 로깅
+        logger.error(f"Full error details: {traceback.format_exc()}")
         return [], None, [], None, "", "", f"API 호출 오류: {str(e)}"
 
 # Gradio 인터페이스 구성 (완전히 동일한 UI)
@@ -395,25 +406,42 @@ def create_gradio_interface():
                 if image is None:
                     return [], None, [], None, "", "", "이미지를 업로드해야 합니다."
                 
-                # 이미지를 임시 파일로 저장
-                temp_path = save_uploaded_file(image)
-                if temp_path is None:
-                    return [], None, [], None, "", "", "이미지 처리에 실패했습니다."
-                
+                temp_path = None
                 try:
+                    # PIL Image를 임시 파일로 저장
+                    if isinstance(image, Image.Image):
+                        temp_path = tempfile.mktemp(suffix='.png')
+                        image.save(temp_path, format='PNG')
+                        logger.info(f"Saved PIL image to temporary file: {temp_path}")
+                    else:
+                        temp_path = save_uploaded_file(image)
+                    
+                    if temp_path is None:
+                        return [], None, [], None, "", "", "이미지 처리에 실패했습니다."
+                    
+                    # API 호출 전 파라미터 로깅 (민감한 정보 제외)
+                    logger.info(f"Calling API with bg_type: {bg_type}, quality_level: {quality_level}")
+                    
                     # API 호출
                     result = api_check_password(
                         password, temp_path, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, 
                         request_text, quality_level, aspect_ratio, output_format, enable_enhancement
                     )
                     return result
+                    
+                except Exception as e:
+                    logger.error(f"Error in check_password wrapper: {e}")
+                    logger.error(f"Full error: {traceback.format_exc()}")
+                    return [], None, [], None, "", "", f"처리 중 오류 발생: {str(e)}"
+                    
                 finally:
                     # 임시 파일 정리
-                    if os.path.exists(temp_path):
+                    if temp_path and os.path.exists(temp_path):
                         try:
                             os.remove(temp_path)
-                        except:
-                            pass
+                            logger.info(f"Cleaned up temporary file: {temp_path}")
+                        except Exception as cleanup_error:
+                            logger.warning(f"Failed to cleanup temp file {temp_path}: {cleanup_error}")
             
             # 프롬프트 생성 버튼 클릭 이벤트 (원본과 동일한 API 이름)
             generate_prompt_btn.click(