Update modules/app.py

modules/app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI, HTTPException, Request, Depends
-from fastapi.responses import FileResponse
+from fastapi.staticfiles import StaticFiles
 from pytube import YouTube
+import uuid
 import logging
 import aiohttp
 import base64
@@ -32,6 +33,13 @@ def generate_hash(length=12):
 async def read_root():
     return {"message": "Saqib's API"}
 
+# Create a directory to store MP3 files if it doesn't exist
+AUDIO_DIR = "audio_files"
+os.makedirs(AUDIO_DIR, exist_ok=True)
+
+# Mount the audio directory
+app.mount("/audio", StaticFiles(directory=AUDIO_DIR), name="audio")
+
 @app.get("/get_audio")
 async def get_audio(url: str):
     if not url:
@@ -40,19 +48,21 @@ async def get_audio(url: str):
     try:
         yt = YouTube(url)
         video = yt.streams.filter(only_audio=True).first()
-        # out_file = video.download(output_path=".")
-        # current_dir
-        out_file = video.download()
+        
+        # Generate a unique filename
+        unique_filename = f"{uuid.uuid4().hex}.mp3"
+        out_file = os.path.join(AUDIO_DIR, unique_filename)
+        
+        # Download the audio
+        video.download(output_path=AUDIO_DIR, filename=unique_filename)
         
         file_stats = os.stat(out_file)
         logging.info(f'Size of audio file in Bytes: {file_stats.st_size}')
         
         if file_stats.st_size <= 30000000:
-            base, ext = os.path.splitext(out_file)
-            new_file = base + '.mp3'
-            os.rename(out_file, new_file)
-            
-            return FileResponse(new_file, media_type="audio/mpeg", filename="audio.mp3")
+            # Construct the URL for the MP3 file
+            mp3_url = f"/audio/{unique_filename}"
+            return {"mp3_url": mp3_url}
         else:
             os.remove(out_file)
             raise HTTPException(status_code=413, detail="Audio file is too large. Limited to about 1.5 hours.")