Update app.py

app.py

@@ -3,8 +3,6 @@ import gradio as gr
 import time
 from moviepy.editor import VideoFileClip
 from faster_whisper import WhisperModel
-from pytube import YouTube
-from pytube.exceptions import VideoUnavailable, PytubeError
 
 # 비디오를 MP3로 변환하는 함수
 def convert_mp4_to_mp3(video_file_path, output_dir):
@@ -40,38 +38,19 @@ 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에서 비디오를 다운로드하는 함수
-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, None
-    except VideoUnavailable:
-        return None, "Video unavailable. Please check the URL."
-    except PytubeError as e:
-        return None, f"An error occurred: {e}"
-
 # 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."
+def process_video(model_size, video_file=None):
+    if not video_file:
+        return "Please upload a video file."
+
+    video_file_path = video_file.name
+    print(f"Using uploaded video file: {video_file_path}")
 
     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")
+    print("Transcription complete")
     return transcription
 
 # Gradio 인터페이스 정의
@@ -79,12 +58,11 @@ iface = gr.Interface(
     fn=process_video,
     inputs=[
         gr.Dropdown(["tiny", "base", "small", "medium", "large"], label="Model Size"),
-        gr.File(label="Upload Video File"),
-        gr.Textbox(label="Video URL")
+        gr.File(label="Upload Video File")
     ],
     outputs="text",
     title="Video to Text Converter using Whisper",
-    description="Upload a video file or provide a video URL, select the Whisper model size, and get the transcribed text.",
+    description="Upload a video file, select the Whisper model size, and get the transcribed text.",
     live=True
 )