.

app.py

@@ -1,8 +1,10 @@
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, HTTPException
 import cv2
 import numpy as np
 from fastapi.responses import StreamingResponse
 import io
+import gdown
+import os
 
 app = FastAPI(
     version="0.0.1",
@@ -15,20 +17,37 @@ app = FastAPI(
 )
 
 @app.post("/detect-edges/")
-async def detect_edges(file: UploadFile = File(...)):
-    # 이미지를 바이트 배열로 읽기
-    image_bytes = await file.read()
-    # 바이트 배열을 numpy 배열로 변환
-    nparr = np.frombuffer(image_bytes, np.uint8)
-    # numpy 배열을 이미지로 디코딩
-    img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-
-    # 엣지 디텍션 수행 (Canny 알고리즘 사용)
-    edges = cv2.Canny(img, 100, 200)
-
-    # 엣지 이미지를 다시 인코딩
-    _, buffer = cv2.imencode('.png', edges)
-    io_buf = io.BytesIO(buffer)
-
-    # 결과 이미지를 사용자에게 전송
-    return StreamingResponse(io_buf, media_type="image/png")
+def detect_edges(url: str):
+    # Google Drive 파일 ID 추출
+    print('got ', str)
+    print('###############')
+    try:
+        file_id = url.split('/')[-2]
+        download_url = f"https://drive.google.com/uc?id={file_id}"
+        
+        # 이미지를 Google Drive에서 다운로드
+        output = 'temp_image.png'
+        gdown.download(download_url, output, quiet=False)
+
+        # 이미지를 읽어들임
+        img = cv2.imread(output)
+
+        # 이미지가 제대로 읽혔는지 확인
+        if img is None:
+            raise HTTPException(status_code=400, detail="Invalid image URL or file not found.")
+
+        # 엣지 디텍션 수행 (Canny 알고리즘 사용)
+        edges = cv2.Canny(img, 100, 200)
+
+        # 엣지 이미지를 다시 인코딩
+        _, buffer = cv2.imencode('.png', edges)
+        io_buf = io.BytesIO(buffer)
+
+        # 임시 파일 삭제
+        os.remove(output)
+
+        # 결과 이미지를 사용자에게 전송
+        return StreamingResponse(io_buf, media_type="image/png")
+    
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))

requirements.txt

@@ -5,4 +5,5 @@ opencv-python-headless  # OpenCV, 이미지 처리 라이브러리, GUI 기능
 Pillow  # Python Imaging Library (PIL), 이미지 작업을 위한 라이브러리
 
 numpy  # NumPy, OpenCV 및 일반 수치 계산에 필요
-python-multipart
+python-multipart
+gdown