From a9043361ecd6c85e30283035016cac85ebaf86e0 Mon Sep 17 00:00:00 2001
From: Jerry Yan <792602257@qq.com>
Date: Sun, 20 Apr 2025 10:58:14 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E9=80=9A=E8=BF=87env?=
 =?UTF-8?q?=E8=8E=B7=E5=8F=96encoder=20args?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env             | 8 ++++++--
 entity/ffmpeg.py | 3 ++-
 index.py         | 2 +-
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/.env b/.env
index 313cda5..3cba4e0 100644
--- a/.env
+++ b/.env
@@ -1,4 +1,8 @@
 TEMPLATE_DIR=template/
-API_ENDPOINT=http://127.0.0.1:8030/task/v1
+API_ENDPOINT=https://zhentuai.com/task/v1
 ACCESS_KEY=TEST_ACCESS_KEY
-TEMP_DIR=tmp/
\ No newline at end of file
+TEMP_DIR=tmp/
+# QSV
+ENCODER_ARGS="-c:v h264_qsv -global_quality 28 -look_ahead 1"
+# NVENC
+#ENCODER_ARGS="-c:v h264_nvenc -rc cbr_ld_hq -b:v 3000k -preset p7 -profile high -bf 0 -bf-strategy 0 -rc-lookahead 0 -qp 28 -spatial-aq 1 -temporal-aq 1 -aq-strength 8 -cq 28"
\ No newline at end of file
diff --git a/entity/ffmpeg.py b/entity/ffmpeg.py
index 1d6bb51..bb2059d 100644
--- a/entity/ffmpeg.py
+++ b/entity/ffmpeg.py
@@ -1,10 +1,11 @@
 import json
+import os
 import time
 import uuid
 from typing import Any
 
 DEFAULT_ARGS = ("-shortest",)
-ENCODER_ARGS = ("-c:v", "h264_qsv", "-global_quality", "28", "-look_ahead", "1", )
+ENCODER_ARGS = ("-c:v", "h264", ) if not os.getenv("ENCODER_ARGS", False) else os.getenv("ENCODER_ARGS", "").split(" ")
 VIDEO_ARGS = ("-profile:v", "high", "-level:v", "4", )
 AUDIO_ARGS = ("-c:a", "aac", "-b:a", "128k", "-ar", "48000", "-ac", "2", )
 MUTE_AUDIO_INPUT = ("-f", "lavfi", "-i", "anullsrc=cl=stereo:r=48000", )
diff --git a/index.py b/index.py
index e7117b3..0ff5ae0 100644
--- a/index.py
+++ b/index.py
@@ -1,7 +1,7 @@
 from time import sleep
 
-import biz.task
 import config
+import biz.task
 from telemetry import init_opentelemetry
 from template import load_local_template
 from util import api