Update app.py

app.py

@@ -299,14 +299,14 @@ def create_gradio_interface():
                     info = gr.Textbox(label="처리 정보", interactive=False)
                     error = gr.Textbox(label="오류 메시지", interactive=False, visible=True)
             
-            # 프롬프트만 생성하는 함수 (API 호출)
-            def generate_prompt_wrapper(password, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, request_text, aspect_ratio):
+            # 프롬프트만 생성하는 함수 (원본과 동일한 엔드포인트)
+            def generate_prompt_with_password_check(password, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, request_text, aspect_ratio):
                 return api_generate_prompt_with_password_check(
                     password, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, request_text, aspect_ratio
                 )
             
-            # 이미지 처리 함수 (API 호출)
-            def process_image_wrapper(password, image, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, request_text, quality_level, aspect_ratio, output_format, enable_enhancement):
+            # 이미지 처리 함수 (원본과 동일한 엔드포인트)
+            def check_password(password, image, bg_type, simple, studio, nature, indoor, special, jewelry, special_effects, request_text, quality_level, aspect_ratio, output_format, enable_enhancement):
                 if image is None:
                     return [], None, [], None, "", "", "이미지를 업로드해야 합니다."
                 
@@ -330,9 +330,9 @@ def create_gradio_interface():
                         except:
                             pass
             
-            # 프롬프트 생성 버튼 클릭 이벤트
+            # 프롬프트 생성 버튼 클릭 이벤트 (원본과 동일한 API 이름)
             generate_prompt_btn.click(
-                fn=generate_prompt_wrapper,
+                fn=generate_prompt_with_password_check,
                 inputs=[
                     password_box,
                     background_type,
@@ -340,12 +340,13 @@ def create_gradio_interface():
                     jewelry_dropdown, special_effects_dropdown,
                     request_text, aspect_ratio
                 ],
-                outputs=[prompt_output]
+                outputs=[prompt_output],
+                api_name="generate_prompt_with_password_check"  # 원본과 동일한 API 이름
             )
             
-            # 편집 버튼 클릭 이벤트
+            # 편집 버튼 클릭 이벤트 (원본과 동일한 API 이름)
             edit_btn.click(
-                fn=process_image_wrapper,
+                fn=check_password,
                 inputs=[
                     password_box, 
                     image, background_type, 
@@ -357,7 +358,8 @@ def create_gradio_interface():
                     original_output, original_download, 
                     enhanced_output, enhanced_download,
                     prompt_output, info, error
-                ]
+                ],
+                api_name="check_password"  # 원본과 동일한 API 이름
             )
         
         logger.info("Gradio interface created successfully")