Update app.py

app.py

@@ -9,10 +9,8 @@ logging.basicConfig(level=logging.INFO)
 def remove_background(input_image):
     """
     입력된 이미지에서 배경을 제거하는 함수
-
     Parameters:
     input_image (PIL.Image 또는 str): 업로드된 이미지 파일의 객체 또는 파일 경로
-
     Returns:
     PIL.Image 또는 str: 배경이 제거된 이미지 또는 에러 메시지
     """
@@ -34,6 +32,10 @@ def remove_background(input_image):
         logging.error(f"Error removing background: {e}")
         return f"에러가 발생했습니다: {str(e)}"
 
+def clear_images():
+    """입력 이미지와 출력 이미지를 모두 초기화하는 함수"""
+    return None, None
+
 # Gradio 인터페이스 구성
 with gr.Blocks() as demo:
     gr.Markdown("# 이미지 배경 제거기")
@@ -52,8 +54,13 @@ with gr.Blocks() as demo:
         )
     
     # 배경 제거 버튼
-    btn = gr.Button("배경 제거")
-    btn.click(fn=remove_background, inputs=input_image, outputs=output_image)
+    with gr.Row():
+        btn_remove = gr.Button("배경 제거")
+        btn_remove.click(fn=remove_background, inputs=input_image, outputs=output_image)
+        
+        # 초기화 버튼 추가
+        btn_reset = gr.Button("초기화")
+        btn_reset.click(fn=clear_images, inputs=None, outputs=[input_image, output_image])
     
     # 예제 이미지
     gr.Examples(
@@ -68,4 +75,4 @@ with gr.Blocks() as demo:
     # gr.Markdown("© 2024 이미지 배경 제거기")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()