Update app.py

app.py

@@ -7,6 +7,7 @@ import logging
 from typing import Optional
 from fastapi.staticfiles import StaticFiles
 from dotenv import load_dotenv
+from pydantic import BaseModel
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -30,10 +31,18 @@ BASE_URL = "https://tecuts-depot.hf.space"
 ARL_TOKEN = os.getenv('ARL')
 dl = DeeLogin(arl=ARL_TOKEN)
 
+
+# 定义请求体模型
+class DownloadRequest(BaseModel):
+    url: str
+    quality: str
+
+
 @app.get("/")
 def read_root():
     return {"message": "running"}
 
+
 # Helper function to get track info
 def get_track_info(track_id: str):
     try:
@@ -48,15 +57,26 @@ def get_track_info(track_id: str):
         logger.error(f"Error fetching track metadata: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
+
 # Fetch track metadata from Deezer API
 @app.get("/track/{track_id}")
 def get_track(track_id: str):
     return get_track_info(track_id)
 
+
 # Download a track and return a download URL
-@app.post("/download/track/{track_id}")
-def download_track(track_id: str, quality: str = "MP3_320"):
+@app.post("/download/track")
+def download_track(request: DownloadRequest):
     try:
+        url = request.url
+        quality = request.quality
+
+        if quality not in ["MP3_320", "MP3_128", "FLAC"]:
+            raise HTTPException(status_code=400, detail="Invalid quality specified")
+
+        # 提取 track_id （假设 url 格式为 https://api.deezer.com/track/{track_id}）
+        track_id = url.split("/")[-1]
+
         # Fetch track info
         track_info = get_track_info(track_id)
         track_link = track_info.get("link")
@@ -66,7 +86,8 @@ def download_track(track_id: str, quality: str = "MP3_320"):
         # Sanitize filename
         track_title = track_info.get("title", "track")
         artist_name = track_info.get("artist", {}).get("name", "unknown")
-        expected_filename = f"{artist_name} - {track_title}.mp3".replace("/", "_")  # Sanitize filename
+        file_extension = "flac" if quality == "FLAC" else "mp3"
+        expected_filename = f"{artist_name} - {track_title}.{file_extension}".replace("/", "_")  # Sanitize filename
 
         # Clear the downloads directory
         for root, dirs, files in os.walk("downloads"):
@@ -85,21 +106,21 @@ def download_track(track_id: str, quality: str = "MP3_320"):
             recursive_download=False
         )
 
-        # Recursively search for the MP3 file in the downloads directory
-        mp3_filepath = None
+        # Recursively search for the file in the downloads directory
+        filepath = None
         for root, dirs, files in os.walk("downloads"):
             for file in files:
-                if file.endswith('.mp3'):
-                    mp3_filepath = os.path.join(root, file)
+                if file.endswith(f'.{file_extension}'):
+                    filepath = os.path.join(root, file)
                     break
-            if mp3_filepath:
+            if filepath:
                 break
 
-        if not mp3_filepath:
-            raise HTTPException(status_code=500, detail="MP3 file not found after download")
+        if not filepath:
+            raise HTTPException(status_code=500, detail=f"{file_extension} file not found after download")
 
         # Return the download URL
-        relative_path = os.path.relpath(mp3_filepath, "downloads")
+        relative_path = os.path.relpath(filepath, "downloads")
         # Remove spaces from the relative path
         relative_path = relative_path.replace(" ", "%20")
         download_url = f"{BASE_URL}/downloads/{relative_path}"
@@ -109,6 +130,7 @@ def download_track(track_id: str, quality: str = "MP3_320"):
         logger.error(f"Error downloading track: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
+
 # Search tracks using Deezer API
 @app.get("/search")
 def search_tracks(query: str, limit: Optional[int] = 10):
@@ -120,4 +142,4 @@ def search_tracks(query: str, limit: Optional[int] = 10):
         raise HTTPException(status_code=500, detail=str(e))
     except Exception as e:
         logger.error(f"Error searching tracks: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))