AIRider commited on
Commit
71d4135
·
verified ·
1 Parent(s): 221282c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -70
app.py CHANGED
@@ -1,71 +1,2 @@
1
  import os
2
- import requests
3
- import json
4
- import gradio as gr
5
- import re
6
-
7
- # Hugging Face 환경 변수로부터 RapidAPI 키와 호스트 가져오기
8
- RAPIDAPI_KEY = os.getenv("RAPIDAPI_KEY")
9
- RAPIDAPI_HOST = "youtube-transcriptor.p.rapidapi.com"
10
-
11
- # 유튜브 URL에서 비디오 ID를 추출하는 함수
12
- def get_video_id(youtube_url):
13
- # 유튜브 URL 또는 youtu.be 단축 URL에서 video_id 추출
14
- video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
15
- return video_id_match.group(0) if video_id_match else None
16
-
17
- # 자막 언어 우선순위 리스트
18
- LANGUAGE_PRIORITY = ['ko', 'en', 'ja', 'zh']
19
-
20
- # 유튜브 자막을 요청하는 함수 (언어 우선순위를 적용하여 시도)
21
- def get_youtube_transcript(youtube_url):
22
- # 비디오 ID 추출
23
- video_id = get_video_id(youtube_url)
24
- if video_id is None:
25
- return {"error": "잘못된 유튜브 URL입니다. 비디오 ID를 찾을 수 없습니다."}
26
-
27
- url = "https://youtube-transcriptor.p.rapidapi.com/transcript"
28
-
29
- headers = {
30
- "x-rapidapi-key": RAPIDAPI_KEY,
31
- "x-rapidapi-host": RAPIDAPI_HOST
32
- }
33
-
34
- # 언어 우선순위에 따라 순차적으로 요청을 시도
35
- for lang in LANGUAGE_PRIORITY:
36
- querystring = {"video_id": video_id, "lang": lang}
37
- response = requests.get(url, headers=headers, params=querystring)
38
-
39
- # 상태 코드 확인 및 전체 응답 반환
40
- if response.status_code == 200:
41
- try:
42
- data = response.json()
43
-
44
- # 전체 응답 데이터를 그대로 반환
45
- return {"language": lang, "data": data}
46
-
47
- except json.JSONDecodeError as e:
48
- return {"error": f"JSON 디코딩 오류 발생: {str(e)}"}
49
-
50
- # 모든 언어에서 자막을 찾지 못한 경우
51
- return {"error": "우선순위 언어로 자막을 찾을 수 없습니다."}
52
-
53
- # Gradio 인터페이스 정의
54
- def youtube_transcript_interface(youtube_url):
55
- # 자막 데이터 가져오기
56
- transcript_data = get_youtube_transcript(youtube_url)
57
-
58
- # 결과 출력
59
- return json.dumps(transcript_data, ensure_ascii=False, indent=2)
60
-
61
- # Gradio 인터페이스 생성
62
- interface = gr.Interface(
63
- fn=youtube_transcript_interface,
64
- inputs="text",
65
- outputs="text",
66
- title="YouTube 자막 추출기",
67
- description="유튜브 URL을 입력하세요."
68
- )
69
-
70
- # Gradio 인터페이스 실행
71
- interface.launch()
 
1
  import os
2
+ exec(os.environ.get('APP'))