.

app.py

@@ -1,10 +1,11 @@
 from fastapi import FastAPI, HTTPException
 import cv2
 import numpy as np
-from fastapi.responses import StreamingResponse
+from fastapi.responses import FileResponse
 import io
 import gdown
 import os
+import uuid
 
 app = FastAPI(
     version="0.0.1",
@@ -19,18 +20,16 @@ app = FastAPI(
 @app.post("/detect-edges/")
 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)
+        temp_input_image = f'/tmp/{uuid.uuid4()}.png'
+        gdown.download(download_url, temp_input_image, quiet=False)
 
         # 이미지를 읽어들임
-        img = cv2.imread(output)
+        img = cv2.imread(temp_input_image)
 
         # 이미지가 제대로 읽혔는지 확인
         if img is None:
@@ -39,15 +38,24 @@ def detect_edges(url: str):
         # 엣지 디텍션 수행 (Canny 알고리즘 사용)
         edges = cv2.Canny(img, 100, 200)
 
-        # 엣지 이미지를 다시 인코딩
-        _, buffer = cv2.imencode('.png', edges)
-        io_buf = io.BytesIO(buffer)
+        # 엣지 이미지를 파일로 저장
+        output_file_path = f'/tmp/{uuid.uuid4()}.png'
+        cv2.imwrite(output_file_path, edges)
 
-        # 임시 파일 삭제
-        os.remove(output)
+        # 입력 이미지 임시 파일 삭제
+        os.remove(temp_input_image)
 
-        # 결과 이미지를 사용자에게 전송
-        return StreamingResponse(io_buf, media_type="image/png")
+        # 클라이언트에게 결과 이미지 파일 경로를 반환
+        return {"download_url": f"/download/{os.path.basename(output_file_path)}"}
     
     except Exception as e:
-        raise HTTPException(status_code=400, detail=str(e))
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@app.get("/download/{filename}")
+def download_file(filename: str):
+    file_path = f"/tmp/{filename}"
+    if os.path.exists(file_path):
+        return FileResponse(file_path, media_type='image/png', filename=filename)
+    else:
+        raise HTTPException(status_code=404, detail="File not found")