File size: 2,890 Bytes
c58c436 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound, VideoUnavailable
import re
app = FastAPI()
# Enable CORS so your Vercel frontend can call this API
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # In production, you can change "*" to your Vercel URL
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
def extract_video_id(url: str) -> str:
pattern = r'(?:youtube\.com\/(?:[^\/]+\/.+\/|(?:v|e(?:mbed)?)\/|.*[?&]v=)|youtu\.be\/)([^"&?\/\s]{11})'
match = re.match(pattern, url)
if match:
return match.group(1)
return None
@app.post("/api/transcript")
async def get_transcript(request: dict):
url = request.get("url")
if not url:
raise HTTPException(status_code=400, detail="Missing URL")
video_id = extract_video_id(url)
if not video_id:
raise HTTPException(status_code=400, detail="Invalid YouTube URL")
try:
# Fetch available transcripts
transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
# Find the best transcript (Prefer manual English, then auto English, then any other)
transcript = None
try:
transcript = transcript_list.find_manually_created_transcript(['en'])
except NoTranscriptFound:
try:
transcript = transcript_list.find_generated_transcript(['en'])
except NoTranscriptFound:
# If no English, get the first available language
for t in transcript_list:
transcript = t
break
if not transcript:
raise HTTPException(status_code=404, detail="No captions found for this video.")
# Fetch the actual text data
transcript_data = transcript.fetch()
# Combine all text lines
full_text = " ".join([line['text'] for line in transcript_data])
# Clean up HTML entities
full_text = full_text.replace("'", "'").replace(""", '"').replace("&", "&").replace("'", "'")
return {"text": full_text}
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 Exception as e:
raise HTTPException(status_code=500, detail=f"Failed to fetch transcript: {str(e)}")
# ========================================================
# KEEP-ALIVE ENDPOINT
# ========================================================
@app.get("/api/health")
async def health_check():
return {"status": "ok", "message": "Transcript API is running!"} |