Update app.py

app.py

@@ -6,7 +6,7 @@ from faster_whisper import WhisperModel
 from pytube import YouTube
 from pytube.exceptions import VideoUnavailable, PytubeError
 
-# Conver Video to audio
+# 비디오를 MP3로 변환하는 함수
 def convert_mp4_to_mp3(video_file_path, output_dir):
     video = VideoFileClip(video_file_path)
     audio = video.audio
@@ -16,7 +16,7 @@ def convert_mp4_to_mp3(video_file_path, output_dir):
     video.close()
     return output_path
 
-# using Whisper Model 
+# Whisper 모델을 사용하여 MP3 파일을 텍스트로 변환하는 함수
 def transcribe_audio(model_size, audio_file):
     model = WhisperModel(model_size, device="cpu", compute_type="int8")
     start_time = time.time()
@@ -40,35 +40,41 @@ 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 to download video 
+# YouTube URL에서 비디오를 다운로드하는 함수
 def download_youtube_video(url, output_dir):
     try:
         yt = YouTube(url)
         stream = yt.streams.filter(file_extension='mp4').first()
         output_path = stream.download(output_dir)
-        return output_path
+        return output_path, None
     except VideoUnavailable:
         return None, "Video unavailable. Please check the URL."
     except PytubeError as e:
         return None, f"An error occurred: {e}"
 
-# function for process video
+# Gradio 인터페이스에서 사용할 메인 함수
 def process_video(model_size, video_file=None, video_url=None):
     if video_url and not video_file:
+        print(f"Downloading video from URL: {video_url}")
         video_file_path, error = download_youtube_video(video_url, '/tmp')
         if error:
+            print(f"Error downloading video: {error}")
             return error
+        print(f"Downloaded video to: {video_file_path}")
     elif video_file and not video_url:
         video_file_path = video_file.name
+        print(f"Using uploaded video file: {video_file_path}")
     else:
         return "Please upload a video file or provide a video URL, but not both."
 
     save_path = "/tmp"
     mp3_file_path = convert_mp4_to_mp3(video_file_path, save_path)
+    print(f"Converted video to MP3: {mp3_file_path}")
     transcription = transcribe_audio(model_size, mp3_file_path)
+    print(f"Transcription complete")
     return transcription
 
-# Gradio interface
+# Gradio 인터페이스 정의
 iface = gr.Interface(
     fn=process_video,
     inputs=[