Update app.py

app.py

@@ -1,10 +1,12 @@
-import requests
 import hashlib
 import time
+import re
+import logging
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import FileResponse
 import tempfile
 import os
+import requests
 
 # Constants
 BASE_URL = 'https://www.qobuz.com/api.json/0.2/'
@@ -19,8 +21,6 @@ HEADERS = {
 }
 app = FastAPI()
 
-
-
 # Define a global temporary download directory
 global_download_dir = tempfile.mkdtemp()
 os.makedirs(global_download_dir, exist_ok=True)
@@ -30,27 +30,29 @@ BASE_URL = "https://tecuts-request.hf.space"
 def md5(string):
     """Generate MD5 hash of a string."""
     return hashlib.md5(string.encode('utf-8')).hexdigest()
-    
+
 @app.get('/')
 def main():
     return {"appStatus": "running"}
 
 @app.post('/track')
-
-def fetch_data_for_url:
+async def fetch_data_for_url(request: Request):
     try:
         data = await request.json()
     except Exception as e:
         logging.error(f"Error parsing JSON: {e}")
         raise HTTPException(status_code=400, detail="Invalid JSON")
+    
     url = data.get('url')
+    if not url:
+        raise HTTPException(status_code=400, detail="URL is required")
 
-    """Fetch data for a specific track URL."""
     print(f'Fetching data for: {url}')
+    
     track_id_match = re.search(r'\d+$', url)
     if not track_id_match:
         print('Track ID not found in your input.')
-        return
+        raise HTTPException(status_code=400, detail="Track ID not found in URL")
     
     track_id = track_id_match.group(0)
     timestamp = int(time.time())
@@ -58,21 +60,5 @@ def fetch_data_for_url:
     rSig = md5(rSigRaw)
     
     download_url = f'{BASE_URL}track/getFileUrl?format_id=27&intent=stream&track_id={track_id}&request_ts={timestamp}&request_sig={rSig}'
-return {
-    "download_url": download_url
-}
     
-#    try:
-#       download_response = requests.get(download_url, headers=HEADERS)
-  #      download_response.raise_for_status()
- #       
-  #      file_url = download_response.json().get('url')
-   #     if file_url:
-    #        print(f'Download link: {file_url}')
-     #   else:
-      #      print('Error fetching download URL.')
-    #
-    #except requests.exceptions.RequestException:
-     #   print('Error fetching data for the URL.')
-
-
+    return {"download_url": download_url}