Update app.py

app.py

@@ -5,7 +5,7 @@ import logging
 import requests
 import subprocess
 from fastapi import FastAPI, Request, HTTPException
-from fastapi.responses import FileResponse
+from fastapi.staticfiles import StaticFiles
 import tempfile
 import os
 import uuid
@@ -26,6 +26,7 @@ app = FastAPI()
 
 # Temporary directory for file storage
 global_download_dir = tempfile.mkdtemp()
+logging.basicConfig(level=logging.DEBUG)
 
 def md5(string):
     """Generate MD5 hash of a string."""
@@ -47,11 +48,11 @@ async def fetch_data_for_url(request: Request):
     if not url:
         raise HTTPException(status_code=400, detail="URL is required")
 
-    print(f'Fetching data for: {url}')
+    logging.info(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.')
+        logging.error('Track ID not found in your input.')
         raise HTTPException(status_code=400, detail="Track ID not found in URL")
     
     track_id = track_id_match.group(0)
@@ -63,10 +64,12 @@ async def fetch_data_for_url(request: Request):
     
     response = requests.get(download_url, headers=HEADERS)
     if response.status_code != 200:
+        logging.error(f"Failed to fetch the track file URL: {response.status_code}")
         raise HTTPException(status_code=500, detail="Failed to fetch the track file URL")
     
     file_url = response.json().get('url')
     if not file_url:
+        logging.error("No file URL returned from Qobuz")
         raise HTTPException(status_code=500, detail="No file URL returned")
     
     # Download the FLAC file
@@ -76,14 +79,27 @@ async def fetch_data_for_url(request: Request):
         with open(flac_file_path, 'wb') as f:
             for chunk in r.iter_content(chunk_size=8192):
                 f.write(chunk)
+    logging.info(f"FLAC file downloaded at {flac_file_path}")
 
     # Convert FLAC to ALAC
     alac_file_path = os.path.join(global_download_dir, f'{uuid.uuid4()}.m4a')
     subprocess.run(['ffmpeg', '-i', flac_file_path, '-c:a', 'alac', alac_file_path], check=True)
+    logging.info(f"Converted to ALAC: {alac_file_path}")
 
-    # Generate a URL for the converted ALAC file
-    # You can use a URL or an endpoint that serves the file, e.g., your server file path or public URL
-    alac_file_url = f'{BASE_URL}/files/{os.path.basename(alac_file_path)}'  # Adjust this depending on your setup
+    # Generate the file URL for ALAC
+    alac_file_url = f'{BASE_URL}/file/{os.path.basename(alac_file_path)}'
+    logging.info(f"Returning ALAC file URL: {alac_file_url}")
 
-    # Return the URL to the ALAC file as a JSON response
+    # Return the ALAC file URL to the client as JSON
     return {"url": alac_file_url}
+
+# Mount StaticFiles to serve files from the global download directory
+app.mount("/file", StaticFiles(directory=global_download_dir), name="downloads")
+
+# Middleware to set the correct MIME type
+@app.middleware("http")
+async def set_mime_type_middleware(request: Request, call_next):
+    response = await call_next(request)
+    if request.url.path.endswith(".m4a"):
+        response.headers["Content-Type"] = "audio/mp4"
+    return response