Update app.py

app.py

@@ -12,40 +12,57 @@ def search_youtube_videos(keyword, limit=5, order='date'):
     video_urls = [video['link'] for video in results['result']]
     return video_urls
 
-def get_transcript(url):
-    loader = YoutubeLoader()
-    transcript = loader.load(url)
-    text = " ".join([segment['text'] for segment in transcript['segments']])
-    return text
+# Gradio 인터페이스 설정
+interface = gr.Interface(
+    fn=search_youtube_videos,
+    inputs=[
+        gr.Textbox(label="검색 키워드", placeholder="YouTube에서 검색할 키워드를 입력하세요."),
+        gr.Slider(minimum=1, maximum=10, default=5, label="검색 결과 개수"),
+        gr.Radio(choices=['date', 'relevance', 'rating', 'title', 'viewCount'], label="정렬 순서", default='date')
+    ],
+    outputs=gr.Textbox(label="검색 결과"),
+    title="YouTube 비디오 검색",
+    description="YouTube에서 비디오를 검색하고 결과의 링크를 받아보세요."
+)
 
-def summarize_text(text):
-    response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=f"요약: {text}",
-        max_tokens=150
-    )
-    return response.choices[0].text.strip()
+# Gradio 앱 실행
+if __name__ == "__main__":
+    interface.launch()
+    
+# def get_transcript(url):
+#     loader = YoutubeLoader()
+#     transcript = loader.load(url)
+#     text = " ".join([segment['text'] for segment in transcript['segments']])
+#     return text
 
-def process_keyword(keyword):
-    video_urls = search_youtube_videos(keyword)
-    summaries = []
-    for url in video_urls:
-        try:
-            text = get_transcript(url)
-            summary = summarize_text(text)
-            summaries.append(f"URL: {url}\nSummary: {summary}\n")
-        except Exception as e:
-            summaries.append(f"URL: {url}\nError: {str(e)}\n")
-    return "\n".join(summaries)
+# def summarize_text(text):
+#     response = openai.Completion.create(
+#         engine="text-davinci-003",
+#         prompt=f"요약: {text}",
+#         max_tokens=150
+#     )
+#     return response.choices[0].text.strip()
 
-# Gradio 인터페이스
-interface = gr.Interface(
-    fn=process_keyword,
-    inputs=gr.Textbox(label="검색 키워드"),
-    outputs=gr.Textbox(label="결과"),
-)
+# def process_keyword(keyword):
+#     video_urls = search_youtube_videos(keyword)
+#     summaries = []
+#     for url in video_urls:
+#         try:
+#             text = get_transcript(url)
+#             summary = summarize_text(text)
+#             summaries.append(f"URL: {url}\nSummary: {summary}\n")
+#         except Exception as e:
+#             summaries.append(f"URL: {url}\nError: {str(e)}\n")
+#     return "\n".join(summaries)
+
+# # Gradio 인터페이스
+# interface = gr.Interface(
+#     fn=process_keyword,
+#     inputs=gr.Textbox(label="검색 키워드"),
+#     outputs=gr.Textbox(label="결과"),
+# )
 
-interface.launch()
+# interface.launch()
 
 
 # from youtubesearchpython import VideosSearch