Update modules/app.py

modules/app.py

@@ -1,8 +1,7 @@
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.staticfiles import StaticFiles
-from pytubefix import YouTube
-from pytubefix.exceptions import PytubeFixError
 from concurrent.futures import ThreadPoolExecutor
+from youtube_transcript_api import YouTubeTranscriptApi
 import asyncio
 import aiohttp
 import aiofiles
@@ -207,6 +206,26 @@ async def get_youtube_audio(request: Request):
         print(f"An error occurred: {str(e)}")
         print(traceback.format_exc())
         raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
+    
+@app.get("/get_youtube_transcript")
+async def get_youtube_transcript(request: Request):
+    try:
+        # Get the video ID from the query parameters
+        video_id = request.query_params.get("video_id")
+        if not video_id:
+            raise HTTPException(status_code=400, detail="video_id parameter is required")
+
+        # Get the transcript for the video
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+
+        # Format the transcript into a single string
+        transcript_text = " ".join([i["text"] for i in transcript])
+
+        return {"transcript": transcript_text}
+    except Exception as e:
+        print(f"An error occurred: {str(e)}")
+        print(traceback.format_exc())
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
 
 @app.post("/whisper")
 async def whisper(request: Request):