Update app.py

app.py

@@ -1,5 +1,5 @@
 import shutil
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException, Request, Body
 from deezspot.deezloader import DeeLogin
 from deezspot.spotloader import SpoLogin
 import requests
@@ -9,7 +9,7 @@ import json
 from typing import Optional
 from fastapi.staticfiles import StaticFiles
 from dotenv import load_dotenv
-from pydantic import BaseModel
+from pydantic import BaseModel, Field, HttpUrl
 from urllib.parse import quote
 from pathlib import Path
 import uuid
@@ -18,7 +18,8 @@ from collections import defaultdict
 import time
 from datetime import timedelta
 import gc
-from typing import Dict
+from typing import Dict, Union, Literal
+import urllib.parse
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -306,6 +307,70 @@ def search_tracks(query: str, limit: Optional[int] = 10):
         logger.error(f"Error searching tracks: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
+
+# --- Request Body Model ---
+# Defines the expected structure and validation for the request body.
+class SpotDlRequest(BaseModel):
+    url: HttpUrl = Field(..., description="The URL to be processed.")
+    quality: Literal["128", "320", "FLAC"] = Field(
+        ...,
+        description="The desired quality. Currently, only '128' is supported for link generation."
+    )
+
+# --- Response Body Model ---
+# Defines the structure for a successful response.
+class SpotDlResponse(BaseModel):
+    download_url: str
+
+# --- Error Response Model ---
+# Defines the structure for an error response.
+class ErrorResponse(BaseModel):
+    detail: str
+
+# --- API Endpoint ---
+@app.post(
+    "/spot_dl",
+    response_model=SpotDlResponse,
+    responses={
+        400: {"model": ErrorResponse, "description": "Bad Request - Invalid input"},
+        422: {"model": ErrorResponse, "description": "Validation Error - Input data is not valid"}
+    },
+    summary="Generate SpotDL Link",
+    description="Accepts a URL and quality, returns a processed URL if quality is '128', "
+                "otherwise returns an error for higher qualities."
+)
+async def create_spot_dl_link(request: SpotDlRequest = Body(...)):
+    """
+    Processes a URL and quality to generate a download link.
+
+    - **url**: The URL to process (must be a valid HTTP/HTTPS URL).
+    - **quality**: The desired quality. Must be one of "128", "320", "FLAC".
+                   Currently, only "128" will result in a successful link generation.
+    """
+    # Log the received request for debugging (optional)
+    print(f"Received request: url='{request.url}', quality='{request.quality}'")
+
+    # Check the quality
+    if request.quality == "128":
+        # URL-encode the input URL
+        encoded_url = urllib.parse.quote(str(request.url))
+        # Construct the target URL
+        output_url = f"https://velynapi.vercel.app/api/downloader/spotifydl?url{encoded_url}"
+        return SpotDlResponse(download_url=output_url)
+    elif request.quality == "320" or request.quality == "FLAC":
+        # If quality is 320 or FLAC, return an error as per requirements
+        raise HTTPException(
+            status_code=400,
+            detail=f"Quality '{request.quality}' is for Premium Users Only. '128' is allowed."
+        )
+    else:
+        # This case should ideally be caught by Pydantic's Literal validation,
+        # but as a fallback or for more general invalid quality values:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Invalid quality value: '{request.quality}'. Allowed values are '128', '320', 'FLAC'."
+        )
+
 # Spotify credentials from environment variables
 '''SPOTIFY_USERNAME = os.getenv("SPOTIFY_USERNAME")
 SPOTIFY_CREDENTIALS = os.getenv("SPOTIFY_CREDENTIALS")