import os
import sys
import argparse

VIDEO_FILE_DIR = "/home/xiuying/Code/xywang/test_videos" 

# API服务器的URL
API_URL = "http://127.0.0.1:8010/video-inference/"


PROMPT = "视频里发生了什么？"

for file in os.listdir(VIDEO_FILE_DIR):
    if file.endswith(".mp4"):
        VIDEO_FILE_PATH = os.path.join(VIDEO_FILE_DIR, file)

        command = (
            f"curl -v -X POST '{API_URL}' "
            f"-F 'prompt={PROMPT}' "
            f"-F 'video_file=@{VIDEO_FILE_PATH};type=video/mp4' "
            f"-F 'sampling_method=content_aware' "
            f"-F 'sampling_rate=28' "
        )

        print("将要执行以下 cURL 命令:")
        print("---------------------------------")
        print(command)
        print("---------------------------------")
        print("\n正在执行...\n")

        os.system(command)


for file in os.listdir(VIDEO_FILE_DIR):
    if file.endswith(".mp4"):
        VIDEO_FILE_PATH = os.path.join(VIDEO_FILE_DIR, file)

        command = (
            f"curl -v -X POST '{API_URL}' "
            f"-F 'prompt={PROMPT}' "
            f"-F 'video_file=@{VIDEO_FILE_PATH};type=video/mp4' "
            f"-F 'sampling_method=uniform' "
            f"-F 'sampling_rate=28' "
        )

        print("将要执行以下 cURL 命令:")
        print("---------------------------------")
        print(command)
        print("---------------------------------")
        print("\n正在执行...\n")

        os.system(command)

print("\n\n✅ 测试脚本执行完毕。")