AIRider commited on
Commit
64cd965
Β·
verified Β·
1 Parent(s): c8e948c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -13
app.py CHANGED
@@ -8,7 +8,6 @@ import re
8
  AA_KEY = os.getenv("AA_KEY")
9
  AA_HOST = "youtube-transcriptor.p.rapidapi.com"
10
 
11
- # 유튜브 URLμ—μ„œ λΉ„λ””μ˜€ IDλ₯Ό μΆ”μΆœν•˜λŠ” ν•¨μˆ˜
12
  def get_video_id(youtube_url):
13
  # λ‹€μ–‘ν•œ YouTube URL ν˜•μ‹μ— λŒ€ν•œ μ •κ·œν‘œν˜„μ‹ νŒ¨ν„΄
14
  patterns = [
@@ -47,21 +46,27 @@ def get_youtube_transcript(youtube_url):
47
  # μ–Έμ–΄ μš°μ„ μˆœμœ„μ— 따라 순차적으둜 μš”μ²­μ„ μ‹œλ„
48
  for lang in LANGUAGE_PRIORITY:
49
  querystring = {"video_id": video_id, "lang": lang}
50
- response = requests.get(url, headers=headers, params=querystring)
51
-
52
- # μƒνƒœ μ½”λ“œ 확인 및 전체 응닡 λ°˜ν™˜
53
- if response.status_code == 200:
54
- try:
55
- data = response.json()
56
-
57
- # 전체 응닡 데이터λ₯Ό κ·ΈλŒ€λ‘œ λ°˜ν™˜
58
- return {"language": lang, "data": data}
 
 
59
 
60
- except json.JSONDecodeError as e:
61
- return {"error": f"JSON λ””μ½”λ”© 였λ₯˜ λ°œμƒ: {str(e)}"}
 
 
 
 
62
 
63
  # λͺ¨λ“  μ–Έμ–΄μ—μ„œ μžλ§‰μ„ 찾지 λͺ»ν•œ 경우
64
- return {"error": "μš°μ„ μˆœμœ„ μ–Έμ–΄λ‘œ μžλ§‰μ„ 찾을 수 μ—†μŠ΅λ‹ˆλ‹€."}
65
 
66
  # Gradio μΈν„°νŽ˜μ΄μŠ€ μ •μ˜
67
  def youtube_transcript_interface(youtube_url):
 
8
  AA_KEY = os.getenv("AA_KEY")
9
  AA_HOST = "youtube-transcriptor.p.rapidapi.com"
10
 
 
11
  def get_video_id(youtube_url):
12
  # λ‹€μ–‘ν•œ YouTube URL ν˜•μ‹μ— λŒ€ν•œ μ •κ·œν‘œν˜„μ‹ νŒ¨ν„΄
13
  patterns = [
 
46
  # μ–Έμ–΄ μš°μ„ μˆœμœ„μ— 따라 순차적으둜 μš”μ²­μ„ μ‹œλ„
47
  for lang in LANGUAGE_PRIORITY:
48
  querystring = {"video_id": video_id, "lang": lang}
49
+ try:
50
+ response = requests.get(url, headers=headers, params=querystring)
51
+ response.raise_for_status() # 였λ₯˜ μƒνƒœ μ½”λ“œμ— λŒ€ν•΄ μ˜ˆμ™Έ λ°œμƒ
52
+
53
+ data = response.json()
54
+
55
+ # μžλ§‰μ΄ μ—†λŠ” 경우 처리
56
+ if "transcript" not in data or not data["transcript"]:
57
+ continue
58
+
59
+ return {"language": lang, "data": data}
60
 
61
+ except requests.exceptions.RequestException as e:
62
+ print(f"Error for language {lang}: {str(e)}")
63
+ continue
64
+ except json.JSONDecodeError as e:
65
+ print(f"JSON Decode Error for language {lang}: {str(e)}")
66
+ continue
67
 
68
  # λͺ¨λ“  μ–Έμ–΄μ—μ„œ μžλ§‰μ„ 찾지 λͺ»ν•œ 경우
69
+ return {"error": "μš°μ„ μˆœμœ„ μ–Έμ–΄λ‘œ μžλ§‰μ„ 찾을 수 μ—†μŠ΅λ‹ˆλ‹€. ν•΄λ‹Ή λ™μ˜μƒμ— μžλ§‰μ΄ μ—†κ±°λ‚˜ μ ‘κ·Όν•  수 μ—†μŠ΅λ‹ˆλ‹€."}
70
 
71
  # Gradio μΈν„°νŽ˜μ΄μŠ€ μ •μ˜
72
  def youtube_transcript_interface(youtube_url):