app.py

# URL To YoutubeID
from urllib.parse import urlparse
from youtube_transcript_api import YouTubeTranscriptApi
from youtube_transcript_api.formatters import TextFormatter
import openai
import gradio as gr

# 유튜브 비디오 ID 추출 함수
def get_yt_video_id(url):
    from urllib.parse import urlparse, parse_qs

    if url.startswith(('youtu', 'www')):
        url = 'http://' + url
        
    query = urlparse(url)
    
    if 'youtube' in query.hostname:
        if query.path == '/watch':
            return parse_qs(query.query)['v'][0]
        elif query.path.startswith(('/embed/', '/v/')):
            return query.path.split('/')[2]
    elif 'youtu.be' in query.hostname:
        return query.path[1:]
    else:
        raise ValueError("유효한 유튜브 링크가 아닙니다.")

# 유튜브 영상의 자막을 가져오는 함수
def transcribe(youtubeId):
    transcription = YouTubeTranscriptApi.get_transcript(youtubeId)
    return transcription

# 자막을 텍스트로 변환하는 함수
formatter = TextFormatter()

def transcriptToText(transcript):
    text = formatter.format_transcript(transcript)
    text = text.replace("\n", " ")
    return text

# 텍스트를 요약하는 함수 (OpenAI API 사용)
def textToSummary(text, OpenAIkey):
    openai.api_key = OpenAIkey
    response = openai.Completion.create(
        model="text-davinci-003",
        prompt="Summarize this in 200 words or less:\n\n" + text,
        temperature=0.7,
        max_tokens=400,
        top_p=1.0,
        frequency_penalty=0.0,
        presence_penalty=1
    )
    return response["choices"][0]["text"].replace("\n", " ").strip()

# 전체 요약 프로세스를 처리하는 함수
def summarize(url, OpenAIkey):
    try:
        videoId = get_yt_video_id(url)
        transcript = transcribe(videoId)
        text = transcriptToText(transcript)
        summary = textToSummary(text, OpenAIkey)
        return summary
    except Exception as e:
        return f"요약에 실패했습니다: {str(e)}"

# Gradio 인터페이스 설정
description = "요약할 유튜브 동영상 링크를 입력하세요"

gr.Interface(fn=summarize,
             inputs=["text", "text"],
             outputs="textbox", 
             description=description
            ).launch()