Update app.py

app.py

@@ -146,18 +146,19 @@ def upscale_image(
         if os.path.exists(temp_input_path):
             os.remove(temp_input_path)
 
-# Gradio 인터페이스 생성
 def create_interface():
-    with gr.Blocks(title="Clarity Upscaler - 이미지 화질 개선") as demo:
-        gr.Markdown("# Clarity Upscaler - 이미지 화질 개선 도구")
-        gr.Markdown("Replicate의 Clarity Upscaler 모델을 사용하여 이미지의 화질을 개선합니다.")
-        
+    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="확대 비율")
+                    scale_factor = gr.Slider(1, 4, value=2, step=0.5, label="확대 비율")
                     output_format = gr.Radio(["jpg", "png"], value="jpg", label="출력 형식")
                     sd_model = gr.Dropdown(
                         choices=[
@@ -169,53 +170,61 @@ def create_interface():
                         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)
-                
-                submit_btn = gr.Button("화질 개선 시작", variant="primary")
+                    resemblance = gr.Slider(0.0, 1.0, value=0.6, step=0.05, label="원본 유사도")
+                    creativity  = gr.Slider(0.0, 1.0, value=0.35, step=0.05, label="창의성")
+                    dynamic     = gr.Slider(1, 10, value=6, step=1, label="다이나믹 임계값")
+                    sharpen     = gr.Slider(0, 2, value=0, step=0.1, label="선명도")
+                    seed        = gr.Number(value=1337, label="시드", precision=0)
                 
+                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
-            
-            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)
+                download_btn  = gr.DownloadButton(
+                    label="다운로드", 
+                    file_name="upscaled.jpg", 
+                    mime="image/jpeg"
                 )
-                return result
-            except Exception as e:
-                logger.error(f"이미지 처리 중 오류 발생: {str(e)}")
+
+        def process_image(
+            password, image, scale_factor, output_format, sd_model,
+            resemblance, creativity, dynamic, sharpen, seed
+        ):
+            if password != "1089":
+                raise ValueError("잘못된 비밀번호입니다.")
+            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)
+            )
+
+        # 1) 실행 버튼 → output_image
         submit_btn.click(
-            process_image,
+            fn=process_image,
             inputs=[
-                input_image, scale_factor, output_format, sd_model, 
+                password_input, input_image, scale_factor, output_format, sd_model,
                 resemblance, creativity, dynamic, sharpen, seed
             ],
             outputs=[output_image]
         )
-        
+        # 2) 다운로드 버튼 → output_image 그대로 받아서 파일로 제공
+        download_btn.click(
+            fn=lambda img: img,
+            inputs=[output_image],
+            outputs=[download_btn]
+        )
+
     return demo
 
-# 메인 실행
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch(share=True)