tregu0458 commited on
Commit
8a7264a
·
verified ·
1 Parent(s): e1df389

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py CHANGED
@@ -4,6 +4,21 @@ from langchain.document_loaders import YoutubeLoader
4
  app = FastAPI()
5
 
6
  @app.post("/transcripts",tags=["Transcript"])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  def get_youtube_transcript(youtube_url: str, language: str = "ja"):
8
  try:
9
  loader = YoutubeLoader.from_youtube_url(
 
4
  app = FastAPI()
5
 
6
  @app.post("/transcripts",tags=["Transcript"])
7
+ def get_youtube_transcript(youtube_url: str, language: str = "ja"):
8
+ try:
9
+ loader = YoutubeLoader.from_youtube_url(
10
+ youtube_url=youtube_url,
11
+ add_video_info=True,
12
+ language=[language],
13
+ )
14
+ docs = loader.load()
15
+ # transcript = " ".join([doc.page_content for doc in docs])
16
+ return {"transcript": str(docs)}
17
+ except Exception as e:
18
+ error_msg = str(e)
19
+ return {"messsage": error_msg}
20
+
21
+ @app.get("/transcripts",tags=["Transcript"])
22
  def get_youtube_transcript(youtube_url: str, language: str = "ja"):
23
  try:
24
  loader = YoutubeLoader.from_youtube_url(