Update app.py

app.py

@@ -126,6 +126,52 @@ async def get_video_url(youtube_url: str):
 
 
 
+@app.get("/script")
+async def get_transcript(youtube_url: str):
+    try:
+        # 定义 ydl_opts
+        ydl_opts = {
+            'skip_download': True,
+            'writesubtitles': True,
+            'writeautomaticsub': True,
+            'subtitleslangs': ['en'],  # 可以根据需要修改语言
+            'format': 'bestaudio/best',
+            'outtmpl': '%(id)s.%(ext)s',
+            'noplaylist': True
+        }
+        cookiefile = "firefox-cookies.txt"
+        ydl_opts["cookiefile"] = cookiefile
+
+        with YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(youtube_url, download=False)
+            subtitles = info.get('subtitles', {})
+            auto_subtitles = info.get('automatic_captions', {})
+
+            transcript = None
+            if subtitles:
+                # 优先使用手动字幕
+                for lang in ydl_opts['subtitleslangs']:
+                    if lang in subtitles:
+                        transcript = subtitles[lang]
+                        break
+            if not transcript and auto_subtitles:
+                # 如果没有手动字幕，使用自动生成的字幕
+                for lang in ydl_opts['subtitleslangs']:
+                    if lang in auto_subtitles:
+                        transcript = auto_subtitles[lang]
+                        break
+
+            if transcript:
+                text = ' '.join([entry['text'] for entry in transcript])
+                return jsonify({'transcript': text})
+            else:
+                return jsonify({'transcript': 'No transcript available'})
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+
 # Define a global temporary download directory
 global_download_dir = tempfile.mkdtemp()