Update app.py

app.py

@@ -14,7 +14,6 @@ from transformers import pipeline  # 새로 추가된 import
 load_dotenv()
 
 
-
 # 기존 코드 유지 (로깅, 함수 등 모든 기능 코드)
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -1187,14 +1186,14 @@ with gr.Blocks(css=custom_css) as demo:
                         bg_remove_status = gr.Textbox(label="상태", lines=2, elem_classes="custom-input")
                         bg_remove_btn = gr.Button('✂️ 배경 제거하기', elem_classes="custom-button primary")
         
-        # 이미지 필터 탭
+        # 이미지 필터 탭 UI 개선
         with gr.TabItem("🖌️ 이미지 필터", elem_classes="tab-content"):
             with gr.Row(equal_height=True):
                 # 왼쪽 열: 비율 3
                 with gr.Column(scale=3, elem_classes="panel"):
                     gr.HTML('<div class="section-title"><i class="fas fa-sliders-h"></i> 이미지 필터 설정</div>')
                     # 이미지 업로드
-                    filter_input_image = gr.Image(type="pil", label="이미지 업로드", elem_classes="image-container")
+                    filter_input_image = gr.Image(type="pil", label="이미지 업로드", elem_classes="image-container", height=400)
                 
                     # 조정 슬라이더
                     with gr.Group():
@@ -1211,14 +1210,13 @@ with gr.Blocks(css=custom_css) as demo:
                 # 오른쪽 열: 비율 7
                 with gr.Column(scale=7, elem_classes="panel"):
                     gr.HTML('<div class="section-title"><i class="fas fa-image"></i> 필터 적용 결과</div>')
-                    # 처리된 이미지 출력
-                    filter_output_image = gr.Image(type="pil", label="처리된 이미지", elem_classes="image-container")
+                    # 처리된 이미지 출력 - 높이를 입력 이미지와 동일하게 설정
+                    filter_output_image = gr.Image(type="pil", label="처리된 이미지", elem_classes="image-container", height=400)
                 
-                    # 변환된 이미지 다운로드 버튼
-                    with gr.Row():
-                        gr.HTML('<div class="section-title"><i class="fas fa-download"></i> 이미지 저장</div>')
-                        download_button = gr.Button("JPG로 변환하기", elem_classes="custom-button primary")
-                        download_output = gr.File(label="JPG 이미지 다운로드")
+                    # 다운로드 버튼 - "결과이미지저장" 섹션으로 변경하고 파일 다운로드를 그 아래에 배치
+                    gr.HTML('<div class="section-title"><i class="fas fa-download"></i> 결과이미지저장</div>')
+                    download_button = gr.Button("결과이미지저장", elem_classes="custom-button primary")
+                    download_output = gr.File(label="JPG 이미지 다운로드")
             
             # 이미지 처리 함수 연결
             inputs = [