Update modules/app.py

modules/app.py

@@ -9,6 +9,7 @@ import base64
 import io
 import os
 import random
+import traceback
 import string
 import json
 
@@ -42,20 +43,24 @@ os.makedirs(AUDIO_DIR, exist_ok=True)
 app.mount("/audio", StaticFiles(directory=AUDIO_DIR), name="audio")
 
 @app.get("/get_audio")
-async def get_audio(url: str):
-    print(url)
-    
+async def get_audio(url: str):    
     if not url:
         raise HTTPException(status_code=400, detail="URL is required")
 
     try:
-        time.sleep(2)
+        print(f"Attempting to process YouTube URL: {url}")
         yt = YouTube(url)
         video = yt.streams.filter(only_audio=True).first()
         
+        if not video:
+            print(f"No audio stream found for URL: {url}")
+            raise HTTPException(status_code=404, detail="No audio stream found for this video")
+        
         # Generate a unique filename
         unique_filename = f"{uuid.uuid4().hex}.mp3"
         out_file = os.path.join(AUDIO_DIR, unique_filename)
+
+        print((f"Downloading audio to: {out_file}"))
         
         # Download the audio
         video.download(output_path=AUDIO_DIR, filename=unique_filename)
@@ -70,9 +75,16 @@ async def get_audio(url: str):
         else:
             os.remove(out_file)
             raise HTTPException(status_code=413, detail="Audio file is too large. Limited to about 1.5 hours.")
+    except PytubeError as e:
+        print(f"PytubeError occurred: {str(e)}")
+        raise HTTPException(status_code=400, detail=f"Error processing YouTube video: {str(e)}")
+    except HTTPException as he:
+        # Re-raise HTTP exceptions
+        raise he
     except Exception as e:
-        print(e)
-        raise HTTPException(status_code=500, detail=str(e))
+        print(f"Unexpected error occurred: {str(e)}")
+        print(traceback.format_exc())
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
 
 @app.post("/whisper")
 async def whisper(request: Request):