Update app.py

app.py

@@ -8,6 +8,10 @@ from fpdf import FPDF
 from fpdf.enums import XPos, YPos
 from datetime import datetime
 
+# RapidAPI 설정
+RAPIDAPI_KEY = os.getenv("RAPIDAPI_KEY")
+RAPIDAPI_HOST = "youtube-transcriptor.p.rapidapi.com"
+
 # 클라이언트 생성 함수
 def create_client(model_name):
     return InferenceClient(model_name, token=os.getenv("HF_TOKEN"))
@@ -21,6 +25,40 @@ def call_api(content, system_message, max_tokens, temperature, top_p):
     response = client.chat_completion(messages=messages, max_tokens=max_tokens, temperature=temperature, top_p=top_p, seed=random_seed)
     return response.choices[0].message.content
 
+# 유튜브 비디오 ID 추출 함수
+def get_video_id(youtube_url):
+    video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
+    return video_id_match.group(0) if video_id_match else None
+
+# 유튜브 트랜스크립트 추출 함수
+def get_transcript(youtube_url):
+    video_id = get_video_id(youtube_url)
+    if not video_id:
+        return "Invalid YouTube URL. Please enter a valid URL."
+
+    if not RAPIDAPI_KEY:
+        return "Error: RAPIDAPI_KEY is not set in the environment variables."
+
+    url = f"https://youtube-transcriptor.p.rapidapi.com/transcript?video_id={video_id}&lang=en"
+
+    headers = {
+        "X-RapidAPI-Key": RAPIDAPI_KEY,
+        "X-RapidAPI-Host": RAPIDAPI_HOST
+    }
+
+    try:
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()  # Raises a HTTPError if the status is 4xx, 5xx
+        transcript_data = response.json()
+        
+        if isinstance(transcript_data, list) and transcript_data:
+            transcript_text = " ".join([entry.get('text', '') for entry in transcript_data])
+            return f"Transcript:\n\n{transcript_text}"
+        else:
+            return "No transcript available or unexpected response format."
+    except requests.RequestException as e:
+        return f"Error fetching transcript: {str(e)}"
+
 # 정보 분석 함수
 def analyze_info(category, style, transcripts):
     transcript_list = transcripts.split("\n\n---\n\n")
@@ -41,28 +79,6 @@ def summarize_transcript(transcripts, system_message, max_tokens, temperature, t
     summary = call_api(transcripts, system_message, max_tokens, temperature, top_p)
     return summary
 
-# 유튜브 비디오 ID 추출 함수
-def get_video_id(youtube_url):
-    video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
-    return video_id_match.group(0) if video_id_match else None
-
-# 유튜브 트랜스크립트 추출 함수
-def get_transcript(youtube_url):
-    video_id = get_video_id(youtube_url)
-    if not video_id:
-        return "Invalid YouTube URL. Please enter a valid URL."
-
-    language_order = ['ko', 'en', 'ja', 'zh-Hans', 'pt', 'es', 'it', 'fr', 'de', 'ru']
-    for lang in language_order:
-        try:
-            transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=[lang])
-            return f"Transcript in {lang}:\n\n{' '.join([entry['text'] for entry in transcript])}"
-        except NoTranscriptFound:
-            continue
-        except Exception as e:
-            return f"Error: {str(e)}"
-    return "No transcript available in the specified languages."
-
 # 카테고리별 프롬프트 함수
 def get_blog_post_prompt(category):
     if category == "일반형":