upodated

main.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound, VideoUnavailable, TooManyRequests, NoTranscriptAvailable
+from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound, VideoUnavailable
 import re
 from pydantic import BaseModel
 
@@ -74,18 +74,20 @@ async def get_transcript(req: TranscriptRequest):
 
     except HTTPException:
         raise # Re-raise our custom HTTP exceptions
-    except TooManyRequests:
-        raise HTTPException(status_code=429, detail="YouTube is rate-limiting requests from this server. Please try again in a minute.")
     except TranscriptsDisabled:
         raise HTTPException(status_code=403, detail="Captions are disabled for this video.")
     except VideoUnavailable:
         raise HTTPException(status_code=404, detail="Video is unavailable.")
-    except NoTranscriptAvailable:
-        raise HTTPException(status_code=404, detail="No transcript is available for this video.")
     except Exception as e:
         # Catch-all: Log the exact error to Hugging Face logs so we can see what went wrong!
-        print(f"UNHANDLED ERROR for {video_id}: {type(e).__name__} - {str(e)}")
-        raise HTTPException(status_code=500, detail=f"Failed to fetch transcript: {type(e).__name__} - {str(e)}")
+        error_name = type(e).__name__
+        print(f"UNHANDLED ERROR for {video_id}: {error_name} - {str(e)}")
+        
+        # Check if it's a rate limit error dynamically
+        if "TooManyRequests" in error_name or "429" in str(e):
+             raise HTTPException(status_code=429, detail="YouTube is rate-limiting requests from this server. Please try again in a minute.")
+        
+        raise HTTPException(status_code=500, detail=f"Failed to fetch transcript: {error_name} - {str(e)}")
 
 # KEEP-ALIVE ENDPOINT
 @app.get("/api/health")