Update app.py

app.py

@@ -37,85 +37,6 @@ BASE_URL = "https://chrunos-depot.hf.space"
 
 # Deezer ARL token (required for deezspot downloads)
 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")
-
-# Create a temporary credentials.json file
-CREDENTIALS_PATH = "/tmp/credentials.json"
-with open(CREDENTIALS_PATH, "w") as f:
-    json.dump({
-        "username": SPOTIFY_USERNAME,
-        "credentials": SPOTIFY_CREDENTIALS,
-        "type": "AUTHENTICATION_STORED_SPOTIFY_CREDENTIALS"
-    }, f)
-
-# Initialize Spotify client
-spo = SpoLogin(credentials_path=CREDENTIALS_PATH)
-
-
-# 定义请求体模型
-class DownloadRequest(BaseModel):
-    url: str
-    quality: str
-
-
-def cleanup_dir(dir_path: Path):
-    """Background task to clean up directory"""
-    try:
-        shutil.rmtree(dir_path)
-        logger.info(f"Cleaned up directory: {dir_path}")
-    except Exception as e:
-        logger.error(f"Error cleaning up {dir_path}: {e}")
-
-
-# Download a Spotify track and return a download URL       
-@app.post("/spot-track/{track_id}")
-async def download_spotify_track(
-    track_id: str,
-    background_tasks: BackgroundTasks
-):
-    try:
-        downloads_dir = Path("downloads")
-        # Create unique directory for this download
-        download_id = uuid.uuid4().hex
-        download_dir = downloads_dir / download_id
-        download_dir.mkdir(parents=True, exist_ok=True)
-
-        # Download to unique directory
-        logger.info(f"Downloading to {download_dir}")
-        spo.download_track(
-            link_track=f"https://open.spotify.com/track/{track_id}",
-            output_dir=str(download_dir),
-            quality_download="VERY_HIGH",
-            recursive_quality=False,
-            recursive_download=False,
-            not_interface=False,
-            method_save=1
-        )
-
-        # Find downloaded file
-        filepath = next(download_dir.glob(f"**/*.ogg"), None)
-        if not filepath:
-            raise HTTPException(status_code=500, detail="File not found after download")
-
-        # Schedule cleanup after response is sent
-        #background_tasks.add_task(cleanup_dir, download_dir)
-
-        # Return path with unique directory
-        relative_path = quote(str(filepath.relative_to(downloads_dir)))
-        return {"download_url": f"{BASE_URL}/downloads/{relative_path}"}
-
-    except Exception as e:
-        logger.error(f"Error downloading track: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
-
-
 
 @app.get("/")
 def read_root():
@@ -147,6 +68,7 @@ def get_track(track_id: str):
 @app.post("/download/track")
 def download_track(request: DownloadRequest):
     try:
+        dl = DeeLogin(arl=ARL_TOKEN)
         url = request.url
         quality = request.quality
 
@@ -229,3 +151,83 @@ def search_tracks(query: str, limit: Optional[int] = 10):
     except Exception as e:
         logger.error(f"Error searching tracks: {e}")
         raise HTTPException(status_code=500, detail=str(e))
+
+# 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")
+
+# Create a temporary credentials.json file
+CREDENTIALS_PATH = "/tmp/credentials.json"
+with open(CREDENTIALS_PATH, "w") as f:
+    json.dump({
+        "username": SPOTIFY_USERNAME,
+        "credentials": SPOTIFY_CREDENTIALS,
+        "type": "AUTHENTICATION_STORED_SPOTIFY_CREDENTIALS"
+    }, f)
+
+# Initialize Spotify client
+spo = SpoLogin(credentials_path=CREDENTIALS_PATH)
+
+
+# 定义请求体模型
+class DownloadRequest(BaseModel):
+    url: str
+    quality: str
+
+
+def cleanup_dir(dir_path: Path):
+    """Background task to clean up directory"""
+    try:
+        shutil.rmtree(dir_path)
+        logger.info(f"Cleaned up directory: {dir_path}")
+    except Exception as e:
+        logger.error(f"Error cleaning up {dir_path}: {e}")
+
+
+# Download a Spotify track and return a download URL       
+@app.post("/spot-track/{track_id}")
+async def download_spotify_track(
+    track_id: str,
+    background_tasks: BackgroundTasks
+):
+    try:
+        downloads_dir = Path("downloads")
+        # Create unique directory for this download
+        download_id = uuid.uuid4().hex
+        download_dir = downloads_dir / download_id
+        download_dir.mkdir(parents=True, exist_ok=True)
+
+        # Download to unique directory
+        logger.info(f"Downloading to {download_dir}")
+        spo.download_track(
+            link_track=f"https://open.spotify.com/track/{track_id}",
+            output_dir=str(download_dir),
+            quality_download="VERY_HIGH",
+            recursive_quality=False,
+            recursive_download=False,
+            not_interface=False,
+            method_save=1
+        )
+
+        # Find downloaded file
+        filepath = next(download_dir.glob(f"**/*.ogg"), None)
+        if not filepath:
+            raise HTTPException(status_code=500, detail="File not found after download")
+
+        # Schedule cleanup after response is sent
+        #background_tasks.add_task(cleanup_dir, download_dir)
+
+        # Return path with unique directory
+        relative_path = quote(str(filepath.relative_to(downloads_dir)))
+        return {"download_url": f"{BASE_URL}/downloads/{relative_path}"}
+
+    except Exception as e:
+        logger.error(f"Error downloading track: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+'''
+
+
+