ALBERT SITE

from youtube_search import YoutubeSearch
from langchain_community.document_loaders import YoutubeLoader
from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
from langchain_classic.chains.combine_documents import create_stuff_documents_chain
from dotenv import load_dotenv

load_dotenv()

model = ChatOpenAI(model="gpt-4o-mini")

' 내가 좋아하는 고양이 채널 크집사 동영상 가져옴
videos = YoutubeSearch("크집사", max_results=3).to_dict()

' 영상 길이가 60분 이하인 영상만 남깁니다.
videos = [v for v in videos if len(v['duration'].split(':')) < 3]

for v in videos:
    ' url_suffix를 이용하여 video_url을 만듭니다.
    v['video_url'] = 'https://youtube.com' + v['url_suffix']

    ' YoutubeLoader를 이용하여 비디오를 로드합니다.
    loader = YoutubeLoader.from_youtube_url(
        v['video_url'],
        language=['ko', 'en']
    )

    v['content'] = loader.load()
    ' print(v['content'])

' 동영상 요약 프롬프트 작성
prompt = ChatPromptTemplate.from_messages(
    [("system", "다음 영상에 대한 요약을 한국어로 만들어줘 :\\n\\n{context}")]
)

chain = create_stuff_documents_chain(model, prompt)

i = 1

for v in videos:
    v['summary'] = chain.invoke({"context": v['content']})
    print(f'------ {i} ------')
    print(v['summary'])
    i += 1

------ 1 ------
이 영상은 고양이 '토토'와 '디디'의 일상과 성장 이야기를 담고 있습니다. 엄마가 맛있는 계란을 주며 두 고양이를 부르고, 토토는 자신이 가장 큰 고양이라는 사실을 깨닫고 점점 야망을 가지게 됩니다. 토토가 배고픔을 표현하고 사람과의 커뮤니케이션을 시도하는 모습, 그리고 디디와의 피지컬 비교를 통해 토토가 많이 성장했음을 보여줍니다.

영상에서는 두 고양이의 성격과 행동 차이도 강조되며, 특히 최근에 토토가 자신의 힘을 인식하고 엄마에게 도전하는 모습이 유머러스하게 묘사됩니다. 또, 엄마는 토토의 힘이 세졌음을 보여주기 위해 때때로 강하게 대처하기도 합니다. 영상의 결론은 고양이와의 소통과 함께 엄마의 근육 발전으로 더욱 훌륭한 집사가 되었다는 내용으로 마무리됩니다.

------ 2 ------
이 영상은 고양이와 강아지의 일상적인 모습과 그 속에서의 재미있는 에피소드를 담고 있습니다. 주인공 루루와 코코는 새로운 캣타워를 보고 신나지만, 디디는 들어갈 수 없어서 안타까운 상황이 발생합니다. 이어서, 주인공은 친구의 멍멍이 덕배를 10일 동안 맡게 된 경험을 이야기합니다. 덕배는 처음에는 고양이들과 어색했지만, 시간이 지날수록 재미있는 상황들이 펼쳐집니다.

주인공은 멍멍이와 함께하는 것이 쉽지 않다는 것을 깨닫고, 멍멍이의 감정 표현과 행동에 대해서도 다양한 이야기를 나눕니다. 마지막에는 캣타워 조립이 완료되고, 새로운 공간에서 아기 고양이들이 자리를 잡는 모습으로 마무리됩니다. 전체적으로 고양이와 강아지의 귀엽고 유쾌한 일상이 잘 표현된 영상입니다.

------ 3 ------
이 영상은 귀여운 고양이 티티와 라라의 일상 모습을 담고 있습니다. 티티는 일이 바쁜 주인 옆에서 애교를 부리며 기다리고, 고양이 면접 같은 재미있는 아이디어를 제안합니다. 루루는 기운이 없어 보이고, 라라는 적극적으로 주인의 관심을 끌려고 합니다. 영상에서는 이들의 다양한 행동과 주인과의 소통이 사랑스럽게 그려지며, 고양이들의 사랑스러운 매력을 느낄 수 있습니다.