AIRider commited on
Commit
828fe8d
ยท
verified ยท
1 Parent(s): 315190f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -15,14 +15,14 @@ def get_video_id(youtube_url):
15
  return video_id_match.group(0) if video_id_match else None
16
 
17
  # ์œ ํŠœ๋ธŒ ์ž๋ง‰์„ ์š”์ฒญํ•˜๋Š” ํ•จ์ˆ˜
18
- def get_youtube_transcript(youtube_url, lang_code):
19
  # ๋น„๋””์˜ค ID ์ถ”์ถœ
20
  video_id = get_video_id(youtube_url)
21
  if video_id is None:
22
  return {"error": "์ž˜๋ชป๋œ ์œ ํŠœ๋ธŒ URL์ž…๋‹ˆ๋‹ค. ๋น„๋””์˜ค ID๋ฅผ ์ฐพ์„ ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค."}
23
 
24
  url = f"https://{RAPIDAPI_HOST}/transcript"
25
- querystring = {"video_id": video_id, "lang": lang_code}
26
 
27
  headers = {
28
  "x-rapidapi-key": RAPIDAPI_KEY,
@@ -39,9 +39,9 @@ def get_youtube_transcript(youtube_url, lang_code):
39
  return {"error": str(e)}
40
 
41
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ •์˜
42
- def youtube_transcript_interface(youtube_url, lang_code):
43
  # ์ž๋ง‰ ๋ฐ์ดํ„ฐ ๊ฐ€์ ธ์˜ค๊ธฐ
44
- transcript_data = get_youtube_transcript(youtube_url, lang_code)
45
 
46
  # JSON ํฌ๋งท์œผ๋กœ ์ถœ๋ ฅ
47
  return json.dumps(transcript_data, ensure_ascii=False, indent=2)
@@ -49,10 +49,10 @@ def youtube_transcript_interface(youtube_url, lang_code):
49
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ƒ์„ฑ
50
  interface = gr.Interface(
51
  fn=youtube_transcript_interface,
52
- inputs=["text", "text"],
53
  outputs="text",
54
  title="YouTube ์ž๋ง‰ ์ถ”์ถœ๊ธฐ",
55
- description="์œ ํŠœ๋ธŒ URL๊ณผ ์–ธ์–ด ์ฝ”๋“œ๋ฅผ ์ž…๋ ฅํ•˜์„ธ์š” (์˜ˆ: 'ko' for Korean)."
56
  )
57
 
58
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์‹คํ–‰
 
15
  return video_id_match.group(0) if video_id_match else None
16
 
17
  # ์œ ํŠœ๋ธŒ ์ž๋ง‰์„ ์š”์ฒญํ•˜๋Š” ํ•จ์ˆ˜
18
+ def get_youtube_transcript(youtube_url):
19
  # ๋น„๋””์˜ค ID ์ถ”์ถœ
20
  video_id = get_video_id(youtube_url)
21
  if video_id is None:
22
  return {"error": "์ž˜๋ชป๋œ ์œ ํŠœ๋ธŒ URL์ž…๋‹ˆ๋‹ค. ๋น„๋””์˜ค ID๋ฅผ ์ฐพ์„ ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค."}
23
 
24
  url = f"https://{RAPIDAPI_HOST}/transcript"
25
+ querystring = {"video_id": video_id}
26
 
27
  headers = {
28
  "x-rapidapi-key": RAPIDAPI_KEY,
 
39
  return {"error": str(e)}
40
 
41
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ •์˜
42
+ def youtube_transcript_interface(youtube_url):
43
  # ์ž๋ง‰ ๋ฐ์ดํ„ฐ ๊ฐ€์ ธ์˜ค๊ธฐ
44
+ transcript_data = get_youtube_transcript(youtube_url)
45
 
46
  # JSON ํฌ๋งท์œผ๋กœ ์ถœ๋ ฅ
47
  return json.dumps(transcript_data, ensure_ascii=False, indent=2)
 
49
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์ƒ์„ฑ
50
  interface = gr.Interface(
51
  fn=youtube_transcript_interface,
52
+ inputs="text",
53
  outputs="text",
54
  title="YouTube ์ž๋ง‰ ์ถ”์ถœ๊ธฐ",
55
+ description="์œ ํŠœ๋ธŒ URL์„ ์ž…๋ ฅํ•˜์„ธ์š”."
56
  )
57
 
58
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค ์‹คํ–‰