Update app.py

app.py

@@ -44,20 +44,27 @@ def save_comments(comments):
         json.dump(comments, file)
 
 def download_audio(video_url):
-    yt = YouTube(video_url)
-    audio = yt.streams.filter(only_audio=True).first()
-    audio_path = audio.download(output_path=".")
-    
-    file_stats = os.stat(audio_path)
-    logging.info(f'Size of audio file in Bytes: {file_stats.st_size}')
-    
-    if file_stats.st_size <= 30000000:  # Check the file size limit
-        base, ext = os.path.splitext(audio_path)
-        new_file = base + '.mp3'
-        os.rename(audio_path, new_file)
-        return new_file
-    else:
-        logging.error('Videos for transcription on this space are limited to about 1.5 hours. Please contact support for more information.')
+    try:
+        yt = YouTube(video_url)
+        audio = yt.streams.filter(only_audio=True).first()
+        if audio is None:
+            logging.error('오디오 스트림을 찾을 수 없습니다.')
+            return None
+        audio_path = audio.download(output_path=".")
+        
+        file_stats = os.stat(audio_path)
+        logging.info(f'오디오 파일 크기(Bytes): {file_stats.st_size}')
+        
+        if file_stats.st_size <= 30000000:  # 파일 크기 제한 확인
+            base, ext = os.path.splitext(audio_path)
+            new_file = base + '.mp3'
+            os.rename(audio_path, new_file)
+            return new_file
+        else:
+            logging.error('파일 크기가 너무 큽니다. 1.5시간 이하의 비디오만 지원됩니다.')
+            return None
+    except Exception as e:
+        logging.error(f"오디오 다운로드 중 오류 발생: {str(e)}")
         return None
 
 def generate_transcript(audio_path):
@@ -68,7 +75,7 @@ def generate_transcript(audio_path):
         result = model.transcribe(audio_path)
         return result['text'].strip()
     except Exception as e:
-        logging.error(f"Exception during transcription: {str(e)}")
+        logging.error(f"전사 중 오류 발생: {str(e)}")
         return f"전사 중 오류가 발생했습니다: {str(e)}"
 
 def generate_reply(comment_text, system_prompt):
@@ -93,7 +100,7 @@ def get_video_comments(video_id):
         request = youtube.commentThreads().list(
             part='snippet',
             videoId=video_id,
-            maxResults=100,   #댓글 읽어들이는 수 정의
+            maxResults=100,   # 댓글 읽어들이는 수 정의
             textFormat='plainText'
         )
         response = request.execute()
@@ -113,7 +120,7 @@ def get_video_comments(video_id):
                     part='snippet',
                     videoId=video_id,
                     pageToken=response['nextPageToken'],
-                    maxResults=100,  #댓글 읽어들이는 수 정의
+                    maxResults=100,  # 댓글 읽어들이는 수 정의
                     textFormat='plainText'
                 )
                 response = request.execute()