Update app.py

app.py

@@ -6,7 +6,7 @@ import asyncio
 import aiohttp
 from huggingface_hub import InferenceClient
 from googleapiclient.discovery import build
-from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api import YouTubeTranscriptApi, NoTranscriptFound
 from youtube_transcript_api.formatters import TextFormatter
 from dotenv import load_dotenv
 import json
@@ -98,9 +98,12 @@ async def get_best_available_transcript(video_id):
         transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=['ko'])
         formatter = TextFormatter()
         return formatter.format_transcript(transcript)
+    except NoTranscriptFound:
+        logging.error(f"No transcript found for video ID {video_id}")
+        return "자막을 찾을 수 없습니다."
     except Exception as e:
-        logging.error(f"Failed to retrieve transcript: {e}")
-        return None
+        logging.error(f"Failed to retrieve transcript for video ID {video_id}: {e}")
+        return "자막을 추출하는 데 실패했습니다."
 
 async def get_video_comments(video_id):
     try:
@@ -133,7 +136,7 @@ async def generate_reply(comment, transcript):
                 return response_json.get("generated_text", "").strip()
             except json.JSONDecodeError:
                 logging.error(f"Failed to decode JSON: {response}")
-                return "답변을 생성할 수 없습니다."
+                return response
     except Exception as e:
         logging.error(f"Error generating reply: {e}")
         return "답변을 생성할 수 없습니다."
@@ -144,25 +147,26 @@ async def create_thread_and_send_reply(message, video_id, comment, reply):
         embed = discord.Embed(description=f"**댓글**: {comment[0]}\n**답글**: {reply}")
         await thread.send(embed=embed)
         webhook_data = {"video_id": video_id, "replies": [{"comment": comment[0], "reply": reply, "comment_id": comment[1]}]}
-        await send_webhook_data(webhook_data, self.session)
+        await send_webhook_data(webhook_data)
     except discord.HTTPException as e:
         if e.code == 160004:
             logging.error("A thread has already been created for this message")
         else:
             logging.error(f"Error in thread creation and reply sending: {e}")
 
-async def send_webhook_data(data, session):
-    data_json = json.dumps(data)
-    for i in range(0, len(data_json), MAX_CHUNK_SIZE):
-        chunk = data_json[i:i+MAX_CHUNK_SIZE]
-        try:
-            async with session.post(WEBHOOK_URL, json=json.loads(chunk)) as response:
-                if response.status == 200:
-                    logging.info("Webhook data sent successfully.")
-                else:
-                    logging.error(f"Failed to send webhook data: HTTP {response.status}")
-        except aiohttp.ClientError as e:
-            logging.error(f"HTTP error occurred while sending webhook data: {e}")
+async def send_webhook_data(data):
+    async with aiohttp.ClientSession() as session:
+        data_json = json.dumps(data)
+        for i in range(0, len(data_json), MAX_CHUNK_SIZE):
+            chunk = data_json[i:i+MAX_CHUNK_SIZE]
+            try:
+                async with session.post(WEBHOOK_URL, json=json.loads(chunk)) as response:
+                    if response.status == 200:
+                        logging.info("Webhook data sent successfully.")
+                    else:
+                        logging.error(f"Failed to send webhook data: HTTP {response.status}")
+            except aiohttp.ClientError as e:
+                logging.error(f"HTTP error occurred while sending webhook data: {e}")
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)