from youtubesearchpython import VideosSearch
import gradio as gr

def search_youtube_videos(keyword, limit=5):
    videos_search = VideosSearch(keyword, limit=limit)
    results = videos_search.result()

    video_urls = [video['link'] for video in results['result']]
    return video_urls

def gradio_interface(keyword):
    video_urls = search_youtube_videos(keyword, limit=5)
    return "\n".join(video_urls)

interface = gr.Interface(
    fn=gradio_interface,
    inputs=gr.Textbox(label="검색 키워드를 입력하세요"),
    outputs=gr.Textbox(label="검색된 유튜브 동영상 URL"),
    title="유튜브 검색 도우미",
    description="키워드를 입력하면 유튜브에서 해당 키워드로 검색한 후 동영상 URL을 보여줍니다."
)

if __name__ == "__main__":
    interface.launch()