Update app.py

app.py

@@ -4,8 +4,9 @@ import time
 from moviepy.editor import VideoFileClip
 from faster_whisper import WhisperModel
 from pytube import YouTube
+from pytube.exceptions import VideoUnavailable, PytubeError
 
-# 비디오를 MP3로 변환하는 함수
+# Conver Video to audio
 def convert_mp4_to_mp3(video_file_path, output_dir):
     video = VideoFileClip(video_file_path)
     audio = video.audio
@@ -15,7 +16,7 @@ def convert_mp4_to_mp3(video_file_path, output_dir):
     video.close()
     return output_path
 
-# Whisper 모델을 사용하여 MP3 파일을 텍스트로 변환하는 함수
+# using Whisper Model 
 def transcribe_audio(model_size, audio_file):
     model = WhisperModel(model_size, device="cpu", compute_type="int8")
     start_time = time.time()
@@ -39,17 +40,24 @@ def transcribe_audio(model_size, audio_file):
 
     return f"{detected_language}\n\nTranscription:\n{result_text}\n\nElapsed time: {elapsed_time:.2f} seconds"
 
-# YouTube URL에서 비디오를 다운로드하는 함수
+# YouTube URL to download video 
 def download_youtube_video(url, output_dir):
-    yt = YouTube(url)
-    stream = yt.streams.filter(file_extension='mp4').first()
-    output_path = stream.download(output_dir)
-    return output_path
+    try:
+        yt = YouTube(url)
+        stream = yt.streams.filter(file_extension='mp4').first()
+        output_path = stream.download(output_dir)
+        return output_path
+    except VideoUnavailable:
+        return None, "Video unavailable. Please check the URL."
+    except PytubeError as e:
+        return None, f"An error occurred: {e}"
 
-# Gradio 인터페이스에서 사용할 메인 함수
+# function for process video
 def process_video(model_size, video_file=None, video_url=None):
     if video_url and not video_file:
-        video_file_path = download_youtube_video(video_url, '/tmp')
+        video_file_path, error = download_youtube_video(video_url, '/tmp')
+        if error:
+            return error
     elif video_file and not video_url:
         video_file_path = video_file.name
     else:
@@ -60,7 +68,7 @@ def process_video(model_size, video_file=None, video_url=None):
     transcription = transcribe_audio(model_size, mp3_file_path)
     return transcription
 
-# Gradio 인터페이스 정의
+# Gradio interface
 iface = gr.Interface(
     fn=process_video,
     inputs=[