from fastapi import FastAPI, HTTPException
from langchain.document_loaders import YoutubeLoader

app = FastAPI()

@app.post("/transcripts",tags=["Transcript"])
def get_youtube_transcript(youtube_url: str, language: str = "ja"):
    try:
        loader = YoutubeLoader.from_youtube_url(
            youtube_url=youtube_url,
            add_video_info=True,
            language=[language],
        )
        docs = loader.load()
        # transcript = " ".join([doc.page_content for doc in docs])
        return {"transcript": str(docs)}
    except Exception as e:
        error_msg = str(e)
        return {"messsage": error_msg}

@app.get("/transcripts",tags=["Transcript"])
def get_youtube_transcript(youtube_url: str, language: str = "ja",length:int = 150000):
    try:
        loader = YoutubeLoader.from_youtube_url(
            youtube_url=youtube_url,
            add_video_info=True,
            language=[language],
        )
        docs = loader.load()
        textTranscript = str(docs)
        if len(textTranscript)<length:
            return {"transcript": textTranscript}
        else:
            return {"transcript": textTranscript[:int(length/2)]+textTranscript[len(textTranscript) - int(length/2):]}
    except Exception as e:
        error_msg = str(e)
        return {"messsage": error_msg}