Create app.py

app.py

@@ -0,0 +1,51 @@
+import gradio as gr
+import whisper
+import os
+from pytube import YouTube
+from moviepy.editor import VideoFileClip
+
+model = whisper.load_model("base")
+
+def download_youtube_audio(url):
+    yt = YouTube(url)
+    stream = yt.streams.filter(only_audio=True).first()
+    audio_file = stream.download(filename="yt_audio.mp4")
+    return audio_file
+
+def transcribe_from_file(file):
+    filename, ext = os.path.splitext(file)
+    if ext == ".mp4":
+        video = VideoFileClip(file)
+        audio_path = filename + ".mp3"
+        video.audio.write_audiofile(audio_path)
+        result = model.transcribe(audio_path)
+        os.remove(audio_path)
+    else:
+        result = model.transcribe(file)
+    return result["text"]
+
+def transcribe_combined(file, url):
+    if url:
+        try:
+            downloaded = download_youtube_audio(url)
+            return transcribe_from_file(downloaded)
+        except Exception as e:
+            return f"❌ 유튜브 다운로드 실패: {str(e)}"
+    elif file:
+        return transcribe_from_file(file)
+    else:
+        return "⚠️ 파일이나 URL 중 하나를 입력해주세요."
+
+demo = gr.Interface(
+    fn=transcribe_combined,
+    inputs=[
+        gr.File(label="🎬 파일 업로드 (선택)", file_types=[".mp3", ".wav", ".mp4"]),
+        gr.Textbox(label="📎 YouTube 링크 (선택)", placeholder="https://youtu.be/...")
+    ],
+    outputs="text",
+    title="🎧 Whisper 전사기 + YouTube 지원",
+    description="mp4/mp3/wav 파일을 업로드하거나 YouTube 영상 주소를 붙여넣어 텍스트로 전사하세요!",
+    allow_flagging="never"
+)
+
+demo.launch()