Update app.py

app.py

@@ -1,6 +1,7 @@
 import shutil
 from fastapi import FastAPI, HTTPException
 from deezspot.deezloader import DeeLogin
+from deezspot.spotloader import SpoLogin
 import requests
 import os
 import logging
@@ -8,6 +9,7 @@ from typing import Optional
 from fastapi.staticfiles import StaticFiles
 from dotenv import load_dotenv
 from pydantic import BaseModel
+from urllib.parse import quote
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -31,6 +33,18 @@ BASE_URL = "https://tecuts-depot.hf.space"
 ARL_TOKEN = os.getenv('ARL')
 dl = DeeLogin(arl=ARL_TOKEN)
 
+# Spotify credentials from environment variables
+SPOTIFY_USERNAME = os.getenv("SPOTIFY_USERNAME")
+SPOTIFY_CREDENTIALS = os.getenv("SPOTIFY_CREDENTIALS")
+
+if not SPOTIFY_USERNAME or not SPOTIFY_CREDENTIALS:
+    raise RuntimeError("Spotify credentials not found in environment variables")
+
+# Initialize Spotify client
+spo = SpoLogin(credentials_path=None)  # We'll pass credentials directly
+spo.username = SPOTIFY_USERNAME
+spo.credentials = SPOTIFY_CREDENTIALS
+
 
 # 定义请求体模型
 class DownloadRequest(BaseModel):
@@ -38,6 +52,46 @@ class DownloadRequest(BaseModel):
     quality: str
 
 
+
+# Download a Spotify track and return a download URL
+@app.post("/spot-track/{track_id}")
+def download_spotify_track(track_id: str):
+    try:
+        # Download the track using spotloader
+        logger.info(f"Downloading Spotify track: {track_id}")
+        spo.download_track(
+            link_track=f"https://open.spotify.com/track/{track_id}",
+            output_dir="downloads",
+            quality_download="NORMAL",
+            recursive_quality=False,
+            recursive_download=False,
+            not_interface=False,
+            method_save=1
+        )
+
+        # Search for the downloaded file
+        downloaded_files = os.listdir("downloads")
+        logger.info(f"Files in downloads directory: {downloaded_files}")
+
+        # Find the file that matches the track ID
+        matching_files = [file for file in downloaded_files if track_id in file]
+
+        if not matching_files:
+            raise HTTPException(status_code=500, detail="File download failed")
+
+        # Use the first matching file
+        filename = matching_files[0]
+        filepath = os.path.join("downloads", filename)
+
+        # Return the download URL
+        download_url = f"/downloads/{quote(filename)}"
+        logger.info(f"Download successful: {download_url}")
+        return {"download_url": download_url}
+    except Exception as e:
+        logger.error(f"Error downloading Spotify track: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+
 @app.get("/")
 def read_root():
     return {"message": "running"}