Update app.py

app.py

@@ -70,18 +70,25 @@ class MyClient(discord.Client):
         try:
             video_id = extract_video_id(message.content)
             if video_id:
-                transcript = await get_best_available_transcript(video_id)
+                transcript, language = await get_best_available_transcript(video_id)
                 comments = await get_video_comments(video_id)
-                if comments and transcript:
-                    replies = await generate_replies(comments, transcript)
-                    await create_thread_and_send_replies(message, video_id, comments, replies, self.session)
+                if comments:
+                    if transcript:
+                        replies = await generate_replies(comments, transcript)
+                        await create_thread_and_send_replies(message, video_id, comments, replies, self.session)
+                    else:
+                        await message.channel.send("자막을 가져올 수 없습니다. 댓글만을 기반으로 답변을 생성합니다.")
+                        replies = await generate_replies(comments, "")
+                        await create_thread_and_send_replies(message, video_id, comments, replies, self.session)
                 else:
-                    await message.channel.send("자막이나 댓글을 가져올 수 없습니다.")
+                    await message.channel.send("댓글을 가져올 수 없습니다.")
             else:
                 await message.channel.send("유효한 유튜브 비디오 URL을 제공해 주세요.")
         finally:
             self.is_processing = False
 
+
+
     def is_message_in_specific_channel(self, message):
         return message.channel.id == SPECIFIC_CHANNEL_ID or (
             isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
@@ -106,34 +113,43 @@ def extract_video_id(url):
     logging.debug(f'추출된 비디오 ID: {video_id}')
     return video_id
 
-async def get_best_available_transcript(video_id, max_retries=3, delay=5):
+import asyncio
+import aiohttp
+from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound
+
+async def get_best_available_transcript(video_id, max_retries=5, delay=10):
+    async def fetch_transcript(language):
+        try:
+            transcript = await asyncio.to_thread(YouTubeTranscriptApi.get_transcript, video_id, languages=[language])
+            return transcript, language
+        except Exception as e:
+            logging.warning(f'{language} 자막 가져오기 오류: {e}')
+            return None, None
+
     for attempt in range(max_retries):
+        ko_transcript, ko_lang = await fetch_transcript('ko')
+        if ko_transcript:
+            return ko_transcript, ko_lang
+
+        en_transcript, en_lang = await fetch_transcript('en')
+        if en_transcript:
+            return en_transcript, en_lang
+
         try:
-            transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko'])
+            transcripts = await asyncio.to_thread(YouTubeTranscriptApi.list_transcripts, video_id)
+            manual_transcript = await asyncio.to_thread(transcripts.find_manually_created_transcript)
+            transcript = await asyncio.to_thread(manual_transcript.fetch)
+            return transcript, manual_transcript.language_code
         except Exception as e:
-            logging.warning(f'한국어 자막 가져오기 오류: {e}')
-            try:
-                transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=['en'])
-            except Exception as e:
-                logging.warning(f'영어 자막 가져오기 오류: {e}')
-                try:
-                    transcripts = YouTubeTranscriptApi.list_transcripts(video_id)
-                    transcript = transcripts.find_manually_created_transcript().fetch()
-                except Exception as e:
-                    if attempt < max_retries - 1:
-                        logging.error(f'대체 자막 가져오기 오류: {e}')
-                        await asyncio.sleep(delay)
-                        continue
-                    else:
-                        logging.error(f'모든 자막 가져오기 시도 실패: {e}')
-                        return None
+            if attempt < max_retries - 1:
+                logging.error(f'모든 자막 가져오기 시도 실패 (시도 {attempt + 1}/{max_retries}): {e}')
+                await asyncio.sleep(delay)
+            else:
+                logging.error(f'최종 자막 가져오기 실패: {e}')
+                return None, None
 
-        formatter = TextFormatter()
-        transcript_text = formatter.format_transcript(transcript)
-        logging.debug(f'가져온 자막: {transcript_text}')
-        return transcript_text
+    return None, None
 
-    return None
 
 async def get_video_comments(video_id):
     comments = []