Update modules/app.py

modules/app.py

@@ -1,4 +1,6 @@
 from fastapi import FastAPI, HTTPException, Request, Depends
+from pytube import YouTube
+import logging
 import aiohttp
 import base64
 import io
@@ -28,7 +30,34 @@ def generate_hash(length=12):
 @app.get("/")
 async def read_root():
     return {"message": "Saqib's API"}
-    
+
+@app.get("/get_audio")
+async def get_audio(url: str):
+    if not url:
+        raise HTTPException(status_code=400, detail="URL is required")
+
+    try:
+        yt = YouTube(url)
+        video = yt.streams.filter(only_audio=True).first()
+        # out_file = video.download(output_path=".")
+        # current_dir
+        out_file = video.download()
+        
+        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")
+        else:
+            os.remove(out_file)
+            raise HTTPException(status_code=413, detail="Audio file is too large. Limited to about 1.5 hours.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 @app.post("/whisper")
 async def whisper(request: Request):
     data = await request.json()  # Extracting JSON data from request