import os
import json
import logging
import warnings
import multiprocessing
import sys
from dotenv import load_dotenv
load_dotenv()

warnings.filterwarnings("ignore")
logging.getLogger("httpx").setLevel(logging.WARNING)



from VideoAgent import VideoRAG, QueryParam



if __name__ == '__main__':

    multiprocessing.set_start_method('spawn', force=True)

    # video文件地址:
    video_base_path = f'/root/huangjie/VideoAgent_api513/videos/origin'
    video_files = sorted(os.listdir(video_base_path))
    video_paths = [os.path.join(video_base_path, f) for f in video_files]
    
    #工作目录
    videorag = VideoRAG( working_dir=f"/root/huangjie/VideoAgent_api513/working_dir")    
    videorag.insert_video(video_path_list=video_paths)
 
    querys = "SP视频开头前10秒的内容"
    response = videorag.query(query=querys, param=QueryParam)
    print("ans:\n\n", response)