app.py

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()