Chrunos commited on
Commit
4bc7387
·
verified ·
1 Parent(s): 6b74354

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py CHANGED
@@ -1,6 +1,10 @@
1
  from flask import Flask, render_template, jsonify, request
2
  from ytmusicapi import YTMusic
3
  import os
 
 
 
 
4
 
5
  app = Flask(__name__)
6
  ytmusic = YTMusic()
@@ -22,6 +26,40 @@ def searcht():
22
  first_song = next((song for song in search_results if 'videoId' in song and song['videoId']), {}) if search_results else {}
23
  return jsonify(first_song)
24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  @app.route('/get_artist', methods=['GET'])
26
  def get_artist():
27
  artist_id = request.args.get('id')
 
1
  from flask import Flask, render_template, jsonify, request
2
  from ytmusicapi import YTMusic
3
  import os
4
+ import logging
5
+ from datetime import datetime, timedelta
6
+ import time
7
+
8
 
9
  app = Flask(__name__)
10
  ytmusic = YTMusic()
 
26
  first_song = next((song for song in search_results if 'videoId' in song and song['videoId']), {}) if search_results else {}
27
  return jsonify(first_song)
28
 
29
+
30
+ async def get_track_download_url(
31
+ """Get download URL with fallback."""
32
+ apis = [
33
+ f"https://cobalt-api.ayo.tf",
34
+ f"http://34.107.254.11"
35
+ ]
36
+ cHeaders =
37
+
38
+ for api_url in apis:
39
+ try:
40
+ logger.info(f"Attempting to get download URL from: {api_url}")
41
+ response = requests.post(api_url, timeout=20, data={"url": "https://youtu.be/8nBFqZppIF0"}, headers={"Accept": "application/json", "Content-Type": "application/json"})
42
+ if response.status_code == 200:
43
+ download_url = response.json().get("url")
44
+ if download_url:
45
+ logger.info(f"Successfully obtained download URL from {api_url}")
46
+ return download_url
47
+ except requests.exceptions.RequestException as e:
48
+ logger.warning(f"Failed to get download URL from {api_url}: {str(e)}")
49
+ continue
50
+
51
+ logger.error(f"No download URL found for track {track_id}")
52
+ raise HTTPException(status_code=404, detail="Download URL not found")
53
+
54
+ @app.post('/track_dl')
55
+ def track_dl(
56
+ track_id=request.json.get('track_id', '')
57
+ dl_url = await get_track_download_url(url)
58
+ result = {"url": dl_url}
59
+ return result
60
+ )
61
+
62
+
63
  @app.route('/get_artist', methods=['GET'])
64
  def get_artist():
65
  artist_id = request.args.get('id')