Update app.py

app.py

@@ -0,0 +1,72 @@
+# 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()