tregu0458 commited on
Commit
9cce471
·
verified ·
1 Parent(s): e6da56f

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from langchain_community.document_loaders import YoutubeLoader
3
+
4
+ def load_youtube_transcript(url, language):
5
+ """
6
+ YouTubeビデオから日本語のトランスクリプトを取得する。
7
+
8
+ Args:
9
+ url (str): YouTubeビデオのURL。
10
+ language (str): トランスクリプトの言語。
11
+
12
+ Returns:
13
+ list: ドキュメントオブジェクトのリスト。
14
+ """
15
+ loader = YoutubeLoader.from_youtube_url(url, language=language, add_video_info=False)
16
+ return loader.load()
17
+
18
+ # Gradioインターフェースの作成
19
+ iface = gr.Interface(
20
+ fn=load_youtube_transcript,
21
+ inputs=[
22
+ gr.Textbox(label="YouTubeビデオのURL", lines=1),
23
+ gr.Dropdown(label="言語", choices=["ja", "en"], value="ja"),
24
+ ],
25
+ outputs=[
26
+ gr.Textbox(label="トランスクリプト", lines=10),
27
+ ],
28
+ title="YouTubeトランスクリプト取得",
29
+ description="YouTubeビデオのURLと言語を入力して、トランスクリプトを取得してください。"
30
+ )
31
+
32
+ # インターフェースの実行
33
+ iface.launch(share=True)