Update app.py

app.py

@@ -19,7 +19,7 @@ def get_youtube_transcript(youtube_url: str, language: str = "ja"):
         return {"messsage": error_msg}
 
 @app.get("/transcripts",tags=["Transcript"])
-def get_youtube_transcript(youtube_url: str, language: str = "ja"):
+def get_youtube_transcript(youtube_url: str, language: str = "ja",length:int = 150000):
     try:
         loader = YoutubeLoader.from_youtube_url(
             youtube_url=youtube_url,
@@ -28,7 +28,10 @@ def get_youtube_transcript(youtube_url: str, language: str = "ja"):
         )
         docs = loader.load()
         # transcript = " ".join([doc.page_content for doc in docs])
-        return {"transcript": str(docs)}
+        if len(str(docs)<length):
+            return {"transcript": str(docs)}
+        else:
+            return {"transcript": str(docs)[:length]}
     except Exception as e:
         error_msg = str(e)
         return {"messsage": error_msg}