Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import cv2
+import numpy as np
 import os
 
 def convert_to_grayscale(image):
@@ -14,24 +15,40 @@ def convert_and_save(image):
     cv2.imwrite(output_path, gray_image)
     return gray_image, output_path
 
-# 예시 이미지 경로 설정
-example_files = ["1.png", "2.png", "3.png"]  # 미리 준비된 파일 이름
-example_paths = [os.path.abspath(file) for file in example_files]
+# 예시 이미지 파일 이름
+example_files = ["1.png", "2.png", "3.png"]
+
+# 기본 예시 이미지 생성 함수
+def create_example_image(file_path, text):
+    # 기본 흰색 이미지에 텍스트 추가
+    image = np.ones((256, 256, 3), dtype=np.uint8) * 255  # 흰색 배경
+    cv2.putText(
+        image,
+        text,
+        (50, 130),
+        cv2.FONT_HERSHEY_SIMPLEX,
+        1,
+        (0, 0, 0),
+        2,
+        cv2.LINE_AA
+    )
+    cv2.imwrite(file_path, image)
 
-# 예시 파일 존재 여부 확인
-missing_files = [file for file in example_files if not os.path.exists(file)]
-if missing_files:
-    print(f"다음 파일이 실행 디렉토리에 없습니다: {', '.join(missing_files)}")
-    raise FileNotFoundError("필수 예시 파일이 없습니다. 확인 후 다시 실행하세요.")
+# 예시 이미지 준비
+for i, file_name in enumerate(example_files, start=1):
+    if not os.path.exists(file_name):
+        print(f"{file_name} 파일이 존재하지 않습니다. 기본 이미지를 생성합니다.")
+        create_example_image(file_name, f"Example {i}")
 
 # Gradio 인터페이스 정의
+example_paths = [os.path.abspath(file) for file in example_files]
 iface = gr.Interface(
     fn=convert_and_save,
     inputs="image",
     outputs=["image", "file"],
     title="이미지 흑백 변환기",
     description="아래 예시 파일을 클릭하거나 직접 이미지를 업로드하세요.",
-    examples=example_paths  # 업로드된 파일을 예시로 사용
+    examples=example_paths  # 예시 이미지 경로 전달
 )
 
 if __name__ == "__main__":