import subprocess
import os

# 使用当前工作目录存储文本文件
text_file = os.path.join(os.getcwd(), 'temp_text.txt')
with open(text_file, 'w', encoding='utf-8') as f:
    f.write('There is a test')

vf = f"drawtext=textfile='temp_text.txt':fontsize=50:fontcolor=yellow:borderw=3:bordercolor=red:x=(w-text_w)/2:y=610"
print('vf:', vf)

args = ['ffmpeg', '-y', '-i', 'video1/frames/frame_001.jpg', '-vf', vf, '-frames:v', '1', 'test_output.jpg']
result = subprocess.run(args, capture_output=True, text=True)
print('stdout:', result.stdout)
print('stderr:', result.stderr[-500:] if len(result.stderr) > 500 else result.stderr)
print('returncode:', result.returncode)

# 清理
os.remove(text_file)
