Update app.py

app.py

@@ -146,73 +146,65 @@ def upscale_image(
         if os.path.exists(temp_input_path):
             os.remove(temp_input_path)
 
-# Gradio 인터페이스 생성
+# 1) Gradio 인터페이스 생성부에서 변경된 부분만 발췌
 def create_interface():
-    with gr.Blocks(title="Test") as demo:
-        
+    with gr.Blocks(title="Clarity Upscaler") as demo:
         with gr.Row():
             with gr.Column():
+                # 비밀번호 입력 필드 추가
+                password_input = gr.Textbox(
+                    label="비밀번호", 
+                    type="password", 
+                    placeholder="비밀번호를 입력하세요"
+                )
+
                 input_image = gr.Image(label="원본 이미지", type="pil")
-                
-                with gr.Accordion("고급 설정", open=False):
-                    scale_factor = gr.Slider(minimum=1, maximum=4, value=2, step=0.5, label="확대 비율")
-                    output_format = gr.Radio(["jpg", "png"], value="jpg", label="출력 형식")
-                    sd_model = gr.Dropdown(
-                        choices=[
-                            "juggernaut_reborn.safetensors [338b85bc4f]",
-                            "sd_xl_base_1.0.safetensors [39d4e625d]",
-                            "sdxl_base_v0.9-9961.safetensors [3048045c]",
-                            "realisticVisionV51_v51VAE.safetensors [5ecbfa0ac]",
-                        ],
-                        value="juggernaut_reborn.safetensors [338b85bc4f]",
-                        label="SD 모델"
-                    )
-                    resemblance = gr.Slider(minimum=0.0, maximum=1.0, value=0.6, step=0.05, label="원본 유사도")
-                    creativity = gr.Slider(minimum=0.0, maximum=1.0, value=0.35, step=0.05, label="창의성")
-                    dynamic = gr.Slider(minimum=1, maximum=10, value=6, step=1, label="다이나믹 임계값")
-                    sharpen = gr.Slider(minimum=0, maximum=2, value=0, step=0.1, label="선명도")
-                    seed = gr.Number(value=1337, label="시드", precision=0)
-                
+                # … (기존 고급 설정 Accordion)
+
                 submit_btn = gr.Button("실행", variant="primary")
-                
+
             with gr.Column():
                 output_image = gr.Image(label="개선된 이미지")
-        
-        # 로그 캡처 기능 제거 (UI에서 로그 표시하지 않음)
-        
-        # 제출 버튼 클릭 시 동작
-        def process_image(image, scale_factor, output_format, sd_model, resemblance, creativity, dynamic, sharpen, seed):
-            if image is None:
-                return None
+
+        # 2) process_image 함수 시그니처 수정
+        def process_image(
+            password, image, scale_factor, output_format, sd_model, 
+            resemblance, creativity, dynamic, sharpen, seed
+        ):
+            # 비밀번호 검증
+            if password != "1089":
+                raise ValueError("잘못된 비밀번호입니다.")
             
-            try:
-                result = upscale_image(
-                    image=image,
-                    scale_factor=scale_factor,
-                    output_format=output_format,
-                    sd_model=sd_model,
-                    resemblance=resemblance,
-                    creativity=creativity,
-                    dynamic=dynamic,
-                    sharpen=sharpen,
-                    seed=int(seed)
-                )
-                return result
-            except Exception as e:
-                logger.error(f"이미지 처리 중 오류 발생: {str(e)}")
+            if image is None:
                 return None
-        
+
+            # 기존 로직 호출
+            return upscale_image(
+                image=image,
+                scale_factor=scale_factor,
+                output_format=output_format,
+                sd_model=sd_model,
+                resemblance=resemblance,
+                creativity=creativity,
+                dynamic=dynamic,
+                sharpen=sharpen,
+                seed=int(seed)
+            )
+
+        # 3) 클릭 이벤트에 password_input 추가
         submit_btn.click(
             process_image,
             inputs=[
+                password_input,  # 맨 앞에 비밀번호
                 input_image, scale_factor, output_format, sd_model, 
                 resemblance, creativity, dynamic, sharpen, seed
             ],
             outputs=[output_image]
         )
-        
+
     return demo
 
+
 # 메인 실행
 if __name__ == "__main__":
     demo = create_interface()