Update app.py

app.py

@@ -2,18 +2,7 @@ import gradio as gr
 import cv2
 import numpy as np
 
-# 기존 기능: 이미지를 흑백으로 변환
-def convert_to_grayscale(image):
-    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    return gray_image
-
-def convert_and_save(image):
-    gray_image = convert_to_grayscale(image)
-    output_path = "output.jpg"
-    cv2.imwrite(output_path, gray_image)
-    return gray_image, output_path
-
-# 로그인 검증 함수
+# 아이디/비밀번호 검증 함수
 def verify_credentials(username, password):
     if username == "kim" and password == "1234":
         return True, "성공적으로 로그인되었습니다."
@@ -25,17 +14,31 @@ def login(username, password):
     success, message = verify_credentials(username, password)
     if success:
         # 로그인 성공 시 로그인 섹션 숨기고 메인 앱 표시
-        return gr.update(visible=False), gr.update(visible=True), gr.update(value=message, visible=True)
+        return gr.update(visible=False), gr.update(visible=True), gr.update(value=message)
     else:
-        # 로그인 실패 시 메시지 표시
-        return gr.update(visible=True), gr.update(visible=False), gr.update(value=message, visible=True)
+        # 로그인 실패 시 오류 메시지 표시
+        return gr.update(visible=True), gr.update(visible=False), gr.update(value=message)
+
+# 이미지 흑백 변환 함수
+def convert_to_grayscale(image):
+    # 이미지를 흑백으로 변환
+    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    return gray_image
+
+# 이미지 변환 및 저장 함수
+def convert_and_save(image):
+    # 이미지를 흑백으로 변환하고, 다운로드 가능한 JPG 파일로 저장
+    gray_image = convert_to_grayscale(image)
+    output_path = "output.jpg"
+    cv2.imwrite(output_path, gray_image)
+    return gray_image, output_path
 
-# Gradio UI 구성
+# Gradio 인터페이스 정의
 with gr.Blocks() as demo:
     # 로그인 섹션
-    with gr.Row(visible=True) as login_row:
+    with gr.Row() as login_row:
         with gr.Column():
-            gr.Markdown("# 로그인")
+            gr.Markdown("## 로그인")
             username = gr.Textbox(label="아이디")
             password = gr.Textbox(label="비밀번호", type="password")
             login_button = gr.Button("로그인")
@@ -43,25 +46,12 @@ with gr.Blocks() as demo:
     
     # 메인 애플리케이션 섹션 (초기에는 숨김)
     with gr.Row(visible=False) as main_app:
-        with gr.Row():
-            # 왼쪽: 입력 항목
-            with gr.Column(scale=1):
-                gr.Markdown("### 이미지 흑백 변환기")
-                gr.Markdown("이미지를 업로드하면 흑백으로 변환하고, JPG 파일로 다운로드할 수 있습니다.")
-                image_input = gr.Image(type="numpy", label="이미지 업로드")
-                convert_button = gr.Button("변환 및 저장")
-            
-            # 오른쪽: 출력 항목
-            with gr.Column(scale=1):
-                output_image = gr.Image(label="흑백 이미지")
-                download_file = gr.File(label="다운로드 JPG 파일")
-        
-        # 변환 버튼 클릭 시 함수 연결
-        convert_button.click(
-            convert_and_save,
-            inputs=image_input,
-            outputs=[output_image, download_file]
-        )
+        with gr.Column():
+            image_input = gr.Image(type="numpy", label="이미지 업로드")
+            convert_button = gr.Button("흑백 변환")
+        with gr.Column():
+            grayscale_image = gr.Image(label="흑백 이미지")
+            download_file = gr.File(label="변환된 이미지 다운로드")
     
     # 로그인 버튼 이벤트 연결
     login_button.click(
@@ -69,6 +59,13 @@ with gr.Blocks() as demo:
         inputs=[username, password],
         outputs=[login_row, main_app, login_message]
     )
+    
+    # 이미지 변환 버튼 이벤트 연결
+    convert_button.click(
+        convert_and_save,
+        inputs=image_input,
+        outputs=[grayscale_image, download_file]
+    )
 
 if __name__ == "__main__":
     demo.launch()