Chrunos commited on
Commit
ac14607
·
verified ·
1 Parent(s): 697d26d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -41,6 +41,7 @@ ARL_TOKEN = os.getenv('ARL')
41
  class DownloadRequest(BaseModel):
42
  url: str
43
  quality: str
 
44
 
45
  @app.get("/")
46
  def read_root():
@@ -72,9 +73,13 @@ def get_track(track_id: str):
72
  @app.post("/download/track")
73
  def download_track(request: DownloadRequest):
74
  try:
75
- dl = DeeLogin(arl=ARL_TOKEN)
 
 
 
76
  url = request.url
77
  quality = request.quality
 
78
 
79
  if quality not in ["MP3_320", "MP3_128", "FLAC"]:
80
  raise HTTPException(status_code=400, detail="Invalid quality specified")
 
41
  class DownloadRequest(BaseModel):
42
  url: str
43
  quality: str
44
+ arl: str
45
 
46
  @app.get("/")
47
  def read_root():
 
73
  @app.post("/download/track")
74
  def download_track(request: DownloadRequest):
75
  try:
76
+ if request.arl is None or request.arl.strip() == "":
77
+ ARL = ARL_TOKEN
78
+ else:
79
+ ARL = request.arl
80
  url = request.url
81
  quality = request.quality
82
+ dl = DeeLogin(arl=ARL)
83
 
84
  if quality not in ["MP3_320", "MP3_128", "FLAC"]:
85
  raise HTTPException(status_code=400, detail="Invalid quality specified")