Spaces:
Sleeping
Sleeping
File size: 1,663 Bytes
d951942 a4348b8 d951942 47408f6 d951942 40b8269 d951942 61e269a d951942 c2a5f95 d951942 40b8269 d951942 47408f6 d951942 40b8269 |
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 |
from typing import Union
import re
from youtube_transcript_api import YouTubeTranscriptApi
from youtube_transcript_api.formatters import TextFormatter
from transformers import pipeline
from fastapi import FastAPI, Query
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse
app = FastAPI()
text_summary = pipeline("summarization", model="Falconsai/text_summarization")
def extract_video_id(url):
regex = r"(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
match = re.search(regex, url)
if match:
return match.group(1)
return None
def get_youtube_transcript(video_url):
video_id = extract_video_id(video_url)
if not video_id:
return "Video ID could not be extracted."
try:
transcript = YouTubeTranscriptApi.get_transcript(video_id)
formatter = TextFormatter()
text_transcript = formatter.format_transcript(transcript)
return text_transcript
except Exception as e:
return f"An error occurred: {e}"
# Get the input from the frontend
@app.get("/getdata")
def get_data(input):
print(input)
transcript = get_youtube_transcript(input)
if transcript:
summary = text_summary(transcript,min_length=10,max_length=1000,do_sample=False)
return {"summary": summary[0]['summary_text']}
else:
return {"error": "Failed to get transcript from the YouTube video."}
app.mount('/', StaticFiles(directory="static", html=True), name="static")
@app.get('/')
def index() -> FileResponse:
return FileResponse('/app/static/index.html', media_type="text/html")
|