Create app.py

app.py

@@ -0,0 +1,177 @@
+import gradio as gr
+import replicate
+import os
+import requests
+import tempfile
+from pathlib import Path
+import logging
+import base64
+
+# 로깅 설정
+logging.basicConfig(
+    level=logging.DEBUG,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+# API 토큰 설정 (Hugging Face Space에서는 SECRET으로 설정해야 함)
+# REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
+
+def remove_background(input_image):
+    """
+    이미지의 배경을 제거하는 함수
+    """
+    try:
+        logger.info("=== 배경 제거 시작 ===")
+        
+        # Replicate API 토큰 확인
+        api_token = os.getenv("REPLICATE_API_TOKEN")
+        if not api_token:
+            logger.error("REPLICATE_API_TOKEN이 설정되지 않았습니다.")
+            return None, "Error: REPLICATE_API_TOKEN이 설정되지 않았습니다."
+        
+        logger.info(f"API 토큰 확인 완료: {api_token[:5]}...")
+        
+        # 입력 이미지가 유효한지 확인
+        if input_image is None:
+            logger.error("입력 이미지가 없습니다.")
+            return None, "이미지를 업로드해주세요."
+        
+        logger.info(f"입력 이미지 경로: {input_image}")
+        logger.info(f"입력 이미지 타입: {type(input_image)}")
+        
+        # 파일을 열어서 URI 형식으로 변환
+        try:
+            with open(input_image, "rb") as image_file:
+                # Data URI 형식으로 변환
+                image_data = base64.b64encode(image_file.read()).decode('utf-8')
+                image_uri = f"data:image/png;base64,{image_data}"
+                
+            logger.info("이미지를 Data URI로 변환 완료")
+            logger.info(f"Data URI 길이: {len(image_uri)}")
+            
+            # Replicate 모델 실행
+            logger.info("Replicate 모델 실행 시작")
+            
+            output = replicate.run(
+                "851-labs/background-remover:a029dff38972b5fda4ec5d75d7d1cd25aeff621d2cf4946a41055d7db66b80bc",
+                input={"image": image_uri}
+            )
+            
+            logger.info("Replicate 모델 실행 완료")
+            logger.info(f"결과 타입: {type(output)}")
+            
+            # 임시 파일로 결과 저장
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
+                tmp_file.write(output.read())
+                tmp_file_path = tmp_file.name
+            
+            logger.info(f"결과 이미지 저장 완료: {tmp_file_path}")
+            return tmp_file_path, "배경 제거가 완료되었습니다!"
+        
+        except Exception as file_error:
+            logger.error(f"파일 처리 오류: {str(file_error)}")
+            raise file_error
+    
+    except replicate.exceptions.ReplicateError as re:
+        logger.error(f"Replicate API 오류: {str(re)}")
+        logger.error(f"오류 상세: {re.__dict__}")
+        return None, f"Replicate API 오류: {str(re)}"
+    
+    except Exception as e:
+        logger.error(f"예상치 못한 오류: {str(e)}")
+        logger.error(f"오류 타입: {type(e)}")
+        logger.error(f"오류 상세: {e.__dict__ if hasattr(e, '__dict__') else 'no details'}")
+        return None, f"오류 발생: {str(e)}"
+
+# Gradio 인터페이스 생성
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown(
+        """
+        # 🖼️ AI 배경 제거기
+        
+        이 앱은 Replicate의 851-labs/background-remover 모델을 사용하여 이미지의 배경을 제거합니다.
+        
+        ### 사용 방법:
+        1. 이미지를 업로드하세요
+        2. '배경 제거' 버튼을 클릭하세요
+        3. 결과 이미지를 다운로드하세요
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(
+                type="filepath",
+                label="원본 이미지",
+                interactive=True
+            )
+            remove_btn = gr.Button("배경 제거", variant="primary")
+        
+        with gr.Column():
+            output_image = gr.Image(
+                type="filepath",
+                label="결과 이미지",
+                interactive=False
+            )
+            status_text = gr.Textbox(
+                label="상태",
+                interactive=False,
+                show_label=True,
+                lines=3
+            )
+    
+    # 디버그 정보를 위한 추가 UI
+    with gr.Accordion("디버그 정보", open=False):
+        debug_output = gr.Textbox(
+            label="디버그 로그",
+            interactive=False,
+            lines=10
+        )
+    
+    # 이벤트 핸들러
+    def process_and_log(image):
+        # 로그 캡처를 위한 핸들러
+        log_capture = []
+        
+        class ListHandler(logging.Handler):
+            def emit(self, record):
+                log_capture.append(self.format(record))
+        
+        list_handler = ListHandler()
+        list_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
+        logger.addHandler(list_handler)
+        
+        try:
+            result_image, status = remove_background(image)
+            debug_log = "\n".join(log_capture)
+            return result_image, status, debug_log
+        finally:
+            logger.removeHandler(list_handler)
+    
+    remove_btn.click(
+        fn=process_and_log,
+        inputs=[input_image],
+        outputs=[output_image, status_text, debug_output]
+    )
+    
+    gr.Markdown(
+        """
+        ---
+        ### ⚙️ 기술 정보
+        - 모델: 851-labs/background-remover
+        - API: Replicate
+        - Framework: Gradio 5.0+
+        
+        ### 🔒 보안 참고사항
+        이 앱을 사용하려면 Replicate API 토큰이 필요합니다. Hugging Face Space의 Settings에서 
+        REPLICATE_API_TOKEN을 Secret으로 설정해주세요.
+        
+        ### 🐛 디버깅
+        문제가 발생하면 '디버그 정보' 섹션을 펼쳐서 상세 로그를 확인하세요.
+        """
+    )
+
+# Gradio 앱 실행
+if __name__ == "__main__":
+    demo.launch()