Update app.py

app.py

@@ -7,7 +7,10 @@ def convert_to_grayscale(image):
     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     return gray_image
 
-def apply_filter(image, filter_type):
+def apply_filter(image, filter_type, grayscale):
+    if grayscale:
+        image = convert_to_grayscale(image)
+    
     if filter_type == "Soft Glow":
         # Soft Glow 효과
         gaussian = cv2.GaussianBlur(image, (15, 15), 0)
@@ -41,23 +44,25 @@ def apply_filter(image, filter_type):
         # 필터를 적용하지 않음
         return image
 
-def convert_and_save(image, filter_type):
-    # 선택한 필터를 이미지에 적용
-    filtered_image = apply_filter(image, filter_type)
-    # 이미지를 흑백으로 변환
-    gray_image = convert_to_grayscale(filtered_image)
+def convert_and_save(image, filter_type, grayscale):
+    # 선택한 필터와 흑백 변환 여부를 이미지에 적용
+    filtered_image = apply_filter(image, filter_type, grayscale)
     # 이미지를 저장
     output_path = "output.jpg"
-    cv2.imwrite(output_path, gray_image)
-    return gray_image, output_path
+    cv2.imwrite(output_path, filtered_image)
+    return filtered_image, output_path
 
 # Gradio 인터페이스 정의
 iface = gr.Interface(
     fn=convert_and_save, 
-    inputs=["image", gr.Radio(["Soft Glow", "Portrait Enhancer", "Warm Tone", "Cold Tone", "High-Key", "Low-Key", "Haze"], label="필터 선택")], 
+    inputs=[
+        "image", 
+        gr.Radio(["None", "Soft Glow", "Portrait Enhancer", "Warm Tone", "Cold Tone", "High-Key", "Low-Key", "Haze"], label="필터 선택"),
+        gr.Checkbox(label="흑백 변환")
+    ], 
     outputs=["image", "file"], 
-    title="이미지 흑백 변환기",
-    description="이미지를 업로드하고 필터를 선택하면, 흑백으로 변환하고 JPG 파일로 다운로드할 수 있습니다."
+    title="이미지 필터 및 흑백 변환기",
+    description="이미지를 업로드하고 필터와 흑백 변환 여부를 선택하면, 변환된 이미지를 JPG 파일로 다운로드할 수 있습니다."
 )
 
 iface.launch()