.

app.py

@@ -1,9 +1,8 @@
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import FileResponse
 import cv2
-import numpy as np
 from PIL import Image
-import io
+import re
 import gdown
 import os
 import uuid
@@ -21,11 +20,40 @@ app = FastAPI(
     ],
 )
 
+
+def extract_file_id(drive_url: str) -> str:
+    """
+    Google Drive URL에서 파일 ID를 추출합니다.
+
+    Parameters:
+    drive_url (str): Google Drive 파일 URL.
+
+    Returns:
+    str: 추출된 파일 ID.
+    """
+    # 정규 표현식을 사용하여 URL에서 파일 ID 추출
+    match = re.search(r'/d/([a-zA-Z0-9_-]+)', drive_url)
+    if match:
+        return match.group(1)
+    
+    # 다른 URL 형식에서 파일 ID 추출
+    match = re.search(r'file/d/([a-zA-Z0-9_-]+)', drive_url)
+    if match:
+        return match.group(1)
+    
+    # 공유 링크에서 파일 ID 추출
+    match = re.search(r'([a-zA-Z0-9_-]{33,})', drive_url)
+    if match:
+        return match.group(1)
+
+    raise ValueError("Invalid Google Drive URL")
+
+
 @app.post("/detect-edges/")
 def detect_edges(url: str, request: Request):
     try:
         # Google Drive 파일 ID 추출
-        file_id = url.split('/')[-2]
+        file_id = extract_file_id(url)
         download_url = f"https://drive.google.com/uc?id={file_id}"
         
         # 이미지를 Google Drive에서 다운로드