Update app.py

app.py

@@ -146,28 +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:
+    with gr.Blocks(title="Clarity Upscaler - 이미지 화질 개선") as demo:
+        gr.Markdown("# Clarity Upscaler - 이미지 화질 개선 도구")
+        gr.Markdown("Replicate의 Clarity Upscaler 모델을 사용하여 이미지의 화질을 개선합니다.")
+        
         with gr.Row():
             with gr.Column():
-                # 1) 비밀번호 입력
-                password_input = gr.Textbox(
-                    label="비밀번호",
-                    type="password",
-                    placeholder="비밀번호를 입력하세요"
-                )
-
-                # 2) 업스케일할 이미지 입력
                 input_image = gr.Image(label="원본 이미지", type="pil")
-
-                # 3) 고급 설정 아코디언 안에 파라미터들 정의
+                
                 with gr.Accordion("고급 설정", open=False):
-                    scale_factor = gr.Slider(
-                        minimum=1, maximum=4, value=2, step=0.5, label="확대 비율"
-                    )
-                    output_format = gr.Radio(
-                        choices=["jpg", "png"], value="jpg", label="출력 형식"
-                    )
+                    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]",
@@ -178,66 +169,52 @@ 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
-                    )
-
-                # 4) 실행 버튼
-                submit_btn = gr.Button("실행", variant="primary")
-
+                    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")
+                
             with gr.Column():
-                # 5) 결과 이미지 표시
                 output_image = gr.Image(label="개선된 이미지")
-
-        # 6) 버튼 클릭 시 호출할 함수
-        def process_image(
-            password, image, scale_factor, output_format, sd_model,
-            resemblance, creativity, dynamic, sharpen, seed
-        ):
-            # 비밀번호 검증
-            if password != "1089":
-                raise ValueError("잘못된 비밀번호입니다.")
+        
+        # 로그 캡처 기능 제거 (UI에서 로그 표시하지 않음)
+        
+        # 제출 버튼 클릭 시 동작
+        def process_image(image, scale_factor, output_format, sd_model, resemblance, creativity, dynamic, sharpen, seed):
             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)
-            )
-
-        # 7) click 이벤트에 모든 입력 위젯을 순서대로 전달
+            
+            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)}")
+                return None
+        
         submit_btn.click(
             process_image,
             inputs=[
-                password_input,
-                input_image, scale_factor, output_format, sd_model,
+                input_image, scale_factor, output_format, sd_model, 
                 resemblance, creativity, dynamic, sharpen, seed
             ],
             outputs=[output_image]
         )
-
+        
     return demo
 
-
 # 메인 실행
 if __name__ == "__main__":
     demo = create_interface()